Gidhub BE Developer

LeetCode : 2. Add Two Numbers

2021-02-16
goodGid
LeetCode

2. Add Two Numbers

Problem

You are given two non-empty linked lists representing two non-negative integers. The digits are stored in reverse order, and each of their nodes contains a single digit. Add the two numbers and return the sum as a linked list.
You may assume the two numbers do not contain any leading zero, except the number 0 itself.

Example

Input: l1 = [2,4,3], l2 = [5,6,4]
Output: [7,0,8]
Explanation: 342 + 465 = 807.

[1] Code (21. 02. 16)

class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {

        ListNode head = new ListNode(-1);
        ListNode ans = head;
        int isOver = 0;

        while (l1 != null || l2 != null) {
            int sum = isOver;

            if (l1 != null) {
                sum += l1.val;
                l1 = l1.next;
            }

            if (l2 != null) {
                sum += l2.val;
                l2 = l2.next;
            }

            if (sum >= 10) {
                isOver = 1;
            } else {
                isOver = 0;
            }

            ListNode node = new ListNode(sum % 10);
            head.next = node;
            head = head.next;
        }

        if (isOver == 1) {
            ListNode node = new ListNode(1);
            head.next = node;
        }
        return ans.next;
    }
}

Check Point

  • 단순 구현 문제

[2] Code (21. 03. 21)

class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {

        ListNode dummyNode = new ListNode();
        ListNode head = dummyNode;

        int sum;
        int carry = 0;
        while (l1 != null && l2 != null) {
            sum = carry + l1.val + l2.val;
            carry = sum >= 10 ? 1 : 0;

            ListNode node = new ListNode(sum % 10);
            head.next = node;
            head = head.next;

            l1 = l1.next;
            l2 = l2.next;
        }

        while (l1 != null) {
            sum = carry + l1.val;
            carry = sum >= 10 ? 1 : 0;

            ListNode node = new ListNode(sum % 10);
            head.next = node;
            head = head.next;

            l1 = l1.next;
        }

        while (l2 != null) {
            sum = carry + l2.val;
            carry = sum >= 10 ? 1 : 0;

            ListNode node = new ListNode(sum % 10);
            head.next = node;
            head = head.next;

            l2 = l2.next;
        }

        if (carry == 1) {
            head.next = new ListNode(1);
        }

        return dummyNode.next;
    }
}

Check Point

  • 단순 구현문제

Review

  • IDE를 사용하지 않고 풀었다.

  • while (l1 != null && l2 != null) 이렇게 조건을 주니

    while (l1 != null)과 while (l2 != null)과 같은 코드가 들어가야 했다.

    그래서 이 부분을 어떻게 풀어냈을까? 를 다른 코드를 보고 참고해봤다.

class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {

        ...
        
        ListNode first = l1;
        ListNode second = l2;

        ListNode result = new ListNode(-1);
        int carry = 0;
        ListNode head = result;

        while (first != null || second != null) {
            int x = first != null ? first.val : 0; // null 이면 0으로 처리
            int y = second != null ? second.val : 0; // null 이면 0으로 처리
            int z = x + y + carry;
            result.next = new ListNode(z % 10);
            carry = z / 10;
            result = result.next;

            if (first != null) { // null이 아닐 경우에만 다음 Node로 이동
                first = first.next;
            }
            if (second != null) { // null이 아닐 경우에만 다음 Node로 이동
                second = second.next;
            }

        }

        if (carry > 0) {
            result.next = new ListNode(carry);
        }
        return head.next;
    }
}
  • l1 혹은 l2가 null이라면 그냥 0으로 값을 처리한다.

[3] Code (22. 02. 14) (x)

// Runtime: 2 ms
// Memory Usage: 47.3 MB
// Ref : https://leetcode.com/submissions/detail/641226973/
class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        return recur(l1, l2, 0);
    }

    private ListNode recur(ListNode l1, ListNode l2, int prevSum) {
        if (l1 == null && l2 == null) {
            if (prevSum != 0) {
                return new ListNode(prevSum);
            }
            return null;
        } else if (l1 == null) {
            if (prevSum != 0) {
                l1 = new ListNode(prevSum); // [1]
                prevSum = 0;
            } else {
                return l2;
            }
        } else if (l2 == null) {
            if (prevSum != 0) {
                l2 = new ListNode(prevSum); // [2]
                prevSum = 0;
            } else {
                return l1;
            }
        }

        int sum = l1.val + l2.val + prevSum;
        l1.val = sum % 10;
        l1.next = recur(l1.next, l2.next, sum / 10);

        return l1;
    }
}

Algorithm Description

  • [1], [2] : 한쪽만 남아있고 prevSum이 있다면

    남아있는 쪽에 새로운 Node를 prevSum으로 생성해준다.

    그렇게 재귀 함수를 호출하면 원래 있던 값처럼 동작시킬 수 있다.

Reference

Recommend

Index