算法训练 排列问题

算法训练 排列问题
时间限制：1.0s 内存限制：512.0MB
提交此题
问题描述
　　求一个0～N-1的排列（即每个数只能出现一次），给出限制条件（一张N*N的表，第i行第j列的1或0，表示为j-1这个数不能出现在i-1这个数后面，并保证第i行第i列为0），将这个排列看成一个自然数，求从小到大排序第K个排列。
数据规模和约定
　　N<=10，K<=500000
输入格式
　　第一行为N和K,接下来的N行，每行N个数，0表示不能，1表示能
输出格式
　　所求的排列
样例输入
3 2
0 1 1
1 0 0
0 1 0

样例输出

1 0 2
解释：
对于N=3的没有任何限制的情况
第一：0 1 2
第二：0 2 1
第三：1 0 2
第四：1 2 0
第五：2 0 1
第六：2 1 0
根据题目所给的限制条件由于2不能出现在1后面，0不能出现在2后面
第一：0 2 1
第二：1 0 2
第三：2 1 0

#include<iostream>
#include<algorithm>
using namespace std;
int a[10];
int b[10];
int ma[10][10];
int n,m;
int total=0;
int is()//判断是否符合限制条件
{
    for (int i=0;i<n-1;i++)
    {
        if (ma[a[i]][a[i+1]]==1)
            continue;
        else
        {
            return 0;
        }
    }
    return 1;
}
void fun(int k)
{
    if(k>n-1)
    {
        if (is())
        {
            total++;
            if (total==m)
            for (int i=0;i<n;i++)
            {
               cout<<a[i]<<" ";
            }
        }
        return ;
    }
    else
    {
        for (int i=0;i<n;i++)
        {
            if (b[i]==0)
            {
                b[i]=1;
                a[k]=i;
                fun(k+1);
                b[i]=0;
            }
        }
    }
}
int main()
{
    cin>>n>>m;
    for (int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>ma[i][j];
        }
    }
    fun(0);
    return 0;
}