有两个用链表表示的整数,每个结点包含一个数位。这些数位是反向存放的,也就是个位排在链表的首部。编写函数对这两个整数求和,并用链表形式返回结果。
给定两个链表ListNode* A,ListNode* B,请返回A+B的结果(ListNode*)。
测试样例:
{1,2,3},{3,2,1}
返回:{4,4,4}
package 链表;
import java.awt.List;
public class Plus {
public static void main(String[] args) {
//{7,4,0,7,5},{2,7,2,3,4}
ListNode a1=new ListNode(7);
ListNode a2=new ListNode(4);
ListNode a3=new ListNode(0);
ListNode a4=new ListNode(7);
ListNode a5=new ListNode(5);
a1.next=a2;
a2.next=a3;
a3.next=a4;
a4.next=a5;
ListNode b1=new ListNode(2);
ListNode b2=new ListNode(7);
ListNode b3=new ListNode(2);
ListNode b4=new ListNode(3);
ListNode b5=new ListNode(4);
b1.next=b2;
b2.next=b3;
b3.next=b4;
b4.next=b5;
plusAB(a1, b1);
}
public static ListNode plusAB(ListNode a, ListNode b) {
ListNode head=null;
// write code here
//{1,9,9,9},{9,2,}
if(b==null){
return a;
}
if(a==null){
return b;
}
boolean flag=false;
//{7,4,0,7,5},{2,7,2,3,4}
while(a!=null&&b!=null){
head=new ListNode(0);
if(a.val+b.val-10>=0){
head.val=a.val+b.val-10;
flag=true;
}else{
head.val=a.val+b.val;
flag=false;
}
a=a.next;
b=b.next;
break;
}
ListNode last=head;
//{7,4,0,7,5},{2,7,2,3,4} {9,1,3,0,0,1} {9,1,3,0,-1,1}
while(a!=null&&b!=null){
ListNode bian=new ListNode(0);
last.next=bian;
last=bian;
if(flag){
last.val=a.val+b.val+1;
if(last.val-10>=0){
last.val=last.val-10;
flag=true;
}else{
flag=false;
}
}else{
last.val=a.val+b.val;
if(last.val-10>=0){
last.val=a.val+b.val-10;
flag=true;
}else{
last.val=a.val+b.val;
flag=false;
}
}
a=a.next;
b=b.next;
}
if(a==null&&b==null){
if(flag){
last.next=new ListNode(1);
}
return head;
}
if(b==null&&a!=null){
while(a!=null){
last.next=a;
last=a;
if(flag){
last.val=last.val+1;
if(last.val-10>=0){
last.val=last.val-10;
flag=true;
}else{
last.val=last.val;
flag=false;
}
}
a=a.next;
}
if(flag){
last.next=new ListNode(1);
}
return head;
}
if(a==null&&b!=null){
while(b!=null){
last.next=b;
last=b;
if(flag){
last.val=last.val+1;
if(last.val-10>=0){
last.val=last.val-10;
flag=true;
}else{
last.val=last.val;
flag=false;
}
}
b=b.next;
}
if(flag){
last.next=new ListNode(1);
}
return head;
}
return head;
}
}
public static void main(String[] args) {
//{7,4,0,7,5},{2,7,2,3,4}
ListNode a1=new ListNode(7);
ListNode a2=new ListNode(4);
ListNode a3=new ListNode(0);
ListNode a4=new ListNode(7);
ListNode a5=new ListNode(5);
a1.next=a2;
a2.next=a3;
a3.next=a4;
a4.next=a5;
ListNode b1=new ListNode(2);
ListNode b2=new ListNode(7);
ListNode b3=new ListNode(2);
ListNode b4=new ListNode(3);
ListNode b5=new ListNode(4);
b1.next=b2;
b2.next=b3;
b3.next=b4;
b4.next=b5;
plusAB(a1, b1);
}
public static ListNode plusAB(ListNode a, ListNode b) {
ListNode head=null;
// write code here
//{1,9,9,9},{9,2,}
if(b==null){
return a;
}
if(a==null){
return b;
}
boolean flag=false;
//{7,4,0,7,5},{2,7,2,3,4}
while(a!=null&&b!=null){
head=new ListNode(0);
if(a.val+b.val-10>=0){
head.val=a.val+b.val-10;
flag=true;
}else{
head.val=a.val+b.val;
flag=false;
}
a=a.next;
b=b.next;
break;
}
ListNode last=head;
//{7,4,0,7,5},{2,7,2,3,4} {9,1,3,0,0,1} {9,1,3,0,-1,1}
while(a!=null&&b!=null){
ListNode bian=new ListNode(0);
last.next=bian;
last=bian;
if(flag){
last.val=a.val+b.val+1;
if(last.val-10>=0){
last.val=last.val-10;
flag=true;
}else{
flag=false;
}
}else{
last.val=a.val+b.val;
if(last.val-10>=0){
last.val=a.val+b.val-10;
flag=true;
}else{
last.val=a.val+b.val;
flag=false;
}
}
a=a.next;
b=b.next;
}
if(a==null&&b==null){
if(flag){
last.next=new ListNode(1);
}
return head;
}
if(b==null&&a!=null){
while(a!=null){
last.next=a;
last=a;
if(flag){
last.val=last.val+1;
if(last.val-10>=0){
last.val=last.val-10;
flag=true;
}else{
last.val=last.val;
flag=false;
}
}
a=a.next;
}
if(flag){
last.next=new ListNode(1);
}
return head;
}
if(a==null&&b!=null){
while(b!=null){
last.next=b;
last=b;
if(flag){
last.val=last.val+1;
if(last.val-10>=0){
last.val=last.val-10;
flag=true;
}else{
last.val=last.val;
flag=false;
}
}
b=b.next;
}
if(flag){
last.next=new ListNode(1);
}
return head;
}
return head;
}
}