首页 > 代码库 > sgu-223 Little Kings

sgu-223 Little Kings

 题目大意:

给你一个n*n的棋盘(n<=10),然后要你在上面放k(k<=n*n)国王(国际象棋中的国王,可以往周围的8个方向走)并使得他们互不攻击,问有多少种方法。


解题思路:

很裸的一道状压DP,用一个数表示二进制,二进制位上为1则表示放,否则表示不放,然后转移用dfs就行了。


AC代码:

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
#define MAX(a,b) ((a)>(b)?(a):(b))
#define MIN(a,b) ((a)>(b)?(b):(a))

using namespace std;
int n,m;
long long f[20][1100][110]={{{0}}};
int g[2]={0};
int gg[2][20]={{0}};
long long ans=0;
int sum=0;

void dfs(int i,int j,int P,int Q)
{
	if(j==n+1 && i==2)
	{
		f[P][g[1]][Q]+=f[P-1][g[0]][Q-sum];
		return;
	}
	if(j==n+1)
	{
		i=2;
		j=1;
	}
	dfs(i,j+1,P,Q);
	
	if(gg[i-1][j-1]!=0)
		return;
	if(i>1 && (gg[i-2][j-1]!=0 || gg[i-2][j]!=0 || gg[i-2][j+1]!=0))
		return;
	
	gg[i-1][j]=1;
	g[i-1]+=(1<<(j-1));
	if(i==2)
		sum++;
	if(sum<=Q)
		dfs(i,j+1,P,Q);
	if(i==2)
		sum--;
	g[i-1]-=(1<<(j-1));
	gg[i-1][j]=0;
	
	return;
}

int main()
{
	scanf("%d%d",&n,&m);
	
	f[0][0][0]=1;
	for(int i=1;i<=n;i++)
		for(int j=0;j<=m;j++)
			dfs(1,1,i,j);
	
	for(int i=0;i<=1023;i++)
		ans+=f[n][i][m];
	
	cout<<ans<<endl;
	return 0;
}


sgu-223 Little Kings