大厂常见算法50题-两数相加

专栏持续更新50道算法题，都是大厂高频算法题，建议关注, 一起巧‘背’算法!

题目

解法

1. 定义一个节点pre，用于初始化结果链表的头部，cur指向pre，它将在遍历过程中用于构建新的链表。
2. 初始化进位变量carry为0，用于记录每次相加后可能的进位。
3. 使用一个while循环，当至少有一个链表不为空时继续执行。在循环中：
   1. 获取当前链表节点的值，如果链表为空，则赋值为0。
   2. 计算两个节点值与carry的和sum。
   3. 更新carry，将sum除以10的结果作为新的进位。
   4. 将sum对10取模的结果作为新节点的值，创建新节点并连接到cur的下一个节点。
   5. 更新cur指针，使其指向新创建的节点。
   6. 如果l1不为空，移动l1到下一个节点；如果l2不为空，移动l2到下一个节点。
4. 循环结束后，检查carry是否为1，如果是，表示还有进位，需要添加一个新的节点到cur后面。
5. 返回结果链表的头部，即pre.next。

public class AddTwoNumbers {

    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode pre = new ListNode(0);
        ListNode cur = pre;
        int carry = 0;
        while (l1 != null || l2!=null){
            // 取出当前节点的值
            int x = l1 == null ? 0 : l1.val;
            int y = l2 == null ? 0 : l2.val;

            // 算出两数相加的值
            int sum = x + y + carry;

            // 算出需要进位的值
            carry = sum / 10;

            // 算出下一个节点的值
            sum = sum%10;

            // 移动节点
            cur.next = new ListNode(sum);
            cur = cur.next;
            if (l1 != null){
                l1 = l1.next;
            }
            if (l2 != null){
                l2 = l2.next;
            }
        }

        // 如果carry最后为1，表示还要进位，加一个节点
        if (carry == 1){
            cur.next = new ListNode(carry);
        }

        return pre.next;
    }

}

class ListNode {
  int val;
  ListNode next;
  ListNode() {}
  ListNode(int val) {
      this.val = val;
  }
  ListNode(int val, ListNode next) {
      this.val = val;
      this.next = next;
  }
}

总结

在处理链表问题并需要返回新链表的头节点时，通常会先创建一个pre指针。这个pre指针的作用是作为辅助，它的下一个节点是指向实际新链表的头节点head。这是因为，在构建链表过程中，为了防止原始头节点在不断创建新节点和移动指针时被覆盖或丢失，我们使用pre来保持对新链表起点的引用。这样，在链表构建完毕后，可以通过pre.next来准确地获取并返回最终的新链表头节点。