add-two-numbers(LeetCode)

tech2023-01-06  104

原题来自牛客网LeetCode:链接

给定两个代表非负数的链表,数字在链表中是反向存储的(链表头结点处的数字是个位数,第二个结点上的数字是十位数…),求这个两个数的和,结果也用链表表示。

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)

输出: 7 -> 0 -> 8

You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)

Output: 7 -> 0 -> 8

示例1

输入 {0},{0}

输出 {0}

示例2

输入 {0},{1}

输出 {1}

思路

数字已经是反向存储了,因此逐位相加然后进位即可,用一个变量out记录是否有进位。

用尾插法添加每一个结点即可,开头第一个结点可以设为0方便尾插,最后返回head->next即可。

代码如下:

class Solution { public: ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) { if(!l1&&!l2) return NULL; ListNode *ans,*p=NULL; int out=0,num,sum=0; p=new ListNode(0); ans=p; while(l1||l2||out){ sum=0; if(l1){ sum+=l1->val; l1=l1->next; } if(l2){ sum+=l2->val; l2=l2->next; } if(out) sum+=1; out=(sum>=10? 1:0); num=sum%10; ListNode *tmp=new ListNode(num); p->next=tmp; p=tmp; } return ans->next; } };
最新回复(0)