求两个多项式链表的和

/*
题目编号: 1634
题目描述: 给定两个表示多项式的链表，计算它们的和并返回一个新的链表表示结果。
每个链表节点包含一个系数和一个指数，并且链表按照指数降序排列。

思路:
1. 使用Stream API处理链表。
2. 将两个链表转为Map<指数, 系数>形式。
3. 合并两个Map并处理相同指数的情况。
4. 将结果转换为链表。
*/

import java.util.*;
import java.util.stream.*;

class ListNode {
    int coef;
    int exp;
    ListNode next;
    ListNode(int coef, int exp) {
        this.coef = coef;
        this.exp = exp;
        this.next = null;
    }
}

public class Solution {
    public ListNode addPolynomial(ListNode poly1, ListNode poly2) {
        Map<Integer, Integer> map1 = toMap(poly1);
        Map<Integer, Integer> map2 = toMap(poly2);
        Map<Integer, Integer> resultMap = new HashMap<>();

        map1.forEach((exp, coef) -> resultMap.merge(exp, coef, Integer::sum));
        map2.forEach((exp, coef) -> resultMap.merge(exp, coef, Integer::sum));

        return toListNode(resultMap);
    }

    private Map<Integer, Integer> toMap(ListNode poly) {
        return Stream.iterate(poly, Objects::nonNull, node -> node.next)
                     .collect(Collectors.toMap(node -> node.exp, node -> node.coef));
    }

    private ListNode toListNode(Map<Integer, Integer> map) {
        return map.entrySet().stream()
                  .sorted(Map.Entry.<Integer, Integer>comparingByKey().reversed())
                  .map(entry -> new ListNode(entry.getValue(), entry.getKey()))
                  .reduce(null, (next, node) -> { node.next = next; return node; }, (a, b) -> b);
    }
}