Prolog指定长度子集提取及剩余元素返回问题排查

需要实现一个Prolog谓词select_seg/4，功能是从给定列表中生成所有指定长度的可能子集（Segment），同时返回最终未被选中的元素组成的列表（Rest）。

现有代码如下：

select_seg(List, Segment, Rest, Size ):-    
    select_seg(List, Segment, Rest, Size, Size).

select_seg(_,_,_,_, 0):- !.    
select_seg(List, [Head_Segs|Tail_Segs],[Head_Rest|Tail_Rest], Size,Acc ):-        
    select(Head_Segs, List, Head_Rest),        
    Acc >= 0,         
    New_Acc is Acc  - 1,        
    select_seg(Head_Rest, Tail_Segs, Tail_Rest, Size, New_Acc).

调用示例：

select_seg([1,2,3,4,5,6,7,8,9], Seg, R ,3 ).

当前问题：返回的R是每次选择后的中间列表集合，而非最终未被选中的元素列表，需要修正代码实现期望输出。

原代码的问题在于递归过程中把每一步的剩余列表都累积到了Rest参数里，导致最终返回的是中间列表的集合。我们需要调整逻辑，只在递归终止时返回最终的剩余列表。

修正后的代码：

% 对外接口谓词
select_seg(List, Segment, Rest, Size) :-
    select_seg(List, Segment, Rest, Size).

% 递归终止条件：当需要选取的元素数量为0时，Segment为空，Rest即为当前剩余的列表
select_seg(Rest, [], Rest, 0) :- !.

% 递归步骤：从当前列表选一个元素加入Segment，剩余列表继续递归选取剩下的元素
select_seg(List, [Head|Tail], Rest, Size) :-
    Size > 0,
    NewSize is Size - 1,
    select(Head, List, TempRest),
    select_seg(TempRest, Tail, Rest, NewSize).

代码说明 #

1. 终止条件：当Size减到0时，说明已经选够了指定长度的元素，此时Segment为空列表，同时把当前的剩余列表直接赋值给Rest，这就是最终未被选中的元素集合。
2. 递归逻辑：每次从当前列表中选一个元素加入Segment，通过TempRest传递当前选择后的剩余列表，继续递归选取剩下的NewSize个元素，直到终止条件触发，返回最终的Rest。

测试调用上述示例时，比如得到Seg = [1,2,3]，对应的R会是[4,5,6,7,8,9]，符合期望的输出。

内容的提问来源于stack exchange，提问作者snow_white