题目链接 https://cn.vjudge.net/problem/UVA-1262
【题意】
给两个6行5列的字母矩阵,找出满足如下条件的密码,密码长度为5位,密码中的每个字母在两个矩阵的对应列中均出现.比如密码的左数第二个字母必须同时在两个矩阵的第二列出现.你要找出字典序为第K小的密码(K<=7777)无解时输出”NO”
【思路】
这题本意是可以计数计算的,但数据范围小可以直接暴搜
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=10;
int K,num;
char s1[N][N],s2[N][N];
vector<char> g[N];
int p[5];
bool dfs(int a,int b,int c,int d,int e){
++num;
if(num==K){
p[0]=a,p[1]=b,p[2]=c,p[3]=d,p[4]=e;
return true;
}
if(e+1<g[4].size()) return dfs(a,b,c,d,e+1);
else{
if(d+1<g[3].size()) return dfs(a,b,c,d+1,0);
else{
if(c+1<g[2].size()) return dfs(a,b,c+1,0,0);
else{
if(b+1<g[1].size()) return dfs(a,b+1,0,0,0);
else{
if(a+1<g[0].size()) return dfs(a+1,0,0,0,0);
else return false;
}
}
}
}
}
int main(){
int T;
scanf("%d",&T);
while(T--){
scanf("%d",&K);
for(int i=0;i<N;++i) g[i].clear();
for(int i=0;i<6;++i) scanf("%s",s1[i]);
for(int i=0;i<6;++i) scanf("%s",s2[i]);
for(int j=0;j<5;++j){
for(int i=0;i<6;++i){
for(int k=0;k<6;++k){
if(s1[i][j]==s2[k][j]) g[j].push_back(s1[i][j]);
}
}
}
bool ok=true;
for(int i=0;i<5;++i){
if(g[i].empty()){ok=false;break;}
sort(g[i].begin(),g[i].end());
g[i].erase(unique(g[i].begin(),g[i].end()),g[i].end());
}
if(!ok){puts("NO");continue;}
num=0;
if(dfs(0,0,0,0,0)){
for(int i=0;i<5;++i) printf("%c",g[i][p[i]]);
puts("");
}
else puts("NO");
}
return 0;
}