程序员面试金典——链式A+B

题目描述

有两个用链表表示的整数，每个结点包含一个数位。这些数位是反向存放的，也就是个位排在链表的首部。编写函数对这两个整数求和，并用链表形式返回结果。

给定两个链表ListNode* A，ListNode* B，请返回A+B的结果(ListNode*)。

测试样例：

{1,2,3},{3,2,1}

返回：{4,4,4}

依次从a，b处开始累加，每次加的结果有进位放到index里，然后将除开高位剩余的结果保存到a中，用pre节点跟踪a（主要防止a比b短，后面a为null），后面可以直接将两串长度不同时，另外的链表直接放过来。ab一样长时，要判断有无进位，有进位要新造一个高位节点。

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
      	int index=0;//进位
		ListNode head = a;
		ListNode pre = a;
		while(a!=null&&b!=null){
			int result = a.val+b.val+index;
			index = result/10;
			result =result%10;
			a.val = result;
			a=a.next;
			b=b.next;
			if(pre.next!=null)pre=pre.next;
		}//while
		if(a==null&&b!=null){
			b.val+=index;
			pre.next=b;
		}
		if(b==null&&a!=null){
			a.val+=index;
		}
		if(a==null&&b==null){
			if(index>0){
				ListNode last = new ListNode(index);
				pre.next = last;
			}
		}
		return head;
    }
}