9.链式A+B

题目描述

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

给定两个链表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) {

            int len1=0,len2=0;

         ListNode p=a,q=b,c=null;

          while(p!=null){

              len1++;

              p=p.next;

           }

          while(q!=null){

              len2++;

              q=q.next;

           }

         p=a.next;q=b.next;

          int value=(a.val+b.val)%10;

          int carry=(a.val+b.val)/10;

          if(len1>=len2){

              c=a;

          }else{

              c=b;

          }

          c.val=value;

          ListNode tmp=c.next;

          ListNode flag=c;

          while(p!=null && q!=null){

        	  int m=(p.val+q.val+carry)/10;

        	  int n=(p.val+q.val+carry)%10;

              carry=m;

              tmp.val=n;

              if(tmp.next==null) flag=tmp;

              tmp=tmp.next;

              p=p.next;

              q=q.next;

          }

         if(p!=null){

             while(p!=null){

                  tmp.val=(p.val+carry)%10;

                  carry=(p.val+carry)/10;

                  if(tmp.next==null) flag=tmp;

                  tmp=tmp.next;

                  p=p.next;

              }

         }else if(q!=null){

             while(q!=null){

                  tmp.val=(q.val+carry)%10;

                  carry=(q.val+carry)/10;

                  if(tmp.next==null) flag=tmp;

                  tmp=tmp.next;

                  q=q.next;

              }

         }

          if (carry==1){

                 ListNode newE=new ListNode(1);

                 newE.next=null;

                 flag.next=newE;

             }

       return c;

    }

}