codevs1322 单词矩阵

 1 //It is made by jump~
 2 #include <iostream>
 3 #include <cstdlib>
 4 #include <cstring>
 5 #include <cstdio>
 6 #include <cmath>
 7 #include <algorithm>
 8 #include <ctime>
 9 #include <vector>
10 #include <queue>
11 #include <map>
12 #include <set>
13 using namespace std;
14 typedef long long LL;
15 int n;
16 char cc,ch[26];
17 int ans[26];
18 int f[6][6][6][6][6];
19 //f[i][j][k][l1][l2]表示第一行摆了i个，第二行摆了j个，第三行摆了k个，第四行摆了l1个，第五行摆了l2个的方案数，保证五个数不降，才能保证合法性
20 
21 inline int getint()
22 {
23        int w=0,q=0; char c=getchar();
24        while((c<‘0‘ || c>‘9‘) && c!=‘-‘) c=getchar(); if(c==‘-‘) q=1,c=getchar(); 
25        while (c>=‘0‘ && c<=‘9‘) w=w*10+c-‘0‘, c=getchar(); return q ? -w : w;
26 }
27 
28 inline bool check(int x,int num){ return (!ans[x]) || (ans[x]==num); }
29 
30 inline int dfs(int a,int b,int c,int d,int e,int tot){
31     if(tot==25) return 1;
32     int &tmp=f[a][b][c][d][e];
33     if(tmp) return tmp;
34     if(a<5 && check(a,tot+1)) tmp+=dfs(a+1,b,c,d,e,tot+1); //填在第一行
35     if(b<a && check(b+5,tot+1)) tmp+=dfs(a,b+1,c,d,e,tot+1); //保证严格不降
36     if(c<b && check(c+10,tot+1)) tmp+=dfs(a,b,c+1,d,e,tot+1);
37     if(d<c && check(d+15,tot+1)) tmp+=dfs(a,b,c,d+1,e,tot+1);
38     if(e<d && check(e+20,tot+1)) tmp+=dfs(a,b,c,d,e+1,tot+1);
39     return tmp;
40 }
41 
42 inline void work(){
43     cc=getchar(); 
44     if(cc==‘W‘) {
45     scanf("%s",ch); 
46     for(int i=0;i<25;i++) {
47         for(ans[i]=1;ans[i]<ch[i]-‘A‘+1;ans[i]++) {
48         memset(f,0,sizeof(f));
49         n+=dfs(0,0,0,0,0,0);
50         }
51     }
52     printf("%d",n+1);//加上自己
53     }
54     else {
55     n=getint(); int now;
56     for(int i=0;i<25;i++) {
57         for(ans[i]=1;ans[i]<=26;ans[i]++) {//确定状态，搜索在当前状态下的答案数
58         memset(f,0,sizeof(f));//随时清空
59         now=dfs(0,0,0,0,0,0);
60         if(now>=n) break;
61         n-=now;
62         }
63     }
64     for(int i=0;i<25;i++) printf("%c",ans[i]+‘A‘-1);
65     }
66 }
67 
68 int main()
69 {
70   work();
71   return 0;
72 }