题目描述
有两个用链表表示的整数,每个结点包含一个数位。这些数位是反向存放的,也就是个位排在链表的首部。编写函数对这两个整数求和,并用链表形式返回结果。
给定两个链表ListNode* A,ListNode* B,请返回A+B的结果(ListNode*)。
测试样例:
{1,2,3},{3,2,1}
返回:{4,4,4}
思路:让A,B链表的每一位分开进行相加后存入一个新的链表,迭代后得到最终的结果。
注意:使用两个新的链表进行操作,其中链表start表示最终要返回的结果链表,链表current表示正在移动相加的链表(一开始的赋值是地址上的,所以current的变化也会引起start的变化)
import java.util.*;
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Plus {
public ListNode plusAB(ListNode a, ListNode b) {
if(a == null && b == null) return null;
ListNode start = null; //最终返回的链表
ListNode current = null; //添加时的链表
int tmp = 0; //用来存储相加的值
int plus = 0; //用来查看是否需要进位
while(a!=null || b!=null) {
tmp = plus; //若有仅为一开始为1
if(a!=null)
tmp += a.val;
if(b!=null)
tmp += b.val;
if(start == null) {
current = new ListNode(tmp%10); //相加后可能存在进位
start = current;
plus = tmp/10;
}else {
current.next = new ListNode(tmp%10);
current = current.next;
plus = tmp/10;
}
if(a==null && b!=null)
b = b.next;
else if(a != null && b == null)
a = a.next;
else {
a = a.next;
b = b.next;
}
}
//若存在进位,则要进行构建新链表
if(plus == 1)
current.next = new ListNode(1);
return start;
}
}