济南学习 Day 5 T3 am

【题目描述】

   众所不知,rly现在不会玩国际象棋。但是,作为一个OIer,rly当然做过八皇后问题.在这里再啰嗦几句,皇后可以攻击到同行同列同对角线,在

n*n的棋盘中,摆放n个皇后使它们互相不能攻击到,求不同的解的数量,这就是经典的n皇后问题。现在问题推广n皇后问题,这个问题对你而言实

在是小菜一碟。但是因为上次rly把棋盘弄破了,又拿不出新的,所以rly打算难一点点,问题就是在破的棋盘上的n皇后问题。他想知道......(你们懂的..)

  妻子都是相同的

 

【输入说明】

  一行,一个整数N。

  接下来N行,每行N个整数,要么为0,表示没坏,要么为1,表示坏了。

【输出说明】

  一行,输出不同的接的数量。

【样例输入】

  4

  1 0 1 1

  1 1 1 0

  0 1 1 1

  1 1 0 1

【样例输出】

  1

【数据范围】

  对于40%的数据,N<=13。

  对于100%的数据,N<=16。

  对于30%的数据棋盘没有破,你可以认为rly又去买了一个新的。

 1 /*位运算优化n皇后*/
 2 #include<cstdio>
 3 #include<iostream>
 4 #define N 20
 5 using namespace std;
 6 int hang[N],n,ans;
 7 void dfs(int t,int a,int b,int c)//a,b,c都是二进制数,表示哪一列或对角线是否放过皇后,1为放过 
 8 {
 9     if(t==n+1)
10     {
11         ans++;return;
12     }
13     int S=((1<<n)-1)&(~(hang[t]|a|b|c));
14     /*
15       S是一个二进制数,表示当前可以往哪一列放皇后 
16       (hang[t]|a|b|c)表示哪些列从列和对角线的角度来说都放过
17       加上~(取反),表示哪些列可以放皇后 
18     */ 
19     while(S)
20     {
21         int x=S&(-S);//x表示可以当前放皇后的位置 
22         dfs(t+1,a+x,(b+x)<<1,(c+x)>>1);
23         S-=x;
24     }
25 }
26 int main()
27 {
28     scanf("%d",&n);
29     for(int i=1;i<=n;i++)
30       for(int j=1;j<=n;j++)
31       {
32           int x;scanf("%d",&x);
33           hang[i]+=x<<(j-1);
34       }
35     dfs(1,0,0,0);
36     printf("%d",ans);
37     return 0;
38 }

思路:位运算优化后的n皇后问题

posted @ 2016-11-08 15:28  浮华的终成空  阅读(144)  评论(0编辑  收藏  举报

Contact with me