给你一个长度为 n
的整数数组 nums
,以及一个正整数 k
。
将这个数组划分为一个或多个长度为 3
的子数组,并满足以下条件:
nums
中的 每个 元素都必须 恰好 存在于某个子数组中。- 子数组中 任意 两个元素的差必须小于或等于
k
。
返回一个 二维数组 ,包含所有的子数组。如果不可能满足条件,就返回一个空数组。如果有多个答案,返回 任意一个 即可。
示例 1:
输入:nums = [1,3,4,8,7,9,3,5,1], k = 2 输出:[[1,1,3],[3,4,5],[7,8,9]] 解释:可以将数组划分为以下子数组:[1,1,3],[3,4,5] 和 [7,8,9] 。 每个子数组中任意两个元素的差都小于或等于 2 。 注意,元素的顺序并不重要。
示例 2:
输入:nums = [1,3,3,2,7,3], k = 3 输出:[] 解释:无法划分数组满足所有条件。
提示:
n == nums.length
1 <= n <= 105
n
是3
的倍数1 <= nums[i] <= 105
1 <= k <= 105
思路:我们先将所给的数组排序,排序以后三个三个地填入二维数组的每一行,然后检查每一行的第三个元素和第一个元素之差,因为它是每一行中差最大的,如果它小于等于K,那么其他元素的差也小于等于K,否则,直接返回空数组.
class Solution {
public int[][] divideArray(int[] nums, int k) {
Arrays.sort(nums);//先排序
int rows = nums.length;
int Row = rows/3;
int cnt = 0;//记录行
int[][]res = new int[Row][3];
for(int i=1;i<=nums.length;i++){
res[cnt][(i-1)%3] = nums[i-1];
if(i%3==0){
cnt++;
}//填充
}
int[][]array = new int[0][0];
for(int i=0;i<Row;i++){
if(res[i][2]-res[i][0]>k){
return array;
}
}
return res;
}
}