题解 | #牛群的特殊分组#
牛群的特殊分组
https://www.nowcoder.com/practice/ba62e31d82a04c9f8d0b0591ab8f2f06
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型一维数组
* @return int整型二维数组
*/
public int[][] special_groups (int[] nums) {
// write code here
List<List<Integer>> res=new ArrayList<>();
List<Integer> temp=new ArrayList<>();
Arrays.sort(nums);
boolean[] flag=new boolean[nums.length];
backtrace(res,temp,flag,nums,0);
int[][] ans=new int[res.size()][];
for(int i=0;i<res.size();i++){
//此处要申请空间,否则会报错
ans[i]=new int[res.get(i).size()];
for(int j=0;j<res.get(i).size();j++){
ans[i][j]=res.get(i).get(j);
}
}
return ans;
}
public void backtrace(List<List<Integer>> res,List<Integer> temp,boolean[] flag,int[] nums,int start){
if(temp.size()>0){
res.add(new ArrayList<>(temp));
//这个地方不要return
}
for(int i=start;i<nums.length;i++){
if(i>0&&nums[i-1]==nums[i]&&flag[i-1]==false) continue;
if(flag[i]==false){
temp.add(nums[i]);
flag[i]=true;
backtrace(res,temp,flag,nums,i+1);
flag[i]=false;
temp.remove(temp.size()-1);
}
}
}
}
