题目
题目链接
给你一个下标从 0 开始、长度为 n 的整数数组 nums ,以及整数 indexDifference 和整数 valueDifference 。
你的任务是从范围 [0, n - 1] 内找出 2 个满足下述所有条件的下标 i 和 j :
abs(i - j) >= indexDifference 且
abs(nums[i] - nums[j]) >= valueDifference
返回整数数组 answer。如果存在满足题目要求的两个下标,则 answer = [i, j] ;否则,answer = [-1, -1] 。如果存在多组可供选择的下标对,只需要返回其中任意一组即可。
注意:i 和 j 可能 相等 。
示例
示例 1:
输入:nums = [5,1,4,1], indexDifference = 2, valueDifference = 4
输出:[0,3]
解释:在示例中,可以选择 i = 0 和 j = 3 。
abs(0 - 3) >= 2 且 abs(nums[0] - nums[3]) >= 4 。
因此,[0,3] 是一个符合题目要求的答案。
[3,0] 也是符合题目要求的答案。
示例 2:
输入:nums = [2,1], indexDifference = 0, valueDifference = 0
输出:[0,0]
解释:
在示例中,可以选择 i = 0 和 j = 0 。
abs(0 - 0) >= 0 且 abs(nums[0] - nums[0]) >= 0 。
因此,[0,0] 是一个符合题目要求的答案。
[0,1]、[1,0] 和 [1,1] 也是符合题目要求的答案。
示例 3:
输入:nums = [1,2,3], indexDifference = 2, valueDifference = 4
输出:[-1,-1]
解释:在示例中,可以证明无法找出 2 个满足所有条件的下标。
因此,返回 [-1,-1] 。
分析题目
这个题非常巧妙,因为要求下标必须差 indexDifference,所以后一个指针(j)在前indexDifference时一定不会满足条件。
注意:这里一定是对后一个指针(j)进行遍历。
这时后一个指针(j)前面的窗口都是可能使得abs(nums[i] - nums[j]) >= valueDifference,所以只要前面窗口中有一个指针(i)满足nums[i] - nums[j]>= valueDifference或者nums[j] - nums[i]) >= valueDifference
这样问题就转变为求前面窗口的最大最小值了。
如果遍历前一个指针(i),那么一定需要求后面窗口的最大最小值,相当于双重循环,时间复杂度较大,不如遍历后一个指针(j)。
代码
class Solution {
public:
vector<int> findIndices(vector<int>& nums, int indexDifference, int valueDifference) {
//标识前面窗口的最大最小值
int maxIdx = 0, minIdx = 0;
vector<int> res(2,-1);
for (int j = indexDifference; j < nums.size(); j++) {
int i = j - indexDifference;
//更新前面窗口中的最大、最小值
if (nums[i] > nums[maxIdx]) {
maxIdx = i;
} else if (nums[i] < nums[minIdx]) {
minIdx = i;
}
//判断是否满足出口条件
if (nums[maxIdx] - nums[j] >= valueDifference) {
res[0]=maxIdx;
res[1]=j;
return res;
}
if (nums[j] - nums[minIdx] >= valueDifference) {
res[0]=minIdx;
res[1]=j;
return res;
}
}
//没有满足条件的值,返回(-1,-1)
return res;
}
};