Problem solving/LeetCode
[LeetCode] 2. Add Two Numbers (C++)
겸
2023. 6. 16. 02:04
문제
각 자릿수의 역순으로 이루어진 링크드리스트 두개가 있다.
이를 더한 값을 링크드리스트로 나타내라
/**
* 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) {}
* };
*/
생각할 것
- int나 long으로 리스트를 더해서 나타낼까 생각했지만, 문제의 조건에서 연결된 노드의 수가 최대 100개이다.
- carry를 두고 더해야한다.
- carry도 while문에 포함시켜서 모두 null일 때 끝까지 실행되도록 하기
- tmp 노드를 두고 next로 이동해야 한다.
- 만약에 return res; 를 한다면 0 7 0 8이 결괏값으로 출력됨
- 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;
}
};
반응형