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();
    }
}