[SCOI2005] BZOJ 1087 互不侵犯King - 状压dp

121 篇文章 0 订阅
40 篇文章 0 订阅

传送门

题解:

首先求出哪些行的状态是合法的,即没有相邻的棋子。

发现合法的行状态数量只有90,没有2^9=512这么多。

然后枚举相邻两行,预处理能否转移。

然后记q[i]表示第i种合法的行方案是q[i],cnt[i]表示q[i]放了多少个棋子

然后记dp[i][j][k]表示前i行第i行状态为j棋子个数是k的方案数

转移枚举哪些状态可以转移(刚才预处理过的)

然后……就没有然后了。

这提示我们状压dp有些时候可以通过预处理使得复杂度得到玄学优化

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#define N 10
#define MAXT 90
#define lint long long
using namespace std;
int n,all,q[MAXT],top,cnt[MAXT];
lint dp[N][N*N][MAXT];
vector<int> g[MAXT];
inline int getv(int x,int p)
{
    return (x&(1<<(p-1)))>>(p-1);
}
inline int output(int x)
{
    for(int i=1;i<=n;i++)
        cout<<getv(x,i);
    return 0;
}
inline bool check(int s)
{
    for(int i=1;i<n;i++)
        if(getv(s,i)&getv(s,i+1))
            return false;
    return true;
}
inline bool check(int s,int t)
{
    for(int i=1;i<=n;i++)
    {
        int v=getv(s,i);
        if(!v) continue;
        if(i-1&&getv(t,i-1)) return false;
        if(getv(t,i)) return false;
        if(i+1<=n&&getv(t,i+1)) return false;
    }
    return true;
}
inline int calc(int s)
{
    int cnt=0;
    for(int i=1;i<=n;i++)
        cnt+=getv(s,i);
    return cnt;
}
int main()
{
    int k;scanf("%d%d",&n,&k);
    all=(1<<n)-1;
    for(int i=0;i<=all;i++)
        if(check(i)) q[++top]=i;
    for(int i=1;i<=top;i++)
        cnt[i]=calc(q[i]);
    for(int i=1;i<=top;i++)
        for(int j=1;j<=top;j++)
            if(check(q[i],q[j]))
                g[j].push_back(i);
    dp[0][0][1]=1LL;
    for(int i=1;i<=n;i++)
        for(int c=0;c<=k;c++)
            for(int j=1;j<=top;j++)
            {
                if(cnt[j]>c) continue;
                for(int k=g[j].size()-1;k>=0;k--)
                    dp[i][c][j]+=dp[i-1][c-cnt[j]][g[j][k]];
            }
    lint ans=0;
    for(int i=1;i<=top;i++)
        ans+=dp[n][k][i];
    printf("%lld\n",ans);return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值