UVa 1262 - Password（解码）

链接：

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3703

题意：

给两个6行5列的字母矩阵，找出满足如下条件的“密码”：密码中的每个字母在两个矩阵的对应列中均出现。
例如，左数第2个字母必须在两个矩阵中的左数第2列中均出现。
给定k（1≤k≤7777），你的任务是找出字典序第k小的密码。如果不存在，输出NO。

分析：

经典的解码问题。首先求出答案每一列的所有可能字母，然后逐位解码即可。
具体实现见代码。

代码：

 import java.io.*;
 import java.util.*;

 public class Main {
     static char G[][][] = new char[2][6][9];

     public static void main(String args[]) {
         Scanner cin = new Scanner(new BufferedInputStream(System.in));

         int T = cin.nextInt();
         while(T --> 0) {
             int k = cin.nextInt();
             cin.nextLine(); // 清空缓存区
             for(int t = 0; t < 2; t++) {
                 for(int r = 0; r < 6; r++) {
                     G[t][r] = cin.nextLine().toCharArray();
                 }
             }

             @SuppressWarnings("unchecked")
             ArrayList<Character> a[] = new ArrayList[5]; // a[c]: 答案第c列的所有可能字母
             for(int i = 0; i < 5; i++) a[i] = new ArrayList<Character>();

             boolean has[][][] = new boolean[2][5][26]; // 第t个矩阵的第c列是否有字母ch
             for(int c = 0; c < 5; c++) {
                 for(int t = 0; t < 2; t++) {
                     for(int r = 0; r < 6; r++) {
                         int ch = G[t][r][c] - 'A';
                         has[t][c][ch] = true;
                     }
                 }
                 for(int i = 0; i < 26; i++)
                     if(has[0][c][i] && has[1][c][i]) a[c].add((char)(i+'A'));
             }

             int up = 1;
             for(int i = 0; i < 5; i++) up *= a[i].size();
             if(k > up) {
                 System.out.println("NO");
                 continue;
             }

             k--; // 解码
             for(int i = 0; i < 5; i++) {
                 up /= a[i].size();
                 System.out.print(a[i].get(k/up));
                 k %= up;
             }
             System.out.println();
         }
         cin.close();
     }
 }