问题:给定两个单链表,求出单链表中元素组成的数字,两个单链表分别可以生成两个数字,求着两个数字的和。链表的头到尾对应数字的高位到低位。
思路:
1、考虑到链表的头为数字高位,且两数相加应该从低位开始运算,所以用栈来存储链表的数字
2、定义两个栈s1和s2分别用来存储两个链表的数据
3、分别利用while循环将两个链表存储到栈中
4、考虑到两数相加可能会大于10,产生进位,所以设置变量cnt表示进位,并定义返回值指针
5、两个栈不同时为空,就分别取出两个栈的栈顶元素相加+进位cnt;
6、定义一个新的链表节点用来存储求和数值的个位,即sum%10;
7、将新的链表节点作为返回结果的链表的最后一个节点,即temp->next=res;
8、将返回指针前移,即res=temp;
9、链表的节点值加完后,cnt不为0.则新建链表节点来存储cnt,并将其链接到res链表,res前移。
10、直到循环结束,将res输出
#include <iostream>
#include <stack>
using namespace std;
struct ListNode
{
int value;
ListNode* next;
ListNode(int x) :value(x), next(NULL) {}
};
ListNode* addList(ListNode* head1, ListNode* head2)
{
stack<int> s1, s2;//定义两个栈来存储链表的数值
while (head1)//将链表1入栈
{
s1.push(head1->value);
head1 = head1->next;
}
while (head2)//将链表2入栈
{
s2.push(head2->value);
head2 = head2->next;
}
int cnt = 0;//定义进位的变量
ListNode* res = nullptr;//定义返回链表的指针;
while (!s1.empty() || !s2.empty())//两个栈不同时为空进入循环
{
int x1 = s1.empty() ? 0 : s1.top();//若栈1为空,则x1=0,否则x1为栈1的栈顶元素;
int x2 = s2.empty() ? 0 : s2.top();
if (!s1.empty())//栈1不为空,出栈
s1.pop();
if (!s2.empty())
s2.pop();
int sum = x1 + x2 + cnt;//两个数值的求和
ListNode* temp = new ListNode(sum % 10);//定义新的链表节点来存储求和的各位值
temp->next = res;//将定义的节点链接到返回链表
res = temp;//指针前移
}
if (cnt > 0)//若两个链表的数值都相加完,但是进位不为空
{
ListNode* tem = new ListNode(cnt);//定义新的链表节点来存储进位的数值
tem->next = res;//将新的节点链接到res
res = tem;//指针前移
}
return res;
}
void printList(ListNode* head)//打印链表
{
while (head)//只要head指针指向不为空
{
if (head->next)//head->指针指向不为空
cout << head->value << "->";
else//head->指针指向为空
cout << head->value << endl;
head = head->next;
}
}
int main()
{
ListNode* head1 = new ListNode(1);//单链表1的定义
ListNode* n1 = new ListNode(2);
ListNode* n2 = new ListNode(3);
ListNode* n3 = new ListNode(4);
ListNode* n4 = new ListNode(5);
head1->next = n1;
n1->next = n2;
n2->next = n3;
n3->next = n4;
n4->next =NULL;
printList(head1);
ListNode* head2 = new ListNode(1);//单链表2的定义
ListNode* nn1 = new ListNode(2);
ListNode* nn2 = new ListNode(3);
head2->next = nn1;
nn1->next = nn2;
nn2->next = NULL;
printList(head2);
ListNode* d = addList(head1, head2);
printList(d);
system("pause");
return 0;
}
输出结果为: