UVA - 1262 数学

UVA - 1262

题意：

有两个6*5 的大写字母组成的矩阵，需要找出满足条件的字典序第k小的密码：密码中每个字母在两个矩阵的对应的同一列中都出现过

代码：

// 先处理出来每一列可以取的字母，例如：{A,B,C,D},{W,F,T},{R,T},{E,P,K},{V,M} 那么k最大不超过
// 4*3*2*3*2=144，当k<=3*2*3*2=36 时第一个字母一定是‘A’，当 36<k<=72 时第一个字母一定是‘B’，如此来
// 确定每一位字母。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int t,k,nu[],sum[][];
bool a[][],s[][];
int main()
{
    scanf("%d",&t);
    while(t--){
        scanf("%d",&k);
        memset(a,,sizeof(a));
        memset(s,,sizeof(s));
        memset(nu,,sizeof(nu));
        char str[];
        int ans[];
        for(int i=;i<=;i++){
            scanf("%s",str);
            for(int j=;j<;j++)
                a[j+][str[j]-'A']=;
        }
        for(int i=;i<=;i++){
            scanf("%s",str);
            for(int j=;j<;j++){
                if(a[j+][str[j]-'A'])
                    s[j+][str[j]-'A']=;
            }
        }
        for(int i=;i<=;i++){
            int p=;
            for(int j=;j<;j++){
                if(s[i][j]) sum[i][++p]=j;
            }
            nu[i]=p;
        }
        int x=nu[];
        for(int i=;i<=;i++) x*=nu[i];
        if(k>x) printf("NO\n");
        else{
            int l=;
            while(l<=){
                x/=nu[l];
                for(int i=;i<=nu[l];i++){
                    int y=i*x;
                    if(k<=y){
                        k-=(y-x);
                        ans[l]=i;
                        break;
                    }
                }
                l++;
            }
            for(int i=;i<=;i++)
                printf("%c",'A'+sum[i][ans[i]]);
            printf("\n");
        }
    }
    return ;
}