C++搜索与回溯算法之置棋问题

置棋问题

题目描述

在M*N的主格中任意指定X个格子构成一个棋盘，而其它格子是残缺的，不能放棋子。在任一个构成的棋盘上放置K个棋子，要求任意两个棋子不得位于同一行或同一列上。求满足条件的所有方案数。

注意：棋盘是稀疏的，即X＜(M*N)/2

编程任务：

1. 对给定的一个棋盘，求出该棋盘可放置的最多的棋子数P；

2. 该棋盘上恰好放置i个棋子时的方案总数（1≤i≤P）,其中经旋转和镜面反射而得的方案记为不同的方案

输入

第1行：2个整数M,N，表示棋盘的行数和列数( 1＜M，N＜10)

接下来是M行，每行N个空格分开的数字（0或者1），1表示相应的格子在棋盘上，0表示相应的格子不在棋盘上。

输出

第1行：1个整数表示可放置的最多的棋子数P 接下来P行，每行分别列出放i（1≤i≤P）个棋子时的方案总数。

样例输入

5 4
0 1 1 1 
0 1 0 0 
1 1 1 0 
0 0 1 0 
0 0 1 1 


样例输出
4
1:10
2:28
3:24
4:5

实现代码
代码如下：
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,k,v[900],tot[900],large;
int map[900][900];
bool a[900],b[900];
void output()
{
    for(int i=1;i<=large;i++) cout<<i<<":"<<tot[i]<<endl;
}
void search(int x,int y)
{
    for(int i=v[x-1];i<=n;i++)
        for(int j=1;j<=m;j++)
            if(!a[i]&&!b[j]&&map[i][j]==1)
            {
                v[x]=i;
                a[i]=1;
                b[j]=1;
                tot[y]++;
                if(y>large) large=y;
                search(x+1,y+1);
                a[i]=0;
                b[j]=0;
            }
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            cin>>map[i][j];
    search(1,1);
    cout<<large<<endl;
    output();
}