Prolog谓词实现：计算列表首元素到指定元素的距离之和

我来帮你搞定这个Prolog谓词的实现，正好匹配你需要的功能！核心思路是遍历列表时同时跟踪当前元素的索引和所有'w'元素索引的累计和，最后对比累计和与输入数字是否相等。

% 主谓词h1：判断输入数字是否等于列表中所有'w'的索引之和
h1(List, Num) :-
    sum_w_indices(List, 0, 0, Num).

% 递归终止条件：空列表时，累计和就是最终结果
sum_w_indices([], _, TotalSum, TotalSum) :- !.

% 遇到'w'时：把当前索引加入累计和，索引+1后继续遍历剩余列表
sum_w_indices([w|Rest], CurrentIndex, CurrentSum, FinalSum) :-
    NewIndex is CurrentIndex + 1,
    NewSum is CurrentSum + CurrentIndex,
    sum_w_indices(Rest, NewIndex, NewSum, FinalSum).

% 遇到非'w'元素时：仅更新索引，累计和不变，继续遍历
sum_w_indices([_|Rest], CurrentIndex, CurrentSum, FinalSum) :-
    NewIndex is CurrentIndex + 1,
    sum_w_indices(Rest, NewIndex, CurrentSum, FinalSum).

测试验证 #

你可以直接用示例测试：

% 符合条件的情况，返回true
?- h1([e,w,b,a,w,w,c], 10).
true.

% 不符合条件的情况，返回false
?- h1([e,w,b,a,w,w,c], 9).
false.

代码逻辑解释 #

• h1/2是对外的主谓词，调用辅助谓词sum_w_indices/4时，初始索引设为0（列表首元素的索引），初始累计和设为0。
• sum_w_indices/4的四个参数分别是：剩余待遍历的列表、当前元素的索引、当前累计的索引和、目标对比数字。
• 第一个子句是递归终止条件：当列表遍历完成（空列表），当前累计和就是所有'w'的索引总和，直接与目标数字统一即可。
• 第二个子句专门处理遇到'w'的场景：计算新的索引（当前索引+1），更新累计和（加上当前'w'的索引），然后递归处理剩余列表。
• 第三个子句处理非'w'的元素：只需要把索引加1，累计和保持不变，继续递归遍历。

如果你想沿用你原来写的h2命名，只需要把主谓词改成h2就行：

h2(List, H) :-
    sum_w_indices(List, 0, 0, H).

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