火星文Trie插入
对应英文存到数组查询
对于每一个火星文句子，拆成若干单词分别在Trie树中查询
PS：开数组的话要开大，大概100W左右，不然会一直RE……
1 #include<cstdio> 2 #include<cstring> 3 using namespace std; 4 #define MAXN 1000100 5 int trie[MAXN][30],tag[MAXN],cnt,sz; 6 char s[MAXN][30],ans[MAXN],tmp[MAXN]; 7 void insert(int len){ 8 int x=0; 9 for (int i=0;i<len;i++){ 10 if (!trie[x][tmp[i]-'a']) 11 trie[x][tmp[i]-'a']=++sz; 12 x=trie[x][tmp[i]-'a']; 13 } 14 tag[x]=cnt; 15 } 16 int ask(int l,int r){ 17 int x=0; 18 for (int i=l;i<r;i++){ 19 if (!trie[x][tmp[i]-'a']) return 0; 20 x=trie[x][tmp[i]-'a']; 21 } 22 return tag[x]; 23 } 24 int main(){ 25 cnt=1;sz=0; 26 scanf("%s",s[cnt]); 27 while (s[cnt][0]!='E'){ 28 if (s[cnt][0]=='S'){ 29 scanf("%s",s[cnt]); 30 continue; 31 } 32 scanf("%s",tmp); 33 int len=strlen(tmp); 34 insert(len); 35 cnt++; 36 scanf("%s",s[cnt]); 37 } 38 scanf("%s",tmp); 39 while (tmp[0]!='E'){ 40 if (tmp[0]=='S'){ 41 getchar();//这个一定要加上…… 42 gets(tmp);continue; 43 } 44 int k=0,q=0,p,judge,len=strlen(tmp); 45 while (k<len){ 46 judge=0;p=k; 47 while (tmp[k]>='a'&&tmp[k]<='z'){ 48 k++;judge=1; 49 } 50 if (!judge){ 51 ans[q]=tmp[k]; 52 q++;k++; 53 continue; 54 } 55 int pos=ask(p,k); 56 if (pos){ 57 int lenth=strlen(s[pos]); 58 for (int i=0;i<lenth;i++){ 59 ans[q]=s[pos][i];q++; 60 } 61 } 62 else 63 for (int i=p;i<k;i++){ 64 ans[q]=tmp[i]; 65 q++; 66 } 67 } 68 for (int i=0;i<q;i++) 69 printf("%c",ans[i]); 70 printf("\n"); 71 gets(tmp); 72 } 73 return 0; 74 }
