JAVA程序设计：分配重复整数（LeetCode：1655）

给你一个长度为 n 的整数数组 nums ，这个数组中至多有 50 个不同的值。同时你有 m 个顾客的订单 quantity ，其中，整数 quantity[i] 是第 i 位顾客订单的数目。请你判断是否能将 nums 中的整数分配给这些顾客，且满足：

第 i 位顾客 恰好 有 quantity[i] 个整数。
第 i 位顾客拿到的整数都是 相同的 。
每位顾客都满足上述两个要求。
如果你可以分配 nums 中的整数满足上面的要求，那么请返回 true ，否则返回 false 。

 

示例 1：

输入：nums = [1,2,3,4], quantity = [2]
输出：false
解释：第 0 位顾客没办法得到两个相同的整数。
示例 2：

输入：nums = [1,2,3,3], quantity = [2]
输出：true
解释：第 0 位顾客得到 [3,3] 。整数 [1,2] 都没有被使用。
示例 3：

输入：nums = [1,1,2,2], quantity = [2,2]
输出：true
解释：第 0 位顾客得到 [1,1] ，第 1 位顾客得到 [2,2] 。
示例 4：

输入：nums = [1,1,2,3], quantity = [2,2]
输出：false
解释：尽管第 0 位顾客可以得到 [1,1] ，第 1 位顾客没法得到 2 个一样的整数。
示例 5：

输入：nums = [1,1,1,1,1], quantity = [2,3]
输出：true
解释：第 0 位顾客得到 [1,1] ，第 1 位顾客得到 [1,1,1] 。
 

提示：

n == nums.length
1 <= n <= 105
1 <= nums[i] <= 1000
m == quantity.length
1 <= m <= 10
1 <= quantity[i] <= 105
nums 中至多有 50 个不同的数字。

思路：状压dp，将m个顾客表示为m位二进制，每一位表示第i个顾客的状态，定义dp[i]表示m个顾客状态为i时是否能够满足要求。

class Solution {

    private static final int len = 1 << 10;

    public boolean canDistribute(int[] nums, int[] quantity) {

        int n = quantity.length;
        int[] c = new int[1005];
        int[] sum = new int[len];
        boolean[] dp = new boolean[len];
        boolean[] temp = new boolean[len];
        List<Integer> list = new ArrayList<>();

        for (int x : nums) c[x]++;
        for (int i = 0; i <= 1000; i++) {
            if (c[i] > 0)
                list.add(c[i]);
        }
        dp[0] = true;
        for (int i = 1; i < (1 << n); i++)
            for (int j = 0; j < n; j++) {
                if (((i >> j) & 1) != 0)
                    sum[i] += quantity[j];
            }
        int m = list.size();
        for (int k = 0; k < m; k++) {
            for (int i = 0; i < (1 << n); i++)
                temp[i] = dp[i];
            for (int i = 0; i < (1 << n); i++) {
                if (temp[i]) continue;
                if (sum[i] <= list.get(k)) dp[i] = true;
                for (int j = i; j != 0; j = (j - 1) & i) {
                    if (sum[i ^ j] <= list.get(k))
                        dp[i] |= temp[j];
                }
            }
        }
        return dp[(1 << n) - 1];
    }
}