LeetCode OJ (2)

本文介绍了如何优化链表加法的实现方式。通过对比两个不同的代码实现,展示了如何简化代码并提高效率。从初始的复杂版本逐步改进到简洁高效的版本,并解释了优化的关键步骤。

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

初始代码块(初学者水平)

ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
    ListNode *header=NULL,*sum=NULL;

    int carry=0;
    bool flag=false;
    while(l1!=NULL && l2!=NULL){
        ListNode *tmp=new ListNode(0);
        carry=l1->val+l2->val+flag;
        if(carry>=10){
            tmp->val = carry-10;
            flag = true;
        }
        else{
            tmp->val=carry;
            flag=false;
        }
        if(!header)
            header=tmp;
        else
            sum->next=tmp;
        sum=tmp;
        l1=l1->next;
        l2=l2->next;
    }

    ListNode *last;
    if(l1==NULL)
        last=l2;
    else
        last=l1;
    carry=0;
    while(last!=NULL){
        ListNode *tmp=new ListNode(0);
        carry=last->val+flag;
        if(carry>=10){
           tmp->val=carry-10; 
           flag=true;
        }
        else{
            tmp->val=carry;
            flag=false;
        }
        sum->next=tmp;
        sum=tmp;
        last=last->next;
    }
    if(flag){
        ListNode *tmp=new ListNode(0);
        tmp->val=flag;
        sum->next=tmp;
    }
    return header;
}
}

经过优化和学弟提醒之后

ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
    ListNode *header=NULL,*sum=NULL;
    int carry=0;
    while(l1 || l2){
        new ListNode(0);
        if(l1)  carry+=l1->val;
        if(l2)  carry+=l2->val;
        ListNode *tmp=new ListNode(carry%10);
        if(!header)
            header=tmp;
        else
            sum->next=tmp;
        carry/=10;
        sum=tmp;
        if(l1) l1=l1->next;
        if(l2) l2=l2->next;
    }
    if(carry){
    ListNode *tmp=new ListNode(carry);
    sum->next=tmp;
    }
    return header;
}

我太悲伤了,第二个代码块是我奋斗的目标!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值