[SGU Little Queens]

[关键字]：搜索 位运算

[题目大意]：在一个n*n的棋盘里放k个王后，问使它们不能互相攻击的摆放方案有多少种。

//==================================================================

[分析]：就是n皇后问题的稍微变形，但要加上位运算优化。具体可以看matrix67神牛的博文。

[代码]：

View Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

int n,m,goal;
int ans,sum;

void DFS(int row,int ld,int rd,int sum,int now)
{
     if (now>n) return;
     if (sum<m)
     {                   int pos=goal&(~(row|ld|rd));
                         //printf("%d %d %d %d\n",pos,row,ld,rd);
                         while (pos>0)
                         {
                               int p=pos&(-pos);
                               pos-=p;
                               DFS(row+p,(ld+p)<<1,(rd+p)>>1,sum+1,now+1);
                         }
                         DFS(row,ld<<1,rd>>1,sum,now+1);
     }
     else ++ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    goal=(1<<n)-1;
    //printf("%d\n",goal);
    DFS(0,0,0,0,0);
    printf("%d\n",ans);
    system("pause");
    return 0;
}