题目:
现有n种不同形状的宝石,每种宝石有足够多颗。欲将这些宝石排列成m行n列的一个矩阵,m<=n,使矩阵中每一行和每一列的宝石都没有相同的形状。试设计一个算法,计算出对于给定的m和n,有多少种不同的宝石排列方案。
代码:
#include <stdio.h>
#include <algorithm>
using namespace std;
#define m 3
#define n 3
#define N 100
int a[N][N];
int sum=0;
bool ok(int row,int col,int k){
for(int j=1;j<row;j++){
if(a[j][col]==k)
return false;
}
return true;
}
void traceback(int row,int col){
for(int i=col;i<=n;i++){
if(ok(row,col,a[row][i])){
swap(a[row][col],a[row][i]);
if(col==n){
if(row==m){
sum++;
return;
}
else
traceback(row+1,1);
}
else
traceback(row,col+1);
swap(a[row][col], a[row][i]);
}
}
}
int main()
{
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
a[i][j]=j;
traceback(1,1);
printf("%d\n",sum);
return 0;
}