public boolean canPartitionKSubsets(int[] nums, int k) {
int n=nums.length;
Arrays.sort(nums);
int sum=0;
for(int i=0;i<n;i++){
sum+=nums[i];
}
if(sum%k!=0) return false;//判断和是否为k的倍数
int tar=sum/k;
boolean[] mark=new boolean[n];//是否已选择某个数
return dfs(nums,mark,k,tar,0,0);//回溯,适用于无法确定循环次数的遍历
}
boolean dfs(int[] nums,boolean[] mark,int cnt,int tar,int cur,int idx){
//完成一组后,从头开始寻找下一组,重置当前和
if(cur==tar) {
cnt--;
cur = 0;
idx=0;
}
if(cnt==0)//已经分为k组
return true;
//从下一个数开始寻找可以加入当前组的数
for(int i=idx;i<nums.length;i++){
if(mark[i])
continue;
if(nums[i]+cur>tar)//当前数过大,则后面都不满足
break;
mark[i]=true;
if(dfs(nums,mark,cnt,tar,cur+nums[i],i+1))//挑选下一个数
return true;
mark[i]=false;//回溯状态
if (cur == 0) return false;//如果结果为true,当前组第一个数不管选哪个都应为true,因为总能找到一个它所在的组,否则为false
}
return false;
}
回溯算法leetcode.698
最新推荐文章于 2024-07-25 11:44:32 发布