1681. 最小不兼容性

发布时间 2023-06-29 14:57:44作者: Tianyiya

给你一个整数数组 nums​​​ 和一个整数 k 。你需要将这个数组划分到 k 个相同大小的子集中,使得同一个子集里面没有两个相同的元素。

一个子集的 不兼容性 是该子集里面最大值和最小值的差。

请你返回将数组分成 k 个子集后,各子集 不兼容性 的 和 的 最小值 ,如果无法分成分成 k 个子集,返回 -1 。

子集的定义是数组中一些数字的集合,对数字顺序没有要求。

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-incompatibility
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

import java.util.*;

class Solution {
    public int minimumIncompatibility(int[] nums, int k) {
        if (nums == null || nums.length % k != 0) {
            return -1;
        }
        int group = nums.length / k;
        int inf = Integer.MAX_VALUE;
        int n = nums.length;
        int limit = 1 << n;
        int[] dp = new int[limit];
        Arrays.fill(dp, inf);
        dp[0] = 0;
        for (int mask = 1; mask < limit; mask++) {
            if (Integer.bitCount(mask) != group) {
                continue;
            }
            int mx = Integer.MIN_VALUE, mi = Integer.MAX_VALUE;
            Set<Integer> cntSet = new HashSet<>();
            int x = mask;
            while (x != 0) {
                int lowbit = x & (-x);
                int index = Integer.numberOfTrailingZeros(lowbit);
                if (cntSet.contains(nums[index])) {
                    break;
                }
                cntSet.add(nums[index]);
                mx = Math.max(mx, nums[index]);
                mi = Math.min(mi, nums[index]);
                x -= lowbit;
            }
            if (cntSet.size() == group) {
                dp[mask] = mx - mi;
            }
        }
        for (int mask = 1; mask < limit; mask++) {
            if (dp[mask] == inf) {
                continue;
            }
            int noSelected = (limit - 1) ^ mask;
            for (int sub = noSelected; sub != 0; sub = (sub - 1) & noSelected) {
                if (dp[sub] != inf) {
                    dp[mask | sub] = Math.min(dp[mask | sub], dp[mask] + dp[sub]);
                }
            }
        }
        return dp[limit - 1] != inf ? dp[limit - 1] : -1;
    }
}