您现在的位置是:首页 >技术杂谈 >【LeetCode】每日一题 -- 1171. 从链表中删去总和值为零的连续节点 -- Java Version网站首页技术杂谈
【LeetCode】每日一题 -- 1171. 从链表中删去总和值为零的连续节点 -- Java Version
简介【LeetCode】每日一题 -- 1171. 从链表中删去总和值为零的连续节点 -- Java Version
题目链接:https://leetcode.cn/problems/remove-zero-sum-consecutive-nodes-from-linked-list/
1. 题解(1171. 从链表中删去总和值为零的连续节点)
2021年字节二面真题
1.1 暴力解法:穷举
时间复杂度 O(n2),空间复杂度O(1)
【解题思路】:
由题意得,题目中要我们删除的是 总和 值为0
的连续节点组成的序列,这句话的重点有两个:
- 总和为0;
- 连续;
……
明白了这两点,我们很容易想出该题的暴力解法:从头开始,对每个节点都向后遍历一次,找出遍历过程中导致总和为0的节点,然后让该节点的前一节点指向恰好导致总和为0的节点的后一节点,完成删除操作。
PS:定义哨兵节点的原因,就是防止头节点也有可能会被删除(需要前一节点)。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode removeZeroSumSublists(ListNode head) {
ListNode sentinel = new ListNode(Integer.MAX_VALUE,head);
ListNode cur = sentinel.next;
ListNode pre = sentinel;
while(cur != null)
{
ListNode node = cur;
int sum = 0;
while (node != null)
{
sum += node.val;
node = node.next;
if (sum == 0)
{
pre.next = node;
//break;
}
}
pre = cur;
cur = cur.next;
}
return sentinel.next;
}
}
1.2 精简解法:HashMap ⭐
时间复杂度O(n),空间复杂度O(n)
【解题思路】:
使用HashMap
实现类似前缀和的思想,分为两次遍历:
- 第一次遍历,将达到当前节点时的总和 sum 和当前节点一起存入 map;
- 第二次遍历,借助了HashMap的特性,如果出现了重复的key,那么后一个键值对就会覆盖掉前一个,因此,当我们再次遍历时,如果链表中真的存在总和 值为
0
的连续节点组成的序列,那么我们此时通过map.get(sum)
找到的就是这个连续序列的最后一个节点,那么我们就可以让当前节点的next
指向map.get(sum).next
,完成删除序列。
class Solution {
public ListNode removeZeroSumSublists(ListNode head) {
ListNode sentinel = new ListNode(0,head);
HashMap<Integer, ListNode> map = new HashMap<>();
// 首次遍历建立 节点处链表和<->节点 哈希表
// 若同一和出现多次会覆盖,即记录该sum出现的最后一次节点
int sum = 0;
for (ListNode cur = sentinel; cur != null; cur = cur.next)
{
sum += cur.val;
map.put(sum, cur);
}
// 第二遍遍历 若当前节点处sum在下一处出现了则表明两结点之间所有节点和为0 直接删除区间所有节点
sum = 0;
for (ListNode cur = sentinel; cur != null; cur = cur.next)
{
sum += cur.val;
cur.next = map.get(sum).next;
}
return sentinel.next;
}
}
2. 参考资料
风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。