从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]