Prolog连续整数列表压缩问题求助：输出格式不符合预期

需求与示例 #

需求：将包含连续整数段的列表压缩，规则如下：

• 长度大于2的连续整数段，编码为[起始值, 结束值]（结束值包含在内）
• 长度不大于2的元素/段，保持原样

示例输入：

compress([3,4,6,7,8,9,10,15,16,17,20,22,23,24,25], CompressedList).

预期输出：

CompressedList = [3, 4, [6,10],[15, 17],20, [22,25]]

当前错误输出：

CompressedList = [[3, [4]], [6,[10]],[15, [17]],20, [22,[25]]]

现有代码 #

compress([], []).

compress([X], [X]).

compress([X, Y | T], [X | CompressedTail]) :-
    X + 1 =\= Y,
    compress([Y | T], CompressedTail).
    

compress([X, Y | T], [[X, YLast] | CompressedTail]) :-
    compress_consecutive([Y | T], YLast, Next),
    compress(Next, CompressedTail).


compress_consecutive([], Last, []) :- Last = [].
compress_consecutive([X], [X], []) :- !.
compress_consecutive([X, Y | T], Last, Next) :-
    X + 1 =:= Y,
    compress_consecutive([Y | T], Last, Next).
    

compress_consecutive([X, Y | T], [X], [Y | T]) :- X + 1 =\= Y.

问题分析 #

你的代码存在两个核心问题：

1. compress_consecutive返回的结束值被错误包装为列表（如[X]），导致最终结果出现多层嵌套结构
2. 未判断连续段的长度，直接将所有连续段包装为[X, YLast]，不符合“长度≤2的段保持原样”的要求

修正后的代码 #

% 空列表处理
compress([], []).
% 单个元素处理
compress([X], [X]).

% 情况1：当前两元素不连续，保留第一个元素后继续处理剩余列表
compress([X, Y | T], [X | CompressedTail]) :-
    X + 1 =\= Y,
    compress([Y | T], CompressedTail).

% 情况2：当前两元素连续，先定位连续段的结束值，再根据长度决定编码方式
compress([X, Y | T], Result) :-
    X + 1 =:= Y,
    find_end([Y | T], X, End, Next),
    % 计算连续段长度
    Length is End - X + 1,
    (   Length > 2
    ->  Result = [[X, End] | CompressedTail]
    ;   Result = [X, Y | CompressedTail]  % 长度为2时直接保留两个元素
    ),
    compress(Next, CompressedTail).

% 辅助谓词：找到连续段的结束值和未处理的剩余列表
find_end([Last], _, Last, []).
find_end([Current, Next | T], Start, End, Rest) :-
    Current + 1 =:= Next,
    find_end([Next | T], Start, End, Rest).
find_end([Current, Next | T], _, Current, [Next | T]) :-
    Current + 1 =\= Next.

测试验证 #

运行示例输入后，输出结果与预期完全一致：

CompressedList = [3, 4, [6, 10], [15, 17], 20, [22, 25]]

关键修改说明 #

1. 新增find_end辅助谓词，精准定位连续段的结束值和剩余未处理列表，避免原代码的嵌套问题
2. 在主逻辑中加入连续段长度判断，仅当长度大于2时才使用[起始,结束]格式编码
3. 优化分支逻辑，确保不连续元素和短连续段的处理符合需求

内容的提问来源于stack exchange，提问作者k.s