自己从头撸起在Linux环境下用vim 把题目实现,因为对链表和指针都开始遗忘,还是学习到了很多细节。对C++的基础知识还是很多帮助。
#include <iostream>
#include <vector>
#include <assert.h>
using namespace std;
struct listNode{
int val;
listNode * next;
listNode(int data){val=data; next=nullptr;}
};
listNode* list_init(vector<int> v)
{
listNode* list=nullptr;
listNode* head=nullptr;
for(auto& i:v)
{
listNode* temp = new listNode(i) ;
if(list==nullptr){list=temp;head=list;}
else {
list->next =temp;
list = temp;
}
}
return head;
}
void list_free(listNode* head)
{
for(listNode* li= head; li!=nullptr;)
{
listNode* temp= li;
li=li->next;
delete temp;
}
}
void list_show(listNode* head)
{
for(listNode* li=head;li!=nullptr;li=li->next)
{
cout<<li->val<<' ';
}
cout << endl;
}
listNode* addTwoNumber(listNode* list1, listNode* list2)
{
assert((list1!=nullptr)&&(list2!=nullptr));
if(list1==nullptr)
return list2;
if(list2==nullptr)
return list1;
listNode * head=nullptr;
listNode * list = nullptr;
listNode * temp;
int carry=0,num1,num2;
listNode* l1=list1;listNode* l2=list2;
while((l1!=nullptr)||(l2!=nullptr))
{
if(l1){num1=l1->val;l1=l1->next;}
else num1=0;
if(l2){num2=l2->val;l2=l2->next;}
else num2=0;
int count = num1 + num2 +carry;
carry = count/10;
temp = new listNode(count%10) ;
if(head==nullptr){list=temp;head=list;}
else{list->next=temp;list=temp;}
}
if(carry)
{
temp=new listNode(carry);
list->next = temp;
}
return head;
}
int main()
{
listNode* head1=list_init(vector<int>{2,3,6,2});
listNode* head2=list_init(vector<int>{2,9,7});
list_show(head1);
list_show(head2);
listNode* result = addTwoNumber(head1,head2);
list_show(result);
list_free(head1);
list_free(head2);
list_free(result);
}