【TreeMap】LeetCode 1438. 绝对差不超过限制的最长连续子数组

题目链接

1438. 绝对差不超过限制的最长连续子数组

思路

本题一看就是一个要用到滑动窗口的题目，难点在于如何快速求窗口中的最大值和最小值。幸好 Java 中有 TreeMap 这个数据结构，它在构成 map 映射的同时，还能够保证其中的元素有序，这样就能很方便求得窗口中的最大值和最小值了。

代码

class Solution {
    public int longestSubarray(int[] nums, int limit) {
        TreeMap<Integer, Integer> map = new TreeMap<>();
        int left = 0;
        int right = 0;
        int n = nums.length;
        int result = 0;

        while(right < n){
            map.put(nums[right], map.getOrDefault(nums[right], 0) + 1);

            // 直接从 TreeMap 中取窗口中的最大值和最小值
            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++;
            }

            result = Math.max(result, right - left + 1);
            right++;
        }

        return result;
    }
}