写点什么

【LeetCode】绝对差不超过限制的最长连续子数组 Java 题解

用户头像
HQ数字卡
关注
发布于: 2021 年 02 月 21 日

题目

给你一个整数数组 nums ,和一个表示限制的整数 limit,请你返回最长连续子数组的长度,该子数组中的任意两个元素之间的绝对差必须小于或者等于 limit 。


如果不存在满足条件的子数组,则返回 0 。


代码

public class DayCode {    public static void main(String[] args) {        int[] nums = new int[]{1,2,2,3,1,4,2};        int limit = 2;        int ans = new DayCode().longestSubarray(nums, limit);        System.out.println("ans is " + ans);    }
/** * https://leetcode-cn.com/problems/longest-continuous-subarray-with-absolute-diff-less-than-or-equal-to-limit/ * 时间复杂度 O(n log n) * 空间复杂度 O(n) * @param nums * @param limit * @return */ public int longestSubarray(int[] nums, int limit) { TreeMap<Integer, Integer> map = new TreeMap<>(); int n = nums.length; int left = 0; int right = 0; int ans = 0; while (right < n) { map.put(nums[right], map.getOrDefault(nums[right], 0) + 1); while (map.lastKey() - map.firstKey() > limit) { map.put(nums[left], map.get(nums[left]) - 1); if (map.get(nums[left]) == 0) { map.remove(nums[left]); } left++; } ans = Math.max(ans, right - left + 1); right++; }
return ans; }}
复制代码

总结

  • 这个题目是滑动窗口问题,主要是使用了 Java 实现好的 TreeMap 数据结构。

  • TreeMap 是 Red-Black tree 的实现,可以根据 key 按照自然序排序。

  • 每天坚持一题,加油!


发布于: 2021 年 02 月 21 日阅读数: 11
用户头像

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

评论

发布
暂无评论
【LeetCode】绝对差不超过限制的最长连续子数组Java题解