LeetCode 1815 状压dp

 

一句话: 将给定数组排序使得对k取余为0的前缀和数量最多。

给定的数组长度n<=30,k<=9

将所有数对k取余后转化为对所有数分组使得 sum%k==0 的组的数量最多

我们定义这种组为优秀组，容易想到定义状态s 二进制第i位为1表示选了第i位数的优秀组最多数目

--> 时间复杂度 O(2 ^ n) 考虑优化 

因为k很小 会有很多重复的数 最终答案与这些重复的数的选取顺序无关 所以状态中我们只需保留每种数当前取了多少个即可 

总状态数将会被减少至 mul(hah[i]+1)，hah[i]表示余数为i的数的个数 可以证明该状态数 <= (n/k)^k 即每种数的数量相同时积最大

如何表示该状态?

--> 将每种数按维度分开即可

 最后dp方程: dp[new_s]=max(dp[new_s], dp[s]+(状态s里所有数的和%k==0))

#include <iostream>
#include <vector>
using namespace std;
const int N=3005; 
int a[35],dp[N],hah[10];
vector<int>to;

int main()
{
	int n,k; scanf("%d%d", &n, &k);
	for(int i=1; i<=n; i++) scanf("%d", &a[i]);
	int ans=0;
    for(int i=1; i<=n; i++) // 若可以两两组合成优秀组则可以提前删去 
    {
        int v=a[i]%k;
        if(hah[k-v])
        {
            hah[k-v]--;
            ans++;
        }
        else
            hah[v]++;
    }
    ans+=hah[0]; // 自身成为优秀组 
    int tot=1;
    for(int i=1; i<k; i++) // 统计总状态数 
        if(hah[i])
        {
            tot*=(hah[i]+1); // 该位数可以取也可以不取 因此为hah[i]+1
            to.push_back(i);
        }
    for(int s=0; s<tot; s++)
    {
        int sum=0;
        for(int i=0,base=1; i<to.size(); i++) // 先计算当前状态的和 base用来降维 
            sum+=to[i]*((s/base)%(hah[to[i]]+1)),base*=(hah[to[i]]+1);
        sum%=k;

        for(int i=0,base=1; i<to.size(); i++)
        {
            int cnt=(s/base)%(hah[to[i]]+1); // to[i]当前的数量
            if(cnt!=hah[to[i]]) // 还可以加入to[i]
            {
                int new_s=s+base;
                dp[new_s]=max(dp[new_s], dp[s]+(sum==0));
            }
            base*=hah[to[i]]+1;
        }
    }
    cout << ans+dp[tot-1] << '\n';
	return 0;
}

总时间复杂度O((n/k)^k) 由于做了优化 因此实际k的值会减半