Prolog以递归方式计算列表中的数字

时间:2017-11-13 08:28:50

标签: list recursion prolog tail-recursion

我需要一个程序来计算列表中的所有数字,无论它们是多么深刻。我能够在不在另一个列表中的情况下对数字进行计数,但是通过深层嵌套元素进行递归并不成功。到目前为止我有这个:

count([],0).
count([H|Tail], N) :-
    count(Tail, N1),
    (  number(H)
    ->N is N1 + 1
    ;   is_list(H)
    -> count(H,N)  
    ;   N = N1
    ).

所以,如果我打电话给count([a,1,[2,b],3],N),输出应为N=3;但是,我只得到N=2。有人可以帮我加我的第二次案例测试吗?此处所有可用的解决方案都不适用于深层嵌套的数字元素。

谢谢!

1 个答案:

答案 0 :(得分:3)

N1分支的代码不正确:在这种情况下,您忽略N的值,这是不正确的,您希望N1为{{1}的总和在H上计算。

完整代码:

:- use_module(library(clpfd)).

count([], 0).
count([H|T], N) :-
    count(T, N1),
    (   number(H) ->
        N #= N1 + 1
    ;   is_list(H) ->
        N #= N1 + N2,
        count(H, N2)
    ;   N1 = N
    ).