目录
废话:
早上考完英语,心情崩溃,故做一题调理心绪。
首先,上题:https://vjudge.net/problem/POJ-1321
题目描述:
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
输入:
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
输出:
对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。
样例:
2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1
输出:
2
1
思路分析:
看到棋盘、方案数,一兴奋:八皇后
(啪)
让我们看看这道题和八皇后的区别:
八皇后:使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。
棋盘问题:摆放k个棋子,要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列。
因此,这道题虽然和八皇后有点像,但是递归的条件和八皇后不同。但是,方法还是我们熟悉的DFS(深度优先搜索),还是比较简单的。
回到这道题:
因为要涉及到DFS,所以尽量把输入的变量都声明为全局变量。毕竟递归的时候用的是函数,用作全局变量方便读写。之后,按照题目要求读入数据:
int n,k;
int result=0;//记录方案数。
char floor[10][10]={'\0'};//题目要求是8x8,开大一点,就10x10吧(内存白菜价)。
由于涉及多组数据的读入,而且结束的标志是输入-1 -1,所以使用一个while循环来开始整个程序。
int main()
{
while(cin>>n&&cin>>k&&n!=-1&&k!=-1)
{
result=0;
memset(floor,'\0',sizeof(floor));//记得初始化!
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cin>>floor[i][j];
}
}
}
return 0;
}
大体框架搭完了,下面集中精力写搜索吧。
void dfs(int t,int left)
//由于题目限制了棋子数,而且每种方案走到每一步旗子剩余量不同。
//所以需要将其作为参数传入下一层。
//还有一个参数(t)表示现在在搜的层。
{
if(t==n&&left!=0) return;//什么?到最后一行了还没放完?再见(-_-#)
if(left==0)//啊哈,放完啦,加一分(^・ω・^ )
{
result++;
return;
}
else//没放完?继续!
{
for(int i=0;i<=n;i++)//一共有n列,分别是n种情况。但是还有一种情况:不放!
{
if(vis[i]==0&&floor[t][i]=='#'&&i<n)//我可不是你想放就能放的,我有我的底线!
{
vis[i]=1;//放了下面的就不能再放咯~
dfs(t+1,left-1);//继续下一行层
vis[i]=0;//下面都被你搜光了Σ(  ̄□ ̄||)<。。走了。。
}
else if(i==n)//行吧,不放就不放
{
dfs(t+1,left);
}
}
return;
}
}
咳咳……按照上面的方法就能把总的方案数求出来,最后把result输出出来就行了。记得换行哦~
完整代码:
#include <iostream>
#include <cstring>
using namespace std;
int n,k;
int result=0;
char floor[10][10]={'\0'};
bool vis[10]={0};
void dfs(int t,int left)
{
if(t==n&&left!=0) return;
if(left==0)
{
result++;
return;
}
else
{
for(int i=0;i<=n;i++)
{
if(vis[i]==0&&floor[t][i]=='#'&&i<n)
{
vis[i]=1;
dfs(t+1,left-1);
vis[i]=0;
}
else if(i==n)
{
dfs(t+1,left);
}
}
}
}
int main()
{
while(cin>>n&&cin>>k&&n!=-1&&k!=-1)
{
result=0;
memset(floor,'\0',sizeof(floor));
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cin>>floor[i][j];
}
}
dfs(0,k);
cout<<result<<endl;
}
return 0;
}
好的,心情舒畅,下去看高数。
最后,祝大家: