如何修改Prolog程序以输出指定工作年限员工的ID排序结果？

原程序存在三个核心问题：

1. 排序目标错误：sort/4的第二个参数指定按**LastName（结构的第2个元素）**排序，而非需求的员工ID
2. 返回结构不符合需求：收集并返回的是完整的FirstName-LastName-YearsXP-ID结构，而非仅ID列表
3. 无排序方向切换逻辑：无法支持升序/降序的切换

修改后的代码方案 #

方案1：拆分升序/降序独立谓词

employee(a, b, 3, 8).
employee(c, d, 5, 19).
employee(e, f, 3, 34).
employee(g, h, 3, 6).

% 获取指定工作年限的员工ID（未排序）
get_employee_ids(YearsXP, IDs) :-
    findall(ID, employee(_, _, YearsXP, ID), IDs).

% 升序排序指定年限的员工ID
sort_employee_ids_asc(YearsXP, SortedIDs) :-
    get_employee_ids(YearsXP, IDs),
    sort(IDs, SortedIDs). % Prolog内置sort默认升序

% 降序排序指定年限的员工ID
sort_employee_ids_desc(YearsXP, SortedIDs) :-
    get_employee_ids(YearsXP, IDs),
    sort(IDs, AscSorted),
    reverse(AscSorted, SortedIDs).

方案2：单谓词支持排序方向参数

employee(a, b, 3, 8).
employee(c, d, 5, 19).
employee(e, f, 3, 34).
employee(g, h, 3, 6).

% 支持升序(asc)/降序(desc)的员工ID排序
sort_employee_ids(YearsXP, Direction, SortedIDs) :-
    findall(ID, employee(_, _, YearsXP, ID), IDs),
    (   Direction = asc -> sort(IDs, SortedIDs)
    ;   Direction = desc -> sort(IDs, AscSorted), reverse(AscSorted, SortedIDs)
    ).

使用示例 #

• 升序查询：sort_employee_ids_asc(3, Result). 或 sort_employee_ids(3, asc, Result).，结果为 Result = [6, 8, 34]
• 降序查询：sort_employee_ids_desc(3, Result). 或 sort_employee_ids(3, desc, Result).，结果为 Result = [34, 8, 6]

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