专栏持续更新50道算法题,都是大厂高频算法题,建议关注, 一起巧‘背’算法!
文章目录
- 题目
- 解法
- 总结
题目
解法
- 定义一个节点pre,用于初始化结果链表的头部,cur指向pre,它将在遍历过程中用于构建新的链表。
- 初始化进位变量carry为0,用于记录每次相加后可能的进位。
- 使用一个while循环,当至少有一个链表不为空时继续执行。在循环中:
- 获取当前链表节点的值,如果链表为空,则赋值为0。
- 计算两个节点值与carry的和sum。
- 更新carry,将sum除以10的结果作为新的进位。
- 将sum对10取模的结果作为新节点的值,创建新节点并连接到cur的下一个节点。
- 更新cur指针,使其指向新创建的节点。
- 如果l1不为空,移动l1到下一个节点;如果l2不为空,移动l2到下一个节点。
- 循环结束后,检查carry是否为1,如果是,表示还有进位,需要添加一个新的节点到cur后面。
- 返回结果链表的头部,即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来准确地获取并返回最终的新链表头节点。