LeetCode 1171. 从链表中删去总和值为零的连续节点

浮生日记

__今天是状态很差的一天,这几天的睡眠一直不好,对床的哥们半夜把电脑拿上床玩,一直开着最亮的屏幕,反光在墙上太刺眼了,搞得我睡不着,哎。白天在图书馆困的和狗一样,不停地打哈欠,中午想着回去美美的睡个午觉,好不容易睡着了,又被回来的老哥吵醒了。

正文

原题链接
这个题用哈希表直接跳过和为0的链表序列(注意哈希表中的 key 值是唯一的)
以前n个节点的和为 key,第n个节点为value;
后序和与前面和相同即 key 值相同,新的 value 会覆盖旧的 value
重新连接时,按照原序列中的 “和” 顺序连接,先连接了产生了0的一段,此后又指向这一段之前的那一个元素,这样就跳过了和为0的一段。

太妙了!!!!

class Solution {
    
    
public:
    ListNode* removeZeroSumSublists(ListNode* head) {
    
    
        map<int, ListNode*> myMap;
        ListNode *extraHead = new ListNode(0);
        extraHead->next = head;

        int sum = 0;
        for(ListNode* plmm = extraHead; plmm !=nullptr; plmm = plmm->next){
    
    
            sum += plmm->val;
            myMap[sum] = plmm;
        }
        sum = 0;
        for(ListNode* plmm = extraHead; plmm !=nullptr; plmm = plmm->next){
    
    
            sum += plmm->val;
            plmm->next = myMap[sum]->next;
        }
        return extraHead->next;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_43078427/article/details/109146950