题解 | #牛群的特殊分组#
牛群的特殊分组
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); } } } }