Description
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上
左下右上右下八个方向上附近的各一个格子,共8个格子。
Input
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
Output
方案数。
Sample Input
3 2
Sample Output
16
看了看题目和数据范围,果断状态压缩Dp。
复杂度:
O(22n∗n∗k)
感觉很大
我们考虑一些优化:(常数上的)
(1)一些有相邻国王的状态毫无意义,提前删掉,有用的状态放在一个数组里
(2)两行相邻的没有其它的牵制,我们可以提前用邻接表把它们搞成一张图
这省了很多无用的时间,于是很轻松的就过了
常数这样基本会小于1
所以
代码:
/**************************************************************
Problem: 1087
User: waz
Language: C++
Result: Accepted
Time:32 ms
Memory:17024 kb
****************************************************************/
#include <cstdio>
#define Rg register
struct edge
{
int To ,Next ;
}
E[1 << 20] ;
int Last[1 << 10] ,Et ;
bool Flag[1 << 10] ;
inline void Addedge(Rg int a ,Rg int b)
{
E[++Et] = (edge){b ,Last[a]} ,Last[a] = Et ;
}
inline void Add(Rg int a ,Rg int b)
{
Addedge(a ,b) ;
Addedge(b ,a) ;
}
inline bool Check(Rg int S1 ,Rg int S2)
{
for(Rg int i = 0 ;i < 9 ;i++)
{
if((S1 & (1 << i)))
{
if(i > 0) if((S2 & (1 << i - 1))) return 0 ;
if((S2 & (1 << i))) return 0 ;
if(i < 9)if((S2 & (1 << i + 1))) return 0 ;
}
}
return 1 ;
}
inline bool Wrong(Rg int S)
{
for(Rg int i = 0 ;i < 8 ;i++)
if((S & (1 << i)) && (S & (1 << i + 1))) return 1 ;
return 0 ;
}
inline void Init()
{
for(Rg int i = 0 ;i < 512 ;i++)
Flag[i] = Wrong(i) ;
for(Rg int i = 0 ;i < 512 ;i++)
{
if(Flag[i])
continue ;
for(Rg int j = 0 ;j < 512 ;j++)
{
if(Flag[j]) continue ;
if(Check(i ,j)) Addedge(i ,j) ;
}
}
}
long long Dp[10][100][1 << 10] ,Ans ;
#define F(i ,j ,k) Dp[i][j][k]
int N ,K ;
inline int Count(Rg int S)
{
Rg int n = 0 ;
for(Rg int i = 0 ;i < 9 ;i++)
if(S & (1 << i)) n++ ;
return n ;
}
int Cnt[1 << 10] ;
int main()
{
Init() ;
F(0 ,0 ,0) = 1 ;
scanf("%d %d" ,&N ,&K) ;
Rg int Limit = 1 << N ;//Limit-- ;
for(Rg int i = 1 ;i < Limit ;i++)
Cnt[i] = Count(i) ;
for(Rg int i = 1 ;i <= N ;i++)
for(Rg int j = 0 ;j <= K ;j++)
{
for(Rg int k = 0 ;k < Limit ;k++)
{
Rg int J = j - Cnt[k] ;
if(Flag[k] || J < 0) continue ;
for(Rg int v = Last[k] ;v ;v = E[v].Next)
F(i ,j ,k) += F(i - 1 ,J ,E[v].To) ;
}
}
for(Rg int i = 0 ;i < Limit ;i++) Ans += F(N ,K ,i) ;
printf("%lld\n" ,Ans) ;
return 0 ;
}