题意:n*n的棋盘上放k个国王,一个国王相邻的八个格子不能有别的国王,求方案总数
题解:根据数据范围可以想到状压dp,f[i][row][num]表示第row行在第i种状态下(预处理所有可行状态),1~row行已经放了num个国王的方案数,最终结果为f[1~cnt][n][k]之和
细看题面没说结果数据范围,果断判一波longlong居然1A哈哈哈(露出不要脸的笑容)!♪(^∀^●)ノ
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
int n,k;
ll f[260][10][90];//f[i][row][num]
int bt[260];
int st[260],cnt;
inline int count(int x) {
int ret=0;
while (x) {
++ret;
x-=x&-x;
}
return ret;
}
inline bool ok(int s1,int s2) {
return !(s1&(s2<<1))&&!(s1&s2)&&!((s1<<1)&s2);
}
int main() {
cin>>n>>k;
for (int i=0;i<(1<<n);++i)
if (!(i&(i<<1))) {
++cnt;
st[cnt]=i;
bt[cnt]=count(i);
}
for (int i=1;i<=cnt;++i)
f[i][1][bt[i]]=1;
for (int row=2;row<=n;++row)
for (int cur=1;cur<=cnt;++cur)
for (int pre=1;pre<=cnt;++pre)
if (ok(st[cur],st[pre])) {
for (int num=bt[cur];num<=k;++num)
f[cur][row][num]+=f[pre][row-1][num-bt[cur]];
}
ll ans=0;
for (int i=1;i<=cnt;++i)
ans+=f[i][n][k];
cout<<ans<<endl;
return 0;
}