package suanfa.lintcode;

import java.util.Stack;

//鏈表
class ListNode {
    int val;
    ListNode next;

    ListNode(int x) {
        val = x;
        next = null;
    }
}

public class SumListNode {

    /**
     * 鏈表求和1：你有兩個(gè)用鏈表代表的整數(shù)，其中每個(gè)節(jié)點(diǎn)包含一個(gè)數(shù)字。數(shù)字存儲按照在原來整數(shù)中相反的順序，
     * 使得第一個(gè)數(shù)字位于鏈表的開頭。寫出一個(gè)函數(shù)將兩個(gè)整數(shù)相加，用鏈表形式返回和。
     * 
     * @param l1
     *            example：3->1->5->null
     * @param l2
     *            example：5->9->2->null
     * @return    
     *            example：8->0->8->null
     */
    public ListNode addListNode1(ListNode l1, ListNode l2) {
        ListNode result = new ListNode(0);
        ListNode node = result;
        // 進(jìn)位標(biāo)志
        int flag = 0;
        while (l1 != null && l2 != null) {
            int value = l1.val + l2.val + flag;
            flag = value / 10;
            value = value % 10;
            node.next = new ListNode(value);
            node = node.next;
            l1 = l1.next;
            l2 = l2.next;
        }
        while (l1 != null) {
            int value = l1.val + flag;
            flag = value / 10;
            value = value % 10;
            node.next = new ListNode(value);
            node = node.next;
            l1 = l1.next;
        }
        while (l2 != null) {
            int value = l2.val + flag;
            flag = value / 10;
            value = value % 10;
            node.next = new ListNode(value);
            node = node.next;
            l2 = l2.next;
        }
        if (flag == 1) {
            node.next = new ListNode(1);
            node = node.next;
        }
        return result.next;
    }

    /**
     * 鏈表求和2： 假定用一個(gè)鏈表表示兩個(gè)數(shù)，其中每個(gè)節(jié)點(diǎn)僅包含一個(gè)數(shù)字。
     * 假設(shè)這兩個(gè)數(shù)的數(shù)字順序排列，請?jiān)O(shè)計(jì)一種方法將兩個(gè)數(shù)相加，并將其結(jié)果表現(xiàn)為鏈表的形式
     * 
     * @param l1
     *            example：6->1->7
     * @param l2
     *            example：2->9->5
     * @return    
     *            example：9->1->2
     */
    public ListNode addListNode2(ListNode l1, ListNode l2) {
        Stack<Integer> stack1 = getListNodeStack(l1);
        Stack<Integer> stack2 = getListNodeStack(l2);
        ListNode result = new ListNode(0);
        int flag = 0; // 進(jìn)位標(biāo)志

        // 對于兩個(gè)鏈表，長度相等時(shí)計(jì)算
        while (!stack1.isEmpty() && !stack2.isEmpty()) {
            int value = stack1.pop() + stack2.pop() + flag;
            flag = value / 10;
            value = value % 10;

            // 創(chuàng)建一個(gè)新的節(jié)點(diǎn)
            ListNode curr = new ListNode(value);
            // 當(dāng)前節(jié)點(diǎn)的next 指向之前的一個(gè)節(jié)點(diǎn)
            curr.next = result.next;
            // 將當(dāng)前節(jié)點(diǎn)存儲起來
            result.next = curr;
        }

        // 當(dāng)?shù)谝粋€(gè)鏈表長度大于第二個(gè)鏈表長度時(shí)
        while (!stack1.isEmpty()) {
            int value = stack1.pop() + flag;
            flag = value / 10;
            value = value % 10;

            // 創(chuàng)建一個(gè)新的節(jié)點(diǎn)
            ListNode curr = new ListNode(value);
            // 當(dāng)前節(jié)點(diǎn)的next 指向之前的一個(gè)節(jié)點(diǎn)
            curr.next = result.next;
            // 將當(dāng)前節(jié)點(diǎn)存儲起來
            result.next = curr;
        }

        // 當(dāng)?shù)诙€(gè)鏈表長度大于第一個(gè)鏈表長度時(shí)
        while (!stack2.isEmpty()) {
            int value = stack2.pop() + flag;
            flag = value / 10;
            value = value % 10;

            // 創(chuàng)建一個(gè)新的節(jié)點(diǎn)
            ListNode curr = new ListNode(value);
            // 當(dāng)前節(jié)點(diǎn)的next 指向之前的一個(gè)節(jié)點(diǎn)
            curr.next = result.next;
            // 將當(dāng)前節(jié)點(diǎn)存儲起來
            result.next = curr;
        }

        // 如果flag等于1，則需要?jiǎng)?chuàng)建一個(gè)新的節(jié)點(diǎn)
        if (flag == 1) {
            ListNode curr = new ListNode(1);
            curr.next = result.next;
            result.next = curr;
        }

        return result.next;
    }

    /**
     * 將鏈表存儲到棧里面
     * 
     * @param node
     * @return Stack
     */
    public Stack<Integer> getListNodeStack(ListNode node) {
        Stack<Integer> stack = new Stack<>();
        while (node != null) {
            stack.push(node.val);
            node = node.next;
        }
        return stack;
    }
}