Cut Tree in Prolog from some depth

2019-02-21 05:53发布

first(M,N):-
     N>0,M=..[H|T],((T==[],write(H));(T\=[],write(H),Q is N-1,second(T,Q))).
second(M,N):-
     N>0,M=[H|T],first(H,N),((T==[]);(T\=[],second(T,N))).
second(_,0):-!.

I want to cut tree form some depth.I write some program which give me this result.

?- first(s(f(b(m,k),a),t(a,g)),4).
sfbmkatag
True

But I want to get this.

?- first(s(f(b(m,k),a),t(a,g)),2,X). 
X = s(f,t) 
True

?- first(s(f(b(m,k),a),t(a,g)),3,X). 
X = s(f(b,a),t(a,g)) 
True

Please someone help me to solove it.

0条回答
登录 后发表回答