CODEWITHSUNDEEP
listprologhead
Arnau Van Boschken ArnauB
Arnau Van Boschken ArnauB

Reputation: 121

Prolog list of lists get all elements

I have a lists of lists:

decide([[1,2,-3],[-2,3],[6],[4]],K).

I want to return all the possible solutions pressing ';'.

The rule is to first return the values that its list has size 1.

Then I want to return the values that its size is bigger than 1.

size([],0).
size([_|Xs],L) :- size(Xs,N),L is N+1.

head([],[]).
head([X|_],X).

return_list_members([X|_], X).
return_list_members([_|T], X):-return_list_members(T, X).


decide([], []).
decide([L|Ls], Lit):- size(L, N), N == 1, head(L, Lit).
decide([L|Ls], Lit):- size(Ls, N), N == 0, head(L, Lit), !.
decide([L|Ls], Lit):- decide(Ls, Lit) ,return_list_members(Ls, Lit)

Example how should be the resulr:

? - decide([[1,2,-3],[-2,3],[6],[4]],K).
K = 6 ;
K = 4 ;
K = -2 ;
K = 3 ;
K = -3 ;
K = 2 ;
K = 1.

My goal is to return first the list with only one value. Then return all the elem of the others lists, one by one. The form I have the code, only return the first elem of the list, because I have the head call. How I can return not only the head values, but all the others, and without repetead? I tried to creat a function for return all the elem of the lists.

Any suggestion?

Upvotes: 0

Views: 396

Answers (1)

false
false

Reputation: 10102

Taking your other question as a starting point, simply insert your new requirements:

listoflist_member(Xss, X) :-
   ( Xs = [_] ; Xs = [_,_|_] ),           % new
   member(Xs, Xss),
   member(X, Xs).

Upvotes: 4

Related Questions