洛谷p1896互不侵犯

原題

先思考怎麼表示狀態,用f[i][j]表示第i行放j個顯然不行,那麼需要一個可以表示出狀態的變量,由於m很小,可以用二進制表示每個位置是否放置,因此f[i][j][z]表示第i行放置狀態j時共放置z個國王的方案數,本行只收上一行的狀態。

我用的是狀壓dfs,和狀壓dp想法一樣,而且比較好實現,另外複雜度相同。

狀壓dfs(((2^n)^2)*k*n)最壞情況1e9,可以ac。

x表示行,y是列(從0),zhuang是你所求的上一行的狀態,net是本行的狀態(即本行的zhuang),now表示現有的國王總數,start存初值,l保證國王不相鄰(其實可以直接跳兩步,就不需要l了)

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,k,cnt,f[11][1125][101];
int dfs(int x,int y,int zhuang,int net,int now,int start,int l)
{
    if(y==n)
    {
        f[x+1][net][now]+=start;
        return 0;
    }
    dfs(x,y+1,zhuang,net,now,start,l);
    if(now<k&&(((1<<y)&zhuang)==0)&&y-l>1)
    {
        net=net|(1<<y);
        if(y>0)
        net=net|(1<<y-1);
        if(y<n-1)
        net=net|(1<<y+1);
        dfs(x,y+1,zhuang,net,now+1,start,y);
    }
}
int main()
{
    cin>>n>>k;
    memset(f,0,sizeof(f));
    f[1][0][0]=1;
    for(int i=1;i<=n;++i)
     for(int j=0;j<(1<<n);++j)
      for(int z=0;z<=k;++z)
       if(f[i][j][z])
       dfs(i,0,j,0,z,f[i][j][z],-2);
    int ans=0;
    for(int i=0;i<(1<<n);++i)
    ans+=f[n+1][i][k];
    cout<<ans;
    return 0;   
} 

注意:洛谷的數據更改,需要將int改爲長整型

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章