【SGU 448】Controlled Tournament（状态压缩动态规划）

最新推荐文章于 2024-04-07 22:08:13 发布

原创最新推荐文章于 2024-04-07 22:08:13 发布 · 265 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#SGU

状态压缩动态规划同时被 2 个专栏收录

5 篇文章

订阅专栏

子集枚举

1 篇文章

订阅专栏

本文介绍了一道经典的算法竞赛题目，通过状态压缩动态规划的方法求解特定条件下比赛获胜方案的数量。利用记忆化搜索减少重复计算，提高了算法效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

【SGU 448】Controlled Tournament

题目大意

给定比赛人员个数 $n$ ，你希望赢的人的编号 $m$ 以及一张 $n\times n$ 的胜负表，第 $i$ 行第 $j$ 列为 $1$ ，代表 $i$ 能赢 $j$ 。
赛制为淘汰赛，求 $m$ 最后能赢，且总比赛树的最小高度时，一共有多少种可能方案。
$n\le 16$ 。

题解

看到 $n\le 16$ ，果断使用状态压缩动态规划。
$dp[i][height][mask]$ 表示已比赛完的人的状态为 $mask$ ，其中第 $i$ 个人胜利了，且比赛树的高度为 $height$ 。
状态转移：枚举子集，当子集 $submask$ 中包含 $i$ ， $i$ 能胜过补集中的 $j$ 的时候，可以转移。
即： $dp[i][height][mask]=\sum {dp[i][height-1][submask]\times dp[j][height-1][mask-submask]}$ 。
具体细节见代码。

代码no.1

#include <cstdio>
#include <cstring>
#define mxn 16
#define lgn 4
#define rep(i,l,r) for(int i=(l);i<(r);i++)
typedef long long ll;
int n,m,hi,a[mxn][mxn],cnt[1<<mxn];
int dp[mxn][lgn|1][1<<mxn];
int search(int i,int k,int msk) {
    if(~dp[i][k][msk]) return dp[i][k][msk];
    if(msk==1<<i)   return dp[i][k][msk]=1;
    if(1<<k<cnt[msk])   return dp[i][k][msk]=0;
    dp[i][k][msk]=0;
    for(int msk0=msk&(msk-1);msk0;msk0=msk&(msk0-1))
        if(msk0&(1<<i)) {
            int msk1=msk^msk0;
            rep(j,0,n)  if(a[i][j]&&(msk1&(1<<j)))
                dp[i][k][msk]+=search(i,k-1,msk0)*search(j,k-1,msk1);
        }
    return dp[i][k][msk];
}
int main() {
    memset(dp,-1,sizeof(dp));
    scanf("%d%d",&n,&m),m--;
    rep(i,1,1<<n)
        cnt[i]=cnt[i>>1]+(i&1);
    rep(i,0,n)  rep(j,0,n)
        scanf("%1d",a[i]+j);
    for(;1<<hi<n;hi++);
    search(m,hi,(1<<n)-1);
    if(dp[m][hi][(1<<n)-1]==-1)
        dp[m][hi][(1<<n)-1]=0;
    printf("%d\n",dp[m][hi][(1<<n)-1]);
    return 0;
}

代码no.2（加了一些小优化，然并卵）

#include <cstdio>
#include <cstring>
#define mxn 16
#define lgn 4
#define rep(i,l,r) for(int i=(l);i<(r);i++)
typedef long long ll;
int n,m,hi,a[mxn][mxn],cnt[1<<mxn];
int dp[mxn][lgn|1][1<<mxn];
int search(int i,int k,int msk) {
    if(~dp[i][k][msk]) return dp[i][k][msk];
    if(msk==1<<i)   return dp[i][k][msk]=1;
    if(1<<k<cnt[msk])   return dp[i][k][msk]=0;
    if(!(msk&(1<<i)))   return dp[i][k][msk]=0;
    dp[i][k][msk]=0;
    int msk0,msk1,tmsk=msk^(1<<i);
    for(int tmsk0=tmsk&(tmsk-1);tmsk0;tmsk0=tmsk&(tmsk0-1)) {
        msk0=tmsk0|(1<<i);
        msk1=msk^msk0;
        rep(j,0,n)  if(a[i][j]&&(msk1&(1<<j)))
            dp[i][k][msk]+=search(i,k-1,msk0)*search(j,k-1,msk1);
    }
    msk0=1<<i;
    msk1=msk^msk0;
    rep(j,0,n)  if(a[i][j]&&(msk1&(1<<j)))
        dp[i][k][msk]+=search(i,k-1,msk0)*search(j,k-1,msk1);
    return dp[i][k][msk];
}
int main() {
    memset(dp,-1,sizeof(dp));
    scanf("%d%d",&n,&m),m--;
    rep(i,1,1<<n)
        cnt[i]=cnt[i>>1]+(i&1);
    rep(i,0,n)  rep(j,0,n)
        scanf("%1d",a[i]+j);
    for(;1<<hi<n;hi++);
    search(m,hi,(1<<n)-1);
    if(dp[m][hi][(1<<n)-1]==-1)
        dp[m][hi][(1<<n)-1]=0;
    printf("%d\n",dp[m][hi][(1<<n)-1]);
    return 0;
}