专栏持续更新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来准确地获取并返回最终的新链表头节点。