博主在面试字节跳动实习生岗位时,遇见了链表求和的问题。而刷过 leetcode
的同学们对此也不会陌生,因为它就是 Leetcode
上的原题——Add Two Numbers II。因此,博主在此给出解题思路和解题方案,希望能对大家有所帮助。
简介
给定两个非空链表来代表两个非负整数。其中,数字最高位位于链表开始位置。它们的每个节点只存储单个数字。我们需要将这两数相加会返回一个新的链表。例如,(7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4) = 7 -> 8 -> 0 -> 7
思路
经过分析,我们不难发现,根据加法原则,从低位开始相加,并且考虑进位标志。然而,链表开始位置是数字的最高位。因此,我们只需要先将链表进行翻转,然后根据加法原则即可完成此题。
代码
根据前一小节的分析,我们可有代码如下
public class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}
class Solution {
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
if(l1 == null || l2 == null) return null;
// 栈保存数据
Stack<Integer> s1 = new Stack<Integer>();
Stack<Integer> s2 = new Stack<Integer>();
ListNode p = l1;
// 入栈
while(p != null){
s1.push(p.val);
p = p.next;
}
p = l2;
while(p != null){
s2.push(p.val);
p = p.next;
}
// 求和结果
ListNode res = new ListNode(-1);
p = res;
int sum = 0;
while(!s1.empty() || !s2.empty()){
sum /= 10;
if(!s1.empty()) sum += s1.pop();
if(!s2.empty()) sum += s2.pop();
ListNode newNode = new ListNode(sum % 10);
newNode.next = p.next;
p.next = newNode;
}
if(sum / 10 == 1){
ListNode newNode = new ListNode(1);
newNode.next = p.next;
p.next = newNode;
}
return res.next;
}