355,两数相加 II

想了解更多数据结构以及算法题,可以关注微信公众号“数据结构和算法”,每天一题为你精彩解答。也可以扫描下面的二维码关注
在这里插入图片描述

给定两个非空链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储单个数字。将这两数相加会返回一个新的链表。


你可以假设除了数字 0 之外,这两个数字都不会以零开头。


示例:

输入: (7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
输出: 7 -> 8 -> 0 -> 7

结点类如下

public class ListNode {
    int val;
    ListNode next;

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

答案:

public ListNode addTwoNumbers(ListNode list1, ListNode list2) {
    Stack<Integer> s1 = new Stack<>();
    Stack<Integer> s2 = new Stack<>();
    while (list1 != null) {
        s1.push(list1.val);
        list1 = list1.next;
    }
    while (list2 != null) {
        s2.push(list2.val);
        list2 = list2.next;
    }
    int sum = 0;
    ListNode head = new ListNode(0);
    while (!s1.empty() || !s2.empty()) {
        if (!s1.empty())
            sum += s1.pop();
        if (!s2.empty())
            sum += s2.pop();
        head.val = sum % 10;
        ListNode node = new ListNode(sum / 10);
        node.next = head;
        head = node;
        sum /= 10;
    }
    return head.val == 0 ? head.next : head;
}

解析:

思路其实很简单,就是分别把两个链表的结点先存放到两个栈中,因为链表的最高位是在链表的最开始的位置,所以存放到栈中之后,栈底是高位,栈顶是个位(也是低位),然后两个栈中的元素再相加,因为栈是先进后出的,最先出来的肯定是个位(也是低位),最后出来的肯定是高位,也就是这两个数是从个位开始相加,这也符合加法的运算规律。


1,代码中第19行我们只保存相加的个位数,因为链表的每个结点只能保存一位数,如果有进位就会在下一步进行保存。

2,第20行在保留进位的值。其中第20到22行涉及到链表的插入,这个使用的是头插法,在链表节点的头部插入,比较简单,如果看不懂的,还可以看下前面刚讲的352,数据结构-2,链表。


3,代码第25行先判断链表相加之后的最高位是否有进位,如果有就直接返回,如果没有就返回头结点head的下一个结点即可。

代码比较简单,我们就以上面的例子来画一个图加深一下理解,

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

对于加法运算这块,我们还可以再来换一种写法

public ListNode addTwoNumbers(ListNode list1, ListNode list2) {
    Stack<ListNode> stk1 = new Stack();
    Stack<ListNode> stk2 = new Stack();
    while (list1 != null) {
        stk1.push(list1);
        list1 = list1.next;
    }
    while (list2 != null) {
        stk2.push(list2);
        list2 = list2.next;
    }
    int carry = 0;
    ListNode node = null, prev = null;
    while (!stk1.isEmpty() || !stk2.isEmpty()) {
        int no1 = !stk1.isEmpty() ? stk1.pop().val : 0;
        int no2 = !stk2.isEmpty() ? stk2.pop().val : 0;
        node = new ListNode((no1 + no2 + carry) % 10);
        carry = (no1 + no2 + carry) / 10;
        node.next = prev;
        prev = node;
    }
    if (carry != 0) {
        node = new ListNode(carry);
        node.next = prev;
    }
    return node;
}

carry表示的是进位的值,上面两种方式虽然写法上有一点差别,但整体思路还是没变,下面我们再来换种思路,使用递归的方式来解决

public ListNode addTwoNumbers(ListNode list1, ListNode list2) {
    int size1 = getLength(list1);
    int size2 = getLength(list2);
    ListNode head = new ListNode(1);
    head.next = size1 < size2 ? helper(list2, list1, size2 - size1) : helper(list1, list2, size1 - size2);
    if (head.next.val > 9) {
        head.next.val = head.next.val % 10;
        return head;
    }
    return head.next;
}

//这里链表list1的长度是大于等于list2的长度的
public ListNode helper(ListNode list1, ListNode list2, int offset) {
    if (list1 == null)
        return null;
    ListNode result = offset == 0 ? new ListNode(list1.val + list2.val) : new ListNode(list1.val);
    ListNode post = offset == 0 ? helper(list1.next, list2.next, 0) : helper(list1.next, list2, offset - 1);
    if (post != null && post.val > 9) {
        result.val += 1;
        post.val = post.val % 10;
    }
    result.next = post;
    return result;
}

public int getLength(ListNode list) {
    int count = 0;
    while (list != null) {
        list = list.next;
        count++;
    }
    return count;
}

getLength表示的是计算链表的长度,代码很容易理解。这题解法思路奇特的地方在第4行,他先默认两个链表相加,最高位会有进位,然后在第6行进行判断,如果确实有进位,修改一下head.next的值,然后返回head,如果没有进位,直接在第10行返回head.next即可。这段代码的核心是在helper方法中,我们来仔细分析一下

1,list1的长度是大于等于list2的长度的,offset表示的是list1的长度与list2长度的差值,如果list1长度大于list2的长度,那么offset是正数,如果list1长度等于list2的长度,那么offset就是0,offset不可能是负数,因为list1长度不可能小于list2的长度。


2,第17行是创建一个新的结点,如果两个链表长度相等,这个新的结点的值就是这两个链表的头结点相加(注意这里的头结点是一直变的),否则新结点的值就是list1的头结点的值。


3,然后第18行进行这种递归的操作后续结点


4,19行判断后续结点是否有进位,如果有进位再处理进位的问题


5,第23行是链表的连接,连接之后在24行直接返回。


在这里插入图片描述

posted @ 2020-09-25 21:33  数据结构和算法  阅读(101)  评论(0编辑  收藏  举报