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;
}
}
评论