牛客网-两个链表生成相加链表

题目描述：

假设链表中每一个节点的值都在 0 - 9 之间，那么链表整体就可以代表一个整数。

给定两个这种链表，请生成代表两个整数相加值的结果链表。

例如：链表 1 为 9->3->7，链表 2 为 6->3，最后生成新的结果链表为 1->0->0->0。

这道题没啥技巧，思路清晰，就是代码有点多。

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param head1 ListNode类 
     * @param head2 ListNode类 
     * @return ListNode类
     */
    public ListNode addInList (ListNode head1, ListNode head2) {
        // write code here
        //首先将两个链表都进行反转
        ListNode h1 = reverse(head1);
        ListNode h2 = reverse(head2);
        int flag = 0; //表示相加过程中的进位
        ListNode result = null;  //表示结果链表
        //使用尾插法将结果插入结果链表中
        ListNode rear = result;
        //反转完成之后再进行相加操作
        while(h1!=null && h2!=null){
            if(h1.val+h2.val+flag>=10){
                //表示有进位
                if(rear==null){
                    //表示第一个节点
                    result = new ListNode(h1.val+h2.val+flag-10);
                    rear=result;
                } else{
                    //直接尾插法
                    ListNode temp = new ListNode(h1.val+h2.val+flag-10);
                    rear.next=temp;
                    rear=temp;
                }
                flag = 1;
            } else{
                //表示没有进位
                if(rear==null){
                    //表示第一个节点
                    result = new ListNode(h1.val+h2.val+flag);
                    rear=result;
                } else{
                    //直接尾插法
                    ListNode temp = new ListNode(h1.val+h2.val+flag);
                    rear.next=temp;
                    rear=temp;
                }
                flag=0;
            }
            h1=h1.next;
            h2=h2.next;
        }
        if(h1!=null){
            //表示链表h1中还有剩余节点
            while(h1!=null){
                if(h1.val+flag>=10){
                    //表示有进位
                    ListNode temp = new ListNode(h1.val+flag-10);
                    rear.next=temp;
                    rear=temp;
                    flag=1;
                } else{
                    //表示没有进位
                    ListNode temp = new ListNode(h1.val+flag);
                    rear.next=temp;
                    rear=temp;
                    flag=0;
                }
                h1=h1.next;
            }
        }
        if(h2!=null){
            //表示链表h2中还有剩余节点
            while(h2!=null){
                if(h2.val+flag>=10){
                    //表示有进位
                    ListNode temp = new ListNode(h2.val+flag-10);
                    rear.next=temp;
                    rear=temp;
                    flag=1;
                } else{
                    //表示没有进位
                    ListNode temp = new ListNode(h2.val+flag);
                    rear.next=temp;
                    rear=temp;
                    flag=0;
                }
                h2=h2.next;
            }
        }
        if(flag==1){
            ListNode temp = new ListNode(1);
            rear.next=temp;
            rear=temp;
        }
        //最后需要将result链表也进行反转
        return reverse(result);
    }
    public ListNode reverse(ListNode head){
        ListNode p = head;
        ListNode q = p.next;
        p.next=null;
        ListNode temp=null;
        //将q后面的节点使用头插法插入p中
        while(q!=null){
            //首先保存q的next
            temp=q.next;
            q.next = p;
            p=q;
            q=temp;
        }
        return p;
    }
}