LeetCode - 1630 等差子数组【排序】

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;
    }
}

 运行结果