본문 바로가기
Problem solving/LeetCode

[LeetCode] 2. Add Two Numbers (C++)

by 겸 2023. 6. 16.

문제

각 자릿수의 역순으로 이루어진 링크드리스트 두개가 있다.

이를 더한 값을 링크드리스트로 나타내라

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */

생각할 것

  1. int나 long으로 리스트를 더해서 나타낼까 생각했지만, 문제의 조건에서 연결된 노드의 수가 최대 100개이다.
  2. carry를 두고 더해야한다.
  3. carry도 while문에 포함시켜서 모두 null일 때 끝까지 실행되도록 하기
  4. tmp 노드를 두고 next로 이동해야 한다.
    1. 만약에 return res; 를 한다면 0 7 0 8이 결괏값으로 출력됨
    2. tmp가 current를 의미함!

결과 코드

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* res = new ListNode();
        ListNode* tmp = res;
        int carry = 0;
        while(l1 != NULL || l2 != NULL || carry){
            int sum = 0;
            if(l1 != NULL) {
                sum += l1 -> val;
                l1 = l1 -> next;
            }

            if(l2 != NULL) {
                sum += l2 -> val;
                l2 = l2 -> next;      
            }  

            sum += carry;
            carry = sum/10;
            tmp -> next = new ListNode(sum%10);  
            tmp = tmp -> next;
        }    
        
        return res -> next;
    }
};
반응형