1630. 等差子数组
示例1:
输入:nums = [4,6,5,9,3,7], l = [0,0,2], r = [2,3,5]
输出:[true,false,true]
解释:
第 0 个查询,对应子数组 [4,6,5] 。可以重新排列为等差数列 [6,5,4] 。
第 1 个查询,对应子数组 [4,6,5,9] 。无法重新排列形成等差数列。
第 2 个查询,对应子数组 [5,9,3,7] 。可以重新排列为等差数列 [3,5,7,9] 。
示例2:
输入:nums = [-12,-9,-3,-12,-6,15,20,-25,-20,-15,-10], l = [0,1,6,4,8,7], r = [4,4,9,7,9,10]
输出:[false,true,false,false,true,true]
提示
- n == nums.length
- m == l.length
- m == r.length
- 2 <= n <= 500
- 1 <= m <= 500
- 0 <= l[i] < r[i] < n
- - <= nums[i] <=
思路
题意:判断nums数组中,从l[i]到r[i]范围内的子数组是否为等差数列。
方法:根据题意,遍历每个范围[l[i], r[i]],并将原数组nums的元素赋值给新数组array。若array数组大小为2,说明仅有两个元素,肯定为等差数列,结果为true。否则,将array排序,依次判断元素差值,若差值不等结果为false。
代码
class Solution {
public List<Boolean> checkArithmeticSubarrays(int[] nums, int[] l, int[] r) {
List<Boolean> result = new ArrayList<>();
for (int i = 0; i < l.length; i++) {
int k = 0;
int[] array = new int[r[i] - l[i] + 1];
if (array.length == 2) {
result.add(true);
continue;
}
for (int j = l[i]; j <= r[i]; j++)
array[k++] = nums[j];
Arrays.sort(array);
int num = array[1] - array[0];
for (int ii = 2; ii < array.length; ii++) {
if (array[ii] - array[ii - 1] != num) {
result.add(false);
break;
}
else if (ii == array.length - 1)
result.add(true);
}
}
return result;
}
}