题目:
给你一个下标从 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] 。
解题思路:
两层for循环,第一层遍历下标为0到(n-indexDifference),第二层遍历(i+indexDifference)到n,这样就可以保证下标差一定是满足第一个条件的,我们只需要去判断第二个条件是否成立即可。
代码:
class Solution:def findIndices(self, nums: list[int], indexDifference: int, valueDifference: int) -> list[int]:n=len(nums)for i in range(n-indexDifference):for j in range(i+indexDifference,n):if abs(nums[i]-nums[j])>=valueDifference:return [i,j]return [-1,-1]
这段代码定义了一个名为Solution的类,其中包含一个名为findIndices的方法。该方法接受一个整数列表nums,一个整数indexDifference和一个整数valueDifference作为输入。它返回一个包含两个索引i和j的列表,这两个索引位置上的数字的绝对差至少为valueDifference,并且它们的索引距离大于等于indexDifference。
如果存在这样的索引位置,则该方法返回这些索引[i, j],否则返回[-1, -1]。
这个实现循环遍历了指定范围内的索引i和j,并检查是否满足绝对差条件。如果满足条件,方法会返回这两个索引。这段代码存当找到满足条件的第一对索引时就返回,符合题意。