【USACO 1.2】Name That Number

时间:2023-03-09 02:10:16
【USACO 1.2】Name That Number

给你一串数字(≤12个),每个数字可以对应3个字母,求生成的所有字符串里,在字典内的有哪些。

我做的时候想的是字典树(Trie 树),模拟数串生成的所有字符串,然后在字典树里查找一下。

/*
TASK:namenum
LANG:C++
*/
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#define ll long long
#define MAX 26
#define N 5001
#define M 15 using namespace std;
int top,trie[N*M][MAX+];
int search(char s[]){
int i,rt;
for(rt=,i=;rt=trie[rt][s[i]-'A'];)
if(!s[++i])return trie[rt][MAX];
return ;
}
void insert(char s[]){
int rt=;
for(int i=;s[i];i++){
int &nxt=trie[rt][s[i]-'A'];
if(==nxt) nxt=++top;
rt=nxt;
}
trie[rt][MAX]=;
} int cnt;
//虽然只有3个字母,但是后面有\0,故要开到4。
char ma[][]={"","","ABC","DEF","GHI","JKL","MNO","PRS","TUV","WXY"};
char s[M],name[M];
int dfs(int d){
if(!s[d]){
if(search(name)){
cout<<name<<endl;
return ;
}
return ;
}
int ok=;
for(int i=;i<;i++){
name[d]=ma[s[d]-''][i];
if(dfs(d+))ok=;
}
return ok;
}
int main() {
//freopen("namenum.in","r",stdin);
//freopen("namenum.out","w",stdout);
ifstream in("dict.txt");
while(in>>s)insert(s);
cin>>s;
if(!dfs())cout<<"NONE"<<endl;
return ;
}

官方题解里说可以二分,还可以把字典全部转为数字,总共就5000个,这就是逆向思维。

/*
TASK:namenum
LANG:C++
*/
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#define ll long long
#define MAX 26
#define N 5001
#define M 15 using namespace std; char dic[N][M];
int ma[MAX]={,,,,,,,,,,,,,,,,,,,,,,,,};
ll num[N],tot,t;
int main() {
freopen("namenum.in","r",stdin);
freopen("namenum.out","w",stdout);
ifstream in("dict.txt");
while(in>>dic[tot]){
ll &ans=num[tot];
for(int i=;dic[tot][i];i++)
ans=ans*+ma[dic[tot][i]-'A'];
tot++;
}
cin>>t;
int ok=;
for(int i=;i<tot;i++)
if(num[i]==t){ok=;cout<<dic[i]<<endl;}
if(!ok)cout<<"NONE"<<endl;
return ;
}