P2750 [USACO5.5]贰五语言Two Five

题目描述

有一种奇怪的语言叫做“贰五语言”。它的每个单词都由A～Y这25个字母各一个组成。但是，并不是任何一种排列都是一个合法的贰五语言单词。贰五语言的单词必须满足这样一个条件：把它的25个字母排成一个5*5的矩阵，它的每一行和每一列都必须是递增的。比如单词ACEPTBDHQUFJMRWGKNSXILOVY，它排成的矩阵如下所示：

A C E P T

B D H Q U

F J M R W

G K N S X

I L O V Y

因为它的每行每列都是递增的，所以它是一个合法的单词。而单词YXWVUTSRQPONMLKJIHGFEDCBA则显然不合法。 由于单词太长存储不便，需要给每一个单词编一个码。编码方法如下：从左到右，再从上到下，可以由一个矩阵的得到一个单词，再把单词按照字典顺序排序。比如，单词ABCDEFGHIJKLMNOPQRSTUVWXY的编码为1，而单词ABCDEFGHIJKLMNOPQRSUTVWXY的编码为2。

现在，你需要编一个程序，完成单词与编码间的转换。

输入输出格式

输入格式：

第一行为一个字母N或W。N表示把编码转换为单词，W表示把单词转换为编码。

若第一行为N，则第二行为一个整数，表示单词的编码。若第一行为W，则第二行为一个合法的单词。

输出格式：

每行一个整数或单词。

输入输出样例

输入样例#1：

N
2

输出样例#1：

ABCDEFGHIJKLMNOPQRSUTVWXY

输入样例#2：

W 
ABCDEFGHIJKLMNOPQRSUTVWXY

输出样例#2：

2

说明

题目翻译来自NOCOW。

USACO Training Section 5.5

此题一开始先打了个dfs这样的dfs只能针对编码转成字符串的这种情况，而且由于情况太多，导致很快就会搜爆，于是我就有了8分的程序：

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#define inf 999999999
#define For(i,a,b) for(i=a;i<=b;++i)
#define rep(i,a,b) for(i=a;i>=b;--i)
#define mm(a,b) memset(a,b,sizeof(a))
#define ll long long
using namespace std;
int read(){
    int sum=0,flag=1;
    char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')flag=-1;c=getchar();}
    while(c>='0'&&c<='9')sum=sum*10+c-'0',c=getchar();
    return sum*flag;
}
int maxx(int x,int y){
    if(x<y)return y;
    return x;
}
int minn(int x,int y){
    if(x<y)return x;
    return y;
}
int abss(int x){
    if(x>=0)return x;
    return -x;
}
const int maxn = 100010;
struct node{
    char s[1010];
};
node a[1010];
char pos[1010][1010];
int vis[maxn],cnt,k,flag;
char ss[]="ABCDEFGHIJKLMNOPQRSTUVWXY";
void dfs(int h,int now,int sum){
    int i;
    if(flag)return;
    if(sum+26-now<5)return;
    if(sum==5){
        if(h==5){
            ++cnt;
            if(cnt==k){
            flag=1;
                int j;For(i,1,5){For(j,1,5){printf("%c",ss[pos[i][j]-1]);}}
            }
            printf("\n");
            return;
        }
        dfs(h+1,1,0);
        return;
    }
    For(i,now,25){
        if(!vis[i]&&pos[h-1][sum+1]<i){
            vis[i]=1;
            pos[h][sum+1]=i;
            dfs(h,i+1,sum+1);
            vis[i]=0;
            pos[h][sum+1]=0;
        }
    }
}
char sss[10];
int main(){
    int i,j;
    scanf("%s",sss);
    scanf("%d",&k);
    dfs(1,1,0);
    return 0;
}

这样的搜索太暴力了！！！！

我只好在网上搜索，发现只有采用记忆化：

记忆化可以用dp[a][b][c][d][e]表示第一行选a个，第二行。。。这样可以大力减少搜索时间

具体思想：逼近法：

　　先看编号转字符串，枚举字符串开头（即一行一行地填字符矩阵），比如说“AB”开头的字符串有多少个？如果大于要求的编号的话，那么自然答案字符串的开头就是“AB”了。这个思想十分容易理解。

　　字符串转编号，类似地！如果这个字符串是“AC~~~~~”，我们先加上"AB~~~~"的字符串个数，然后一直逼近，直至相同。（以上摘自http://www.cnblogs.com/albert7xie/p/5347496.html）

具体实现在代码里有解释：

#include<cmath>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define For(i,a,b) for(i=a;i<=b;++i)
#define rep(i,a,b) for(i=a;i>=b;--i)
using namespace std;
int dp[6][6][6][6][6];
char s[10010];
char st[2];
bool pd(int h,int now){
    return (!s[h] || s[h]==now+'A');//如果当前这一位没有限制或者满足限制
}
int dfs(int a,int b,int c,int d,int e,int now){//now是现在搜到了哪一个字母
    if(now==25)return 1;
    int res = dp[a][b][c][d][e];
    if(res)return res;
    if(a<5 && pd(a,now)){res+=dfs(a+1,b,c,d,e,now+1);}//
    if(b<a && pd(b+5,now)){res+=dfs(a,b+1,c,d,e,now+1);}//b<a是因为只有这时候b位置的上方才填了数
   if(c<b && pd(c+10,now)){res+=dfs(a,b,c+1,d,e,now+1);}
    if(d<c && pd(d+15,now)){res+=dfs(a,b,c,d+1,e,now+1);}
    if(e<d && pd(e+20,now)){res+=dfs(a,b,c,d,e+1,now+1);}
    return dp[a][b][c][d][e]=res;//记忆化
}
char sss[1010];
int main(){
    int i,j,n;
    scanf("%s",st);
    if(st[0] == 'N'){
        scanf("%d",&n);
        For(i, 0, 24){
            for(s[i]='A' ; ;++s[i]){//枚举第i位选什么数字
                memset(dp,0,sizeof(dp));
                int tmp = dfs(0,0,0,0,0,0);
                if(tmp >= n)break;//若当前枚举的串方案已超过n，则当前这一位不能选当前枚举的字母，只能慢慢逼近
                n-=tmp;
            }
        }
        printf("%s\n",s);
    }
    else{
        int ans=0;
        scanf("%s",sss);
        For(i,0,24){
            for(s[i]='A';s[i]<sss[i];s[i]++){
                memset(dp,0,sizeof(dp));
                int tmp = dfs(0,0,0,0,0,0);
                ans+=tmp;
            }
        }
        printf("%d\n",ans+1);
    }
    return 0;
}