从C ++的链接列表中删除零和连续节点
假设我们给出了链表的开头;我们必须重复删除总计为0的节点的连续序列,直到没有此类序列为止。因此,这样做之后,我们必须返回最终链接列表的头部。因此,如果列表类似于[1,2,-3,3,1],则结果将为[3,1]。
为了解决这个问题,我们将遵循以下步骤-
创建一个名为dummy的节点,并将其存储为0,然后再设置dummy:=head
创建一个映射m,将键0的哑元存储到m中,设置sum=0
而head不为null-
sum:=sum+head的值,设置m[sum]:=head,head:=head的下一个
头:=假人
和:=0
而head不为null
总和:=总和+头值
temp:=m[sum]
如果temp不是头,则头的下一个:=temp的下一个
头:=头下
返回假人的下一个
让我们看下面的实现以更好地理解-
示例
#include <bits/stdc++.h> using namespace std; class ListNode{ public: int val; ListNode *next; ListNode(int data){ val = data; next = NULL; } }; ListNode *make_list(vector<int> v){ ListNode *head = new ListNode(v[0]); for(int i = 1; i<v.size(); i++){ ListNode *ptr = head; while(ptr->next != NULL){ ptr = ptr->next; } ptr->next = new ListNode(v[i]); } return head; } void print_list(ListNode *head){ ListNode *ptr = head; cout << "["; while(ptr){ cout << ptr->val << ", "; ptr = ptr->next; } cout << "]" << endl; } class Solution { public: ListNode* removeZeroSumSublists(ListNode* head) { ListNode* dummy = new ListNode(0); dummy->next = head; unordered_map <int, ListNode*> m; m[0] = dummy; int sum = 0; while(head){ sum += head->val; m[sum] = head; head = head->next; } head = dummy; sum = 0; while(head){ sum += head->val; ListNode* temp = m[sum]; if(temp != head){ head->next = temp->next; } head = head->next; } return dummy->next; } }; main(){ vector<int> v1 = {1,2,-3,3,1}; ListNode *head = make_list(v1); Solution ob; print_list(ob.removeZeroSumSublists(head)); }
输入值
[1,2,-3,3,1]
输出结果
[3,1]