两个单链表生成相加链表

package 链式APlusB;

class ListNode {
    int val;
    ListNode next = null;

    ListNode(int val) {
        this.val = val;
    }
}

public class Plus {

    public static void main(String[] args) {
        ListNode one = new ListNode(9);
        ListNode two = new ListNode(3);
        ListNode three = new ListNode(7);
        ListNode four = new ListNode(6);
        ListNode five = new ListNode(3);
        one.next = two;
        two.next = three;
        four.next = five;
        //print(new Plus().reverse(one));
        new Plus().plusAB(one, four);
    }

    private static void print(ListNode root) {
        if (root == null)
            return;
        System.out.print(root.val + " ");
        print(root.next);
    }

    public ListNode plusAB(ListNode a, ListNode b) {
        // write code here
        a = reverse(a);
        b = reverse(b);
        ListNode ans = null;
        ListNode aHead = a;
        ListNode bHead = b;
        int add = 0;
        while (a != null || b != null) {
            int val = 0;
            val += add;
            if (a != null) {
                val += a.val;
                a = a.next;
            }
            if (b != null) {
                val += b.val;
                b = b.next;
            }
            if (val > 9) {
                add = val / 10;
                val %= 10;

                if (ans == null) {
                    ans = new ListNode(val);
                } else {
                    ListNode tmp = ans;
                    ans = new ListNode(val);
                    ans.next = tmp;
                }
            } else {
                add = 0;
                if (ans == null) {
                    ans = new ListNode(val);
                } else {
                    ListNode tmp = ans;
                    ans = new ListNode(val);
                    ans.next = tmp;
                }
            }
        }
        if(add != 0){
            ListNode tmp = ans;
            ans = new ListNode(add);
            ans.next = tmp;
        }
        aHead = reverse(aHead);
        bHead = reverse(bHead);
        return ans;
    }

    private ListNode reverse(ListNode a) {
        ListNode pre = null;
        ListNode next = null;
        ListNode cur = a;
        while (cur != null) {
            next = cur.next;
            cur.next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值