蓝桥杯 ALGO-52 算法训练 排列问题

问题描述

求一个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表示能

输出格式

所求的排列

样例输入

#include <iostream>
#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
    int n, k, t, a[10], index[10], cnt = 1;
    vector<pair<int, int> > v;
    cin >> n >> k;
    for (int i = 0; i < n; i++) 
        a[i] = i;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cin >> t;
            if (t == 0 && i != j) {
                v.push_back({i, j});
            }
        }
    }
    do {
        int flag = 0;
        for (int i = 0; i < n; i++)
            index[a[i]] = i;
        for (int i = 0; i < v.size(); i++) {
            if (index[v[i].second] - index[v[i].first] == 1) {
                flag = 1;
                break;
            }
        }
        if (flag == 0) {
            if (cnt == k) {
                for (int i = 0; i < n; i++)
                    cout << a[i] << ' ';
                break;
            } else {
                cnt++;
            }
 
        }
 
    } while (next_permutation(a, a + n));
    return 0;
}