class Solution {
public:
vector<int> findIndices(vector<int>& nums, int indexDifference, int valueDifference) {
}
};
2903. 找出满足差值条件的下标 I
给你一个下标从 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 <= 1000 <= nums[i] <= 500 <= indexDifference <= 1000 <= valueDifference <= 50原站题解
rust 解法, 执行用时: 0 ms, 内存消耗: 2.1 MB, 提交时间: 2024-05-25 08:28:38
impl Solution {
pub fn find_indices(nums: Vec<i32>, index_difference: i32, value_difference: i32) -> Vec<i32> {
let mut max_idx = 0;
let mut min_idx = 0;
for j in index_difference as usize..nums.len() {
let i = j - index_difference as usize;
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] >= value_difference {
return vec![max_idx as i32, j as i32];
}
if nums[j] - nums[min_idx] >= value_difference {
return vec![min_idx as i32, j as i32];
}
}
vec![-1, -1]
}
}
javascript 解法, 执行用时: 68 ms, 内存消耗: 42.8 MB, 提交时间: 2023-10-16 17:25:02
/**
* @param {number[]} nums
* @param {number} indexDifference
* @param {number} valueDifference
* @return {number[]}
*/
var findIndices = function(nums, indexDifference, valueDifference) {
let maxIdx = 0, minIdx = 0;
for (let j = indexDifference; j < nums.length; j++) {
var 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 [maxIdx, j];
}
if (nums[j] - nums[minIdx] >= valueDifference) {
return [minIdx, j];
}
}
return [-1, -1];
};
golang 解法, 执行用时: 0 ms, 内存消耗: 2.7 MB, 提交时间: 2023-10-16 17:23:34
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}
}
cpp 解法, 执行用时: 8 ms, 内存消耗: 17.6 MB, 提交时间: 2023-10-16 17:23:19
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};
}
};
java 解法, 执行用时: 0 ms, 内存消耗: 42.8 MB, 提交时间: 2023-10-16 17:23:03
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};
}
}
python3 解法, 执行用时: 40 ms, 内存消耗: 15.9 MB, 提交时间: 2023-10-16 17:22:36
# 双指针，维护最大最小
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]