문제
각 자릿수의 역순으로 이루어진 링크드리스트 두개가 있다.
이를 더한 값을 링크드리스트로 나타내라
/**
* 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;
}
};
반응형
'Problem solving > LeetCode' 카테고리의 다른 글
[LeetCode] 5. Longest Palindromic Substring (C++) (0) | 2023.06.23 |
---|---|
[LeetCode] 4. Median of Two Sorted Arrays (C++) (0) | 2023.06.22 |
[LeetCode] 3. Longest Substring Without Repeating Characters (C++) (0) | 2023.06.20 |
[LeetCode] 1. Two Sum (C++) (0) | 2023.06.16 |