题意:
给定两表(6行5列),只有同列中同时存在的的字母才可作为密码,现在要求求解按字典序排序第n个密码。
解题思路:
先找出可作为密码的字母,其个数分别为a,b,c,d,e,并计算总的种类数。
第一个字母若为A,则n应小于b*c*d*e;
若为B,则n应大于b*c*d*e小于2*b*c*d*e;
....
由此可以推出余下的四个字母。
坑点:
我做题的时候完完全全没注意到重复性,若同列中有两个C,那么可选密码的个数不应为2,而是1.
所以应该去重,我一开始用vector再排序就完全没注意到这个问题。所以使用set就能解决问题。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#define Siz(x) (int)x.size()
using namespace std;
int T, k;
char s1[7][7];
char s2[7][7];
set<int>g[7];
set<int>::iterator it;
inline bool check(int ch,int col){
for (int i = 0; i < 6; ++i){
if (s2[i][col] == ch) return 1;
}
return 0;
}
int sum[7];
int main(){
// freopen("out.txt","w",stdout);
scanf("%d",&T);
while(T--){
scanf("%d",&k);
for (int i = 0; i < 7; ++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 i = 0; i < 6; ++i){
for (int j = 0; j < 5; ++j){
if (check(s1[i][j],j)) g[j].insert(s1[i][j]);
}
}
int cur = 0;
sum[5] = 1;
for (int i = 4; i >= 0; --i){
sum[i] = sum[i+1] * Siz(g[i]);
}
if (k > sum[0]) {
puts("NO");
continue;
}
for (int i = 0; i < 5; ++i){
for (it = g[i].begin(); it != g[i].end(); ++it){
if (cur + sum[i+1] < k){
cur += sum[i+1];
}
else break;
}
putchar(*it);
}
putchar('\n');
}
return 0;
}