请问可以怎么优化下

love_qj

给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。

如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外，这两个数都不会以 0 开头。

示例：

输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出：7 -> 0 -> 8
原因：342 + 465 = 807

/**
* Definition for singly-linked list.
* struct ListNode {
*     int val;
*     struct ListNode *next;
* };
*/


struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2){
    struct ListNode* head, *tail;
    head = (struct ListNode*)malloc(sizeof(struct ListNode));
    tail = head->next;
    head->next = NULL;
    int flag = 0;  //设置进位变量
    while(l1 && l2){
        struct ListNode* l3;
        l3 = (struct ListNode*)malloc(sizeof(struct ListNode));
        //先链接链表，再进行赋值
        if(head->next == NULL)  //头指针为空作判断条件
        {
            tail = l3;
            tail->next = NULL;
            head->next = tail;
        }
        else{
            tail->next = l3;
            tail = tail->next;
            tail->next = NULL;
        }
        //赋值
        if(l1->val + l2->val >= 10){
            flag = 1;
            l3->val = l1->val + l2->val - 10;
        }
        else{
            l3->val = l1->val + l2->val + flag;
            flag = 0;
        }
        l1 = l1->next;
        l2 = l2->next;
    }
    //循环结束条件为l1为空或者l2为空
    if(l1 == NULL){
        while(l2){
            tail->next = l2;
            tail = tail->next;
            tail->next = NULL;
        }
    }
    else if(l2 == NULL){
        while(l1){
            tail->next = l1;
            tail = tail->next;
            tail->next = NULL;
        }
    }
    return head->next;
}
leedcode上得题，提交说超出了时间限制，如下：
最后执行的输入：
[9,9,9,9,9,9,9]
[9,9,9,9]

xieglt

没有考虑进位啊。

xieglt

1. struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2)
   
2. {
   
3.     struct ListNode head, *tail;
   
4.     head.next = NULL;
   
5.         tail = &head;
   
6.         
   
7.     int flag = 0;  //设置进位变量
   
8. 
   
9.     while(l1 || l2)
   
10.         {
    
11.                 tail->next = (ListNode *)malloc(sizeof(ListNode));
    
12.                 tail = tail->next;
    
13.                 tail->next = NULL;
    
14.                
    
15.                 if(l1 != NULL)
    
16.                 {
    
17.                         flag += l1->val;
    
18.                         l1 = l1->next;
    
19.                 }
    
20. 
    
21.                 if(l2 != NULL)
    
22.                 {
    
23.                         flag += l2->val;
    
24.                         l2 = l2->next;
    
25.                 }
    
26. 
    
27.                 tail->val = flag % 10;
    
28.                 flag /= 10;
    
29.     }
    
30. 
    
31.         if(flag != 0)
    
32.         {
    
33.                 tail->next = (ListNode *)malloc(sizeof(ListNode));
    
34.                 tail->next->val = flag;
    
35.                 tail->next->next = NULL;
    
36.         }
    
37.        
    
38.     return head.next;
    
39. }

复制代码

love_qj

xieglt 发表于 2020-10-12 09:25

你这个是肯定有问题的，这是指针函数啊哥哥

xieglt

love_qj 发表于 2020-10-12 09:37
你这个是肯定有问题的，这是指针函数啊哥哥

说话不要那么肯定，不运行一下怎么知道有问题呢？