YbtOJ 状压dp E. 1.最优组队

Mvrk

已于 2022-11-13 04:00:32 修改

阅读量100

点赞数

分类专栏： YBTOJ 文章标签：算法 c++ 数据结构

于 2022-11-07 15:56:44 首次发布

本文链接：https://blog.csdn.net/never_in_time/article/details/127731952

版权

YBTOJ 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

显然设置 $f_i$ 为状态为 $i$ 的情况下最大和谐度。枚举各个子集 $i$ 再枚举 $i$ 的各个子集 $j$ 进行转移，复杂度 $O(2^{2n})$ ， $n\le16$ ，显然TLE。

#include<bits/stdc++.h>//注意位运算的优先级,加括号控制
using namespace std;
const int maxn=1e5+10;
int f[maxn],n,top;
signed main()
{
    cin>>n;
    top=(1<<n)-1;
    for(int i=1;i<=top;i++) cin>>f[i];
    for(int i=1;i<=top;i++)
        for(int j=1;j<=top;j++)
            if(i!=j&&((i|j)==i)) f[i]=max(f[i],f[j]+f[i^j]);
    cout<<f[top]<<endl;
    return 0;
}

所以考虑只枚举 $i$ 的子集进行优化，

for(int j=(i-1)&i;j;j=(j-1)&i)

举个例子

那么复杂度是

$\sum_{N\supseteq I }^{} \sum_{I\supseteq J}^{}1=\sum_{N\supseteq I }^{}2^{|I|}$

由于 $N$ 中大小为 $i$ 的子集个数是 $\begin{pmatrix}n\\I\end{pmatrix}$ ，所以：

$=\sum_{i=1}^{n}\begin{pmatrix}n\\I\end{pmatrix}\times 2^i$

用二项式定理计算：

$=\sum_{i=1}^{n}\begin{pmatrix}n\\I\end{pmatrix}\times 2^i\times1^{n-1}$

$1+2)^n−1$

所以时间复杂度为 $O(n^3)$

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int f[maxn],n,top;
signed main()
{
    cin>>n;
    top=(1<<n)-1;
    for(int i=1;i<=(1<<n)-1;i++) cin>>f[i];
    for(int i=1;i<=top;i++)
        for(int j=(i-1)&i;j;j=(j-1)&i)
            if(i!=j&&((i|j)==i)) f[i]=max(f[i],f[j]+f[i^j]);
    cout<<f[top]<<endl;
    return 0;
}