LeetCode : 2. Add Two Numbers

오우.. 이런 ListNode로 푸는 문제가 c++로는 익숙하지 않아서 힘들었다.

말 그대로 연결리스트의 특성을 잘 살려서 구현하면 된다!

 

[메인 로직]

각 노드의 값을 합하고 나머지는 dummy노드에 넣어주고 몫은 다음으로 넘겨준다.

ListNode만 익숙하다면, 계산하는 과정을 그닥 어렵지 않다.

 

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        bool check = false;
        ListNode* dummy = new ListNode();
        ListNode* curr = dummy;
        int mok = 0;
        while(l1!=NULL || l2!=NULL){
            int x = (l1!=NULL) ? l1->val : 0;
            int y = (l2!=NULL) ? l2->val : 0;
            int sum = x+y+mok;
            
            if(l1!=NULL){
                l1 = l1->next;
            }
            
            if(l2!=NULL){
                l2 = l2->next;
            }
            curr->val = (sum%10);
            mok = (sum/10);
            if(l1==NULL && l2==NULL) break;
            curr->next = new ListNode();
            curr = curr->next;
        }
        
        if(mok!=0){
            curr->next = new ListNode(mok);
        }
                        
        return dummy;
    }
};