leetcode 1438. 绝对差不超过限制的最长连续子数组（滑动窗口+treemap）

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

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

示例 1：

输入：nums = [8,2,4,7], limit = 4
输出：2
解释：所有子数组如下：
[8] 最大绝对差 |8-8| = 0 <= 4.
[8,2] 最大绝对差 |8-2| = 6 > 4.
[8,2,4] 最大绝对差 |8-2| = 6 > 4.
[8,2,4,7] 最大绝对差 |8-2| = 6 > 4.
[2] 最大绝对差 |2-2| = 0 <= 4.
[2,4] 最大绝对差 |2-4| = 2 <= 4.
[2,4,7] 最大绝对差 |2-7| = 5 > 4.
[4] 最大绝对差 |4-4| = 0 <= 4.
[4,7] 最大绝对差 |4-7| = 3 <= 4.
[7] 最大绝对差 |7-7| = 0 <= 4.
因此，满足题意的最长子数组的长度为 2 。

解题思路

利用treemap，在统计窗口元素个数的同时，对元素的大小进行排序，得出窗口内的最大值和最小值

代码

class Solution {
    public int longestSubarray(int[] nums, int limit) {

        TreeMap<Integer,Integer> map=new TreeMap<>();

        int l=0,r=0,res=0;
        while (r<nums.length)
        {
            map.put(nums[r],map.getOrDefault(nums[r],0)+1);
             while (Math.abs(map.firstKey()-map.lastKey())>limit)
            {
                map.put(nums[l],map.get(nums[l])-1);
                if(map.get(nums[l])==0) map.remove(nums[l]);//该元素个数为0，则从窗口中移除
                l++;
                
            }
            res= Math.max(r-l+1,res);
            r++;
        }
        return res;
    }
}