class Solution {
public:
vector<int> findIndices(vector<int>& nums, int indexDifference, int valueDifference) {
}
};
2905. 找出满足差值条件的下标 II
给你一个下标从 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] 。
提示：
1 <= n == nums.length <= 1050 <= nums[i] <= 1090 <= indexDifference <= 1050 <= valueDifference <= 109原站题解
golang 解法, 执行用时: 100 ms, 内存消耗: 8.8 MB, 提交时间: 2023-10-16 20:48:04
func findIndices(nums []int, indexDifference, valueDifference int) []int {
maxIdx, minIdx := 0, 0
for j := indexDifference; j < len(nums); j++ {
i := j - indexDifference
if nums[i] > nums[maxIdx] {
maxIdx = i
} else if nums[i] < nums[minIdx] {
minIdx = i
}
if nums[maxIdx]-nums[j] >= valueDifference {
return []int{maxIdx, j}
}
if nums[j]-nums[minIdx] >= valueDifference {
return []int{minIdx, j}
}
}
return []int{-1, -1}
}
java 解法, 执行用时: 1 ms, 内存消耗: 58.6 MB, 提交时间: 2023-10-16 20:47:48
class Solution {
public int[] findIndices(int[] nums, int indexDifference, int valueDifference) {
int maxIdx = 0, minIdx = 0;
for (int j = indexDifference; j < nums.length; 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) {
return new int[]{maxIdx, j};
}
if (nums[j] - nums[minIdx] >= valueDifference) {
return new int[]{minIdx, j};
}
}
return new int[]{-1, -1};
}
}
cpp 解法, 执行用时: 112 ms, 内存消耗: 78.2 MB, 提交时间: 2023-10-16 20:47:35
class Solution {
public:
vector<int> findIndices(vector<int> &nums, int indexDifference, int valueDifference) {
int max_idx = 0, min_idx = 0;
for (int j = indexDifference; j < nums.size(); j++) {
int i = j - indexDifference;
if (nums[i] > nums[max_idx]) {
max_idx = i;
} else if (nums[i] < nums[min_idx]) {
min_idx = i;
}
if (nums[max_idx] - nums[j] >= valueDifference) {
return {max_idx, j};
}
if (nums[j] - nums[min_idx] >= valueDifference) {
return {min_idx, j};
}
}
return {-1, -1};
}
};
python3 解法, 执行用时: 136 ms, 内存消耗: 31.3 MB, 提交时间: 2023-10-16 20:47:21
class Solution:
def findIndices(self, nums: List[int], indexDifference: int, valueDifference: int) -> List[int]:
max_idx = min_idx = 0
for j in range(indexDifference, len(nums)):
i = j - indexDifference
if nums[i] > nums[max_idx]:
max_idx = i
elif nums[i] < nums[min_idx]:
min_idx = i
if nums[max_idx] - nums[j] >= valueDifference:
return [max_idx, j]
if nums[j] - nums[min_idx] >= valueDifference:
return [min_idx, j]
return [-1, -1]