2019 年百度之星·程序设计大赛 - 初赛二1002 度度熊与排列

1002 度度熊与排列

题目

分析

一共给出了 n 对，随便拿一对，用 dfs 按字典序选择每一个位置，选完后用剩下的 n-1对 检验 check一下，不行就回溯上个位置。

#include <bits/stdc++.h>
using namespace std;
#define ll long long 
#define INF 0x3f3f3f3f
#define mk(x, y) make_pair(x, y)
#define pii pair<string, string>
#define fi first
#define se second
#define fuck(x) cout<<x<<endl
const int N = 1e2 + 5;
const ll mod = 1e9 + 7;

int t, n, m;
pii a[N];
int b[N];
string res;
vector<int> v[30];

bool check(int x, int y){
    for(int i = 1; i < n; i++){
        if(a[i].fi[x] != a[i].se[y]){
            return false;
        }
    }
    return true;
}

int vis[N], vvv;

void dfs(int ss, int pos, string ans){  // dfs 枚举每个位置判断，不行就回溯
    if(ss == m){    //  ans 存结果
        vvv = 1;
        res = ans;
        return ;
    }
    for(auto x : v[a[0].fi[ss] - 'a']){
        if(!vis[x] && check(ss, x)){
            vis[x] = 1;
            dfs(ss + 1, pos + 1, ans+char('0'+x));
            if(vvv)
                return;
        }
    }
}

void sovle(){
    for(int i = 0; i < 30; i++){        // 初始化
        v[i].clear();
    }
    for(int i = 0; i < a[0].se.size(); i++){    // 找出所有可行位置
        v[a[0].se[i] - 'a'].push_back(i);
    }
    memset(vis, 0, sizeof(vis));    // 初始化
    vvv = 0;
    res = "";
    dfs(0, 0, "");
    if(res.size() == 0){
        cout << -1 << "\n";
    }else{
        for(int i = 0; i < res.size(); i++){
            printf("%d%c", res[i] - '0' + 1, " \n"[i == res.size() - 1]);
        }
    }
}   

int main(){
    scanf("%d", &t);
    while(t--){
        scanf("%d%d", &n, &m);
        int flag = 0;
        for(int i = 0; i < n; i++){
            string x, y, b, c;
            cin >> x >> y;
            a[i] = mk(x, y);
            b = x, c = y;
            sort(b.begin(), b.end());
            sort(c.begin(), c.end());
            if(b != c){
                flag = 1;
            }
        }
        if(flag){       // 不相等
            cout << -1 << endl;
            continue;
        }
        sovle();
    }
    return 0;
}