题目来源,待字闺中,原创@陈利人 ,欢迎大家继续关注微信公众账号“待字闺中”
原题:两个单链表(singly linked list),每一个节点里面一个0-9的数字,输入就相当于两个大数了。然后返回这两个数的和(一个新list)。这两个输入的list长度相等。 要求是:不用递归;要求算法在最好的情况下,只遍历两个list一次 ,最差的情况下两遍
分析:如果链表表示的数是从低到高位表示的,则很简单;如果是从高到低,问题就复杂了,进位是万恶之源。
两个单链表,每个节点存储一个0-9的数字,那么一个单链表就表示一个大数。 从高位到低位存,即表头对应的是这个大数的最高位。两个链表的长度相等, 我们要返回一个新的单链表,是这两个输入链表代表的数的和。我们不能使用递归, 不能使用额外的存储空间,即空间复杂度是O(1)。只遍历输入链表一次, 输出链表也是单链表(没有前向指针)。既然只能遍历两个输入链表一次,那我们就从高位加起呗。在这种限制条件下, 这是唯一的出路。然后呢?进位咋整?先加高位,再加低位, 低位产生的进位怎么加到高位去?我们可没有前向指针哦亲。既然没有前向指针, 我们就让一个临时指针指向高位,当低位相加产生进位时,我们就可以操作高位了。 让我们看看图示:
输入链表1: 1 2 3
输入链表2: 1 2 8
输出链表: 2 4
两个指针: p q
当指向输出链表当前结点的指针q发现3+8=11,产生进位,指向高位的p就将结点值加1。 注意,两个0-9的数相加,要么不进位,要么进位为1,只有两种情况。因此, 我们不用考虑进位是其它数,这一点很重要,后面会看到的。这样就OK了吗?当然不是,如果你遇上连续进位,怎么破?请看下面的情况:
输入链表1: 1 2 3 4 5
输入链表2: 1 7 6 5 9
显然,指向高位的指针p总是紧跟着指向当前结点的指针q是不行的, 这样当遇上连续进位时,比p更高位的位也需要改变。既然p不能紧跟着q, 我们就不让它们紧挨着,给它们产生点距离。考虑一下,什么情况下会产生连续进位? 9! 嗯,遇上9的时候。它要连续进位到哪一位?不为9的那一位。因此,指针p 要停留在和不为9的那一位上,看图示:
输入链表1: 1 2 3 4 5
输入链表2: 1 7 6 5 9
输出链表: 2 9 9 9
两个指针: p q
这回当q发现,需要进位了,只需要把p所指结点加1,然后把p,q间的结点都置0即可。 为什么都置0了呢,因为进位只可能是1,9+1=10,留在该位的自然是0了。分析完毕,这种方法在任何时候都只需要遍历输入链表一次,空间复杂度O(1)。具体代码如下:
#include<iostream> #include<memory.h> using namespace std; struct Node { int val; struct Node* next; Node(int x):val(x),next(NULL){} }; Node* AddLinkList(Node* head1,Node* head2) { if(head1 == NULL)return head2; if(head2 == NULL)return head1; Node* head = new Node(head1->val+head2->val);//结果链表 Node* p1 = head1->next,*p2 = head2->next,*p,*q = head,*pre=head;//q是可能会有进位的节点 while(p1 && p2) { int sum = p1->val+p2->val; p = new Node(sum % 10); pre->next = p; pre = p; if(sum > 9) { q->val += 1; for(q=q->next;q!=p;q=q->next)q->val = 0;//p节点加1,中间的9变为0 } else if(sum < 9)q = p;//可能会有进位的节点修改为p p1 = p1->next; p2 = p2->next; } if(head->val > 9) { p = new Node(1); p->next = head; head->val -= 10; head = p; } return head; } Node* MakeLinkedList(int* data,int n) { if(data == NULL || n<=0)return NULL; Node* head = new Node(data[0]); Node* pre = head,*p; int i; for(i = 1;i<n;i++) { p = new Node(data[i]); pre->next = p; pre = p; } return head; } void print(Node* head) { while(head) { cout << head->val << " "; head = head->next; } cout<<endl; } void deleteNode(Node* &head) { if(head == NULL)return; if(head->next)deleteNode(head->next); delete head; head = NULL; } int main() { int n,i; Node* head1,*head2; cout << "----------输入每个链表的长度---------"<<endl; while(cin >> n) { int* data = new int[n]; cout <<"-------输入第一个链表的元素------"<<endl; for(i=0;i<n;i++)cin >> data[i]; head1 = MakeLinkedList(data,n); cout <<"-------第一个链表为-------------"<<endl; print(head1); cout <<"-------输入第二个链表的元素------"<<endl; for(i = 0;i<n;i++)cin >> data[i]; head2 = MakeLinkedList(data,n); cout <<"-------第二个链表为-------------"<<endl; Node* head = AddLinkList(head1,head2); cout <<"-------两个链表的和为-----------"<<endl; print(head); deleteNode(head1); deleteNode(head2); deleteNode(head); cout << "----------输入每个链表的长度---------"<<endl; } return 0; }
原文:http://blog.csdn.net/fangjian1204/article/details/37695379