每日一题记录【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 可能 相等 。

---

思路

方法一：暴力解法

两层for循环，遍历下表i和j，不断判断是否符合条件，时间复杂度O(n)。

class Solution {
    public int[] findIndices(int[] nums, int indexDifference, int valueDifference) {
        for (int i = 0; i < nums.length; i++) {
            for (int j = i; j < nums.length; j++) {
                if (j - i >= indexDifference && Math.abs(nums[j] - nums[i]) >= valueDifference) {
                    return new int[]{i, j};
                }
            }
        }
        return new int[]{-1, -1};
    }
}

---

方法二：双指针

一层for循环，起始位置i从indexDifference开始，另外一个指针指向前面indexDifference位置的元素， 即j = i-indexDifference。另外维护【0， j】范围内最大元素与最小元素的下标。如果nums【i】与最大元素和最小元素的取值之差满足条件二，那么就找到了答案。

class Solution {
    public int[] findIndices(int[] nums, int indexDifference, int valueDifference) {
        int[] res = new int[2];
        int n = nums.length;
        res[0] = -1;
        res[1] = -1;
        int minIdx = 0, maxIdx = 0;
        for(int i=indexDifference;i<n;i++) {
            int j = i-indexDifference;
            minIdx = nums[j]<nums[minIdx]?j:minIdx;
            maxIdx = nums[j]<nums[maxIdx]?maxIdx:j;
            if(Math.abs(nums[i]-nums[minIdx])>=valueDifference) {
                return new int[]{minIdx, i};
            }
            if(Math.abs(nums[i]-nums[maxIdx])>=valueDifference) {
                return new int[]{maxIdx, i};
            }
        }
        return res;
    }
}

---