找了这题回溯挺好的,刚学的菜鸟动动脑子做出来受益不浅啊哈哈。
主体和八皇后差不多,区别在于这题棋子数可以少于行数,八皇后是棋子放完了就退出了,所以在循环结束后加上一个条件语句,如果棋子数不满足k个并且不是最后一行,那么再去继续搜索。还有一点就是上一个搜索退出后除了改变访问过的1还要num--,以便继续搜索。
#include<cstdio>
#include <iostream>
#include<cstring>
using namespace std;
int vis[4][50], tot,num;
char Mao[10][10];
int n,k;
void dfs(int a)
{
if(num==k && a!=n) tot++;
else if(num==k && a==n) tot++;
else
{
for(int i=0; i<n; i++)
{
if(Mao[a][i]=='#' && !vis[0][i])
{
num++;
vis[0][i]=1;
dfs(a+1);
vis[0][i]=0;
num=num-1;
}
}
if(num!=k && a!=n) dfs(a+1);
}
}
int main()
{
while(scanf("%d%d",&n,&k) && n!=-1 && k!=-1)
{
num=0;
tot=0;
memset(vis,0,sizeof(vis));
memset(Mao,0,sizeof(Mao));
for(int i=0; i<n; i++)
scanf("%s",Mao[i]);
dfs(0);
printf("%d\n",tot);
}
return 0;
}