首页 > 代码库 > Add Two Numbers问题

Add Two Numbers问题

问题描述:

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

 

这个问题还是比较简单的,应该主要是在考察对链表的掌握程度吧。

struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2) {    int carry=0;    struct ListNode *p1=l1,*p2=l2,*answer,*p,*tmp;    answer=malloc(sizeof(struct ListNode));    p=answer;    while(p1!=NULL||p2!=NULL||carry){        p->val=carry?(p1?p1->val:0)+(p2?p2->val:0)+1:(p1?p1->val:0)+(p2?p2->val:0);        carry=0;        if(p->val>9){            p->val%=10;            carry=1;        }        p1=p1?p1->next:NULL;        p2=p2?p2->next:NULL;        tmp=malloc(sizeof(struct ListNode));        p->next=tmp;        p=p->next;    }    p->next=NULL;    for(p=answer;p->next->next!=NULL;p=p->next)        ;    tmp=p->next;    p->next=NULL;    free(tmp);    return answer;}

AC倒是AC了,仅仅击败了2.32%的C提交....果断不能忍啊。让我想想,malloc应该开销挺大的,与其创建一个崭新的链表,不如在l1或者l2的基础上修改原来的链表。

struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2) {    int carry=0;    struct ListNode *p1=l1,*p2=l2,*tmp,*p;    while(p1->val!=10||p2!=NULL||carry){        p1->val=carry?(p1->val!=10?p1->val:0)+(p2?p2->val:0)+1:(p1->val!=10?p1->val:0)+(p2?p2->val:0);        carry=0;        if(p1->val>9){            p1->val%=10;            carry=1;        }        p2=p2?p2->next:NULL;        if(p1->next==NULL){            p1->next=malloc(sizeof(struct ListNode));            p1->next->val=10;            p1->next->next=NULL;        }        p1=p1->next;    }    for(p=l1;p->next->next!=NULL;p=p->next)        ;    tmp=p->next;    p->next=NULL;    free(tmp);    return l1;}

时间从35ms减到了18ms,击败53%的C提交。如果在函数开始时先判断l1,l2哪一个长度更长,以更长的链表为基础链表的话应该是能获取更快的速度的,然后我试了一下,竟然还慢了1ms,应该是测试用例的问题吧。

不过这道题也没有多少追求速度的必要,做到这个程度我觉得已经可以收工了。

Add Two Numbers问题