题目描述
有两个用链表表示的整数,每个结点包含一个数位。这些数位是反向存放的,也就是个位排在链表的首部。编写函数对这两个整数求和,并用链表形式返回结果。
给定两个链表ListNode* A,ListNode* B,请返回A+B的结果(ListNode*)。
测试样例:
{1,2,3},{3,2,1}
返回:{4,4,4}
解题思路:
首先判断两种边界情况,即是一个为空时,另一个链表就是输出,然后再判断当两个链表都不为空时,每个节点值相加,然后就是需要判断进位,这里需要注意的是进位值的初始化,每次用完之后立马进行初始化,防止出错。再一个为空或者两个链表都为空的时候,再进行剩下的节点值与进位的加法运算,这里需要注意的是即是所有的节点都加算完成之后,仍需判断是否有进位产生,如果有则需要新建一个节点来保存它,保证数值相加的正确性。
代码如下
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) {
// write code here
if(a == null) {
return b;
}
if(b == null) {
return a;
}
int flag = 0;
int temp = 0;
ListNode node = null;
ListNode pnode = null;
while(a != null && b != null) {
temp = a.val + b.val + flag;
flag = 0;
if(temp >= 10) {
temp %= 10;
flag = 1;
}
ListNode root = new ListNode(temp);
if(node == null) {
node = root;
pnode = root;
}else {
node.next = root;
node = root;
}
a = a.next;
b = b.next;
}
while(a != null) {
temp = a.val + flag;
flag = 0;
if(temp >= 10) {
temp %= 10;
flag = 1;
}
ListNode root = new ListNode(temp);
node.next = root;
node = root;
a = a.next;
}
while(b != null) {
temp = b.val + flag;
flag = 0;
if(temp >= 10) {
temp %= 10;
flag = 1;
}
ListNode root = new ListNode(temp);
node.next = root;
node = root;
b = b.next;
}
if(flag == 1) {
ListNode root = new ListNode(1);
node.next = root;
node = root;
}
return pnode;
}
}