最小不兼容性

给你一个整数数组 nums​​​ 和一个整数 k 。你需要将这个数组划分到 k 个相同大小的子集中,使得同一个子集里面没有两个相同的元素。
一个子集的不兼容性是该子集里面最大值和最小值的差。
请你返回将数组分成 k 个子集后,各子集 不兼容性的和的最小值 ,如果无法分成分成 k 个子集,返回 -1 。

1. 状态压缩 + 动态规划

class Solution {
public:
    int minimumIncompatibility(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> dp(1<<n,INT_MAX); //记录对应状态最小值
        int group = n/k;//每个集合数量

        //状态边界计算,计算只有一个组的不兼容性
        for(int mask=0;mask<(1<<n);mask++){//遍历所有状态
            if(__builtin_popcount(mask)!=group) continue;//剔除不满足条件的状态,这里只计算一组的不兼容性

            int mn = INT_MAX; int mx = INT_MIN;
            vector<bool> vis(n);
            int cnt = 0;
            for(int i=0;i<n;i++){//遍历每个数
                if(mask&(1<<i)){//如果该状态下该数存在
                    if(vis[nums[i]]) break; //该状态下存在重复数,跳出
                    vis[nums[i]] = true;//标记该数
                    //求该组的最大值和最小值
                    mn = min(mn,nums[i]);
                    mx = max(mx,nums[i]);
                    cnt++;//组中数加一
                }
            }
            if(cnt!=group)  continue;
            dp[mask] = mx - mn;//记录该状态的不兼容性
        }

        for(int mask=0;mask<(1<<n);mask++){//遍历所有状态
            if(dp[mask]==INT_MAX) continue;  //只从已经记录的状态转移计算下一个状态
            int sub = ((1<<n)-1)^mask; //求mask的补集
            for(int cur=sub;cur>0;cur = (cur-1)&sub)
                if(dp[cur]!=INT_MAX) dp[mask|cur] = min(dp[mask|cur],dp[mask] + dp[cur]);
        }
        return dp[(1<<n)-1]==INT_MAX?-1:dp[(1<<n)-1];
    }
};

2. 状态压缩 + 记忆化搜索

class Solution {
public:
    int minimumIncompatibility(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> dp(1<<n,INT_MAX/2); //记录对应状态最小值
        int group = n/k;//每个集合数量
        dp[0] = 0;
        function<int(int)> f = [&](int mask) ->int{
            if(dp[mask]==-1) return INT_MAX/2; //不合格的组合,但表示已经计算过
            if(dp[mask]!=INT_MAX/2) return dp[mask]; //记忆化搜索

            int cnt = __builtin_popcount(mask);
            if(cnt==group){
                int mn = INT_MAX; int mx = INT_MIN;
                vector<bool> vis(n);
                for(int i=0;i<n;i++){//遍历每个数
                    if(mask&(1<<i)){//如果该状态下该数存在
                        if(vis[nums[i]]){
                            dp[mask] = -1;
                            return INT_MAX/2;
                        }
                        vis[nums[i]] = true;//标记该数
                        //求该组的最大值和最小值
                        mn = min(mn,nums[i]);
                        mx = max(mx,nums[i]);
                }
            }
            dp[mask] = mx - mn;//记录该状态的不兼容性
            return dp[mask];
            }
 
            //对于一般情况,进行递归回溯
            for(int cur=mask;cur>0;cur = (cur-1)&mask){//枚举所有子集
                if(__builtin_popcount(cur)==group){
                    int sub = mask^cur; //求cur的补集
                    dp[mask] = min(dp[mask], f(sub) + f(cur));
                }
            }
            if(dp[mask]==INT_MAX/2){ //已经计算过,但不合格
                 dp[mask] = -1;
                 return INT_MAX/2;
            }
            return dp[mask];
        };
        return f((1<<n)-1)==INT_MAX/2?-1:f((1<<n)-1);
    }
};
posted @ 2023-06-30 21:04  失控D大白兔  阅读(24)  评论(0编辑  收藏  举报