程序员面试经典(链表) --- 链表A + B

题目描述

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

给定两个链表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;
    }
}