求助SWI-Prolog实现：在原列表每个X后插入指定子列表

需求说明 #

• 遍历原列表，每当遇到指定元素X时，立即在X之后插入给定的子列表
• 示例：原列表[1,2,1,5,3]，X=1，指定子列表[0,7,9]，处理后结果为[1,0,7,9,2,1,0,7,9,5,3]

原代码问题分析 #

你提供的代码存在多处逻辑错误：

insert([], _, [], []) :- !.
insert([H|T], X, Sub, [Res]) :- 
   H =\= X, !.
insert([X|T], X, Sub, [Res]) :- 
   insert(T, X, Sub, [X|Sub]).

• 空列表处理规则参数冗余，第三个参数无需限定为空
• 第二个规则仅判断当前元素不等于X，但未将该元素加入结果，也未递归处理剩余列表
• 第三个规则递归逻辑错误，未正确拼接X、子列表和剩余列表的处理结果

正确实现代码 #

以下是符合需求的Prolog实现：

% 基准情况：空列表处理后仍为空
insert([], _, _, []) :- !.

% 匹配到目标元素X时，拼接X、子列表及剩余列表的处理结果
insert([X|T], X, Sub, Result) :-
    insert(T, X, Sub, Rest),
    append([X|Sub], Rest, Result), !.

% 未匹配到X时，保留当前元素，递归处理剩余列表
insert([H|T], X, Sub, [H|Rest]) :-
    H =\= X,
    insert(T, X, Sub, Rest).

代码逻辑说明 #

1. 基准规则：当原列表为空时，结果也为空，使用!避免回溯
2. 匹配X的情况：先递归处理剩余列表得到Rest，再通过append将[X|Sub]（X加子列表）与Rest拼接成最终结果
3. 不匹配X的情况：将当前元素H放在结果列表的头部，再递归处理剩余列表得到Rest

测试示例查询：

insert([1,2,1,5,3], 1, [0,7,9], R).

返回结果：R = [1, 0, 7, 9, 2, 1, 0, 7, 9, 5, 3]，完全符合需求。

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