HDU1075 字典树板子题

时间:2023-01-24 22:30:46

题意 :给出两组字符串 一一映射,给出一种组成的文字,要求映射成另外一种
思路:使用字典树,把映射的另外一个字符存在字典树的单词节点处  例如 abc   123

则把123存在abc节点中的c处即可

同时这里使用的是静态的数组,操作和写起来都更方便,就是要提前判断开的空间,过大过小都会有莫名其妙的错误

 #include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=5e5+;
const int size=;
struct Trie{
int ch[maxn][size];
char str[maxn][];
bool isEnd[maxn];
int size;
void init(){
size=;
memset(ch,,sizeof(ch));
memset(isEnd,,sizeof(isEnd));
}
int index(char s){
return s-'a';
}
void insert(char*s1,char*s2){
int i,rt;
for(i=rt=;s1[i]!='\0';i++){
int c=index(s1[i]);
if(ch[rt][c]==){
ch[rt][c]=size++;
}
rt=ch[rt][c];
}
strcpy(str[rt],s2);
isEnd[rt]=;
}
const char*find(char*s){
int i,rt;
for(i=rt=;s[i]!='\0';i++){
int c=index(s[i]);
if(!ch[rt][c])return "";
rt=ch[rt][c];
}
return (isEnd[rt])?str[rt]:""; }
}trie;
char s1[maxn],s2[maxn];
int main(){
trie.init();
while(scanf("%s",s1)==){
if(!strcmp(s1,"START"))continue;
if(!strcmp(s1,"END"))break;
scanf("%s",s2);
trie.insert(s2,s1);
}
//cout<<111<<endl;
getchar();
while(gets(s1)){
if(!strcmp(s1,"START"))continue;
if(!strcmp(s1,"END"))break;
int i=;
while(s1[i]!='\0'){
if(s1[i]>='a'&&s1[i]<='z'){
int j=;
while(s1[i]!='\0'&&s1[i]>='a'&&s1[i]<='z'){
s2[j++]=s1[i++];
}
s2[j]='\0';
const char*temp=trie.find(s2);
if(temp[]=='')printf("%s",s2);
else printf("%s",temp);
}
else putchar(s1[i++]);
// puts("");
}
// puts("");
cout<<endl;
}
return ;
}