leetcode 随机刷题记录 1171从链表中删去总和值为零的连续节点

最新推荐文章于 2024-10-07 19:00:27 发布

没事起什么名字

最新推荐文章于 2024-10-07 19:00:27 发布

阅读量300

点赞数

分类专栏： leetcode 文章标签：链表 leetcode list

本文链接：https://blog.csdn.net/weixin_44104745/article/details/121860685

版权

链表操作前缀和子列表删除算法优化数据结构

关键词由CSDN通过智能技术生成

leetcode 专栏收录该内容

70 篇文章 0 订阅

订阅专栏

思路：
遍历得到前缀和，前缀和相等的就可以删除，
得到前后对照的表，右边越右越好，从左往右再遍历，遇到对应的直接跳转，写的挺垃圾的，大家看个乐

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    vector <int> before;
    int left[2100000];
    int right[2100000];
    int ifOk[1100];
    int l;
    ListNode* removeZeroSumSublists(ListNode* head) {
        getBefore(head, 0);
         l = before.size();
        for(int i = 0; i < 2100000; i++){
            left[i] = -1;
            right[i] = -1;
        }
        for(int i = 0; i < l; i++){
            if(left[before[i]] == -1){
                left[before[i]] = i;
            }else{
                right[before[i]] = i;
            }
        }
        ListNode *temp = new ListNode;
        temp->next = head;
        getSon(0, temp, head);
        return temp->next;
    }
    void getBefore(ListNode* node, int val){
        before.push_back(val + 1000005);
        if(node != nullptr){
            getBefore(node->next, val + node->val);
        }
    }

    void getSon(int index, ListNode * p, ListNode *now){
      //  cout << index << endl;
        if(index == l - 1){
            return;
        }
        int temp = right[before[index]];
        if(temp == -1){
            getSon(index + 1, now, now->next);
        }else{
            right[before[index]] = -1;
            now = getNode(now, temp - index);
            if(now == nullptr){
                //cout << "?";
                p->next = now;
                return;
            }
            p->next = now;
            getSon(temp, p, now);
            
        }
    }
    ListNode *getNode(ListNode* node, int index){
        while(index > 0){
            node = node ->next;
            index--;
        }
        return node;
    }
    
};