008-拉丁矩阵问题

题目：

现有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;
}