Prolog中生成仅含红蓝两状态元素的所有排列的实现方法

我懂你想实现的目标——生成所有由红色和蓝色两种元素组成的、长度为n的列表排列对吧？结合你已经写好的基础代码，我们只需要调整一下生成逻辑就能搞定啦！

先回顾下你现有的代码：

% 生成全蓝色列表的谓词
generate_list(0, []) :- !.
generate_list(N, [H | T]) :- 
    N2 is N-1, 
    H = blue, 
    generate_list(N2, T).

% 颜色翻转谓词
flip(blue, red).
flip(red, blue).

你原来的generate_list只能生成全蓝的列表，现在要让每个位置的元素都有两种可能（蓝色或者翻转后的红色），就能覆盖所有排列组合了。

修改后的实现方案 #

% 生成所有长度为N的红蓝元素列表
generate_all_lists(0, []) :- !.
generate_all_lists(N, [H | T]) :-
    N2 is N-1,
    % 每个位置的元素有两种选择：原颜色blue，或者翻转后的red
    (H = blue ; flip(blue, H)),
    generate_all_lists(N2, T).

代码逻辑说明 #

• 递归终止条件和你原来的一致：当长度N为0时，返回空列表，用!避免回溯到其他分支。
• 对于每个非终止的递归步骤，我们给当前元素H两种选择：
  1. 直接赋值为blue（和你原来的逻辑保持一致）
  2. 通过flip(blue, H)得到red（复用你已经定义好的翻转谓词）
• 递归生成剩余长度为N-1的子列表，这样Prolog会自动回溯遍历所有可能的组合，生成全部排列。

测试示例 #

比如查询长度为2的所有红蓝列表，执行以下查询：

?- generate_all_lists(2, List).
List = [blue, blue] ;
List = [blue, red] ;
List = [red, blue] ;
List = [red, red].

如果想要更通用的版本，比如支持从任意初始颜色生成所有排列，可以给谓词加个初始颜色参数：

% 从指定初始颜色生成所有长度为N的双色列表
generate_all_lists_from(0, _, []) :- !.
generate_all_lists_from(N, StartColor, [H | T]) :-
    N2 is N-1,
    (H = StartColor ; flip(StartColor, H)),
    generate_all_lists_from(N2, StartColor, T).

比如查询从红色开始的长度为3的列表：generate_all_lists_from(3, red, List)，就能得到所有由红、蓝组成的3元素列表啦。

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