leetcode 2.两数相加
题目描述
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
解题思路
该题是链表对应值相加,没有难度,但是要考虑的情况比较多
1.两个链表一样长
2.链表的长度不相等
3.进位情况
4.最后一次相加进位的情况
5.单个链表进位的情况
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* cur = new ListNode(-1); // 定义一个链表存储结果
ListNode* resNum = cur;
bool flag = false; // 进位标志
ListNode* cur1 = l1;
ListNode* cur2 = l2;
// 为了防止两个链表不是一样长,只有两个节点都不为空的时候,才进行相加操作
while((cur1 != NULL) && (cur2 !=NULL)){
cur->next = new ListNode(0);
cur = cur->next;
// 进位的情况下,需要在两数相加的结果上加1, 同时判断当前相加是否进位
if(flag){
if((cur1->val + cur2->val + 1)/10 == 0){
flag = false;
}
else{
flag = true;
}
cur->val = (cur1->val + cur2->val + 1)%10;
}
else{ // 不进位的情况下,只要判读相加是否进位。
if((cur1->val + cur2->val)/10 == 0){
flag = false;
}
else{
flag = true;
}
cur->val = (cur1->val + cur2->val)%10;
}
// 移动指针位置
cur1 = cur1->next;
cur2 = cur2->next;
}
// l1比较长的情况,这里需要考虑有多个9的情况,所以需要遍历一下
while(cur1 != NULL){
cur->next = cur1;
cur = cur->next;
if(flag){
cur->val += 1;
if(cur->val == 10){
cur->val = 0;
flag = true;
}
else{
flag = false;
}
}
else{
break;
}
cur1 = cur1->next;
}
// l2比较长的情况,同样需要考虑多个9的情况
while(cur2 != NULL){
cur->next = cur2;
cur = cur->next;
if(flag){
cur->val += 1;
if(cur->val == 10){
cur->val = 0;
flag = true;
}
else{
flag = false;
}
}
else{
break;
}
cur2 = cur2->next;
}
// 最后相加进位的情况
if(flag){
cur->next = new ListNode(1);
}
return resNum->next;
}
};
上面是直接按照思路写的,下面是精简一下代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode(-1); // 存储结果的链表
ListNode* cur = head; // 移动节点
bool flag = false; // 进位
int sum = 0;
while(l1 != NULL || l2 != NULL){
sum = 0; // sum置位
// l1是否循环完毕,取val值
if(l1 != NULL){
sum += l1->val;
l1 = l1->next;
}
// l2是否循环完毕,取val值
if(l2 != NULL){
sum += l2->val;
l2 = l2->next;
}
// 进位加1
if(flag){
sum += 1;
}
cur->next = new ListNode(sum%10); // 向链表中添加值
cur = cur->next;
flag = (sum/10 != 0) ? true:false; // 当前相加的结果是否进位
}
// 最后一位相加是否有进位情况
if(flag){
cur->next = new ListNode(1);
}
return head->next;
}
};
欢迎大家关注我的个人公众号,同样的也是和该博客账号一样,专注分享技术问题,我们一起学习进步