hihoCoder 1014trie树（字典树）

hihoCoder 1014

题目提示已经很清楚了~

贴代码……

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;
const int MAXN = 100000 + 10;
const int alNum = 26;
struct Node{
int cnt;
int next[alNum];
void init(){
memset(next,-1,sizeof(next));
cnt = 0;
return;
}
};
Node trie[MAXN];
int tt;
void build_trie(char str[]){
int len = strlen(str);
int p = 0;
for(int i = 0;i < len;i++){
int ch = str[i] - 'a';
if(trie
.next[ch] == -1){
trie[tt].init();
trie[p].next[ch] = tt++;
}
p = trie[p].next[ch];
trie[p].cnt++;
}
}
int quercy(char str[]){
int len = strlen(str);
int p = 0;
for(int i = 0;i < len;i++){
int ch = str[i] - 'a';
if(trie[p].next[ch] == -1){
return 0;
}
p = trie[p].next[ch];
}
return trie[p].cnt;
}
int main(){
//    freopen("input.txt","r",stdin);
int n,m;
while(~scanf("%d",&n)){
char str[20];
tt = 0;
trie[tt++].init();
for(int i = 0;i < n;i++){
scanf("%s",str);
build_trie(str);
}
scanf("%d",&m);
for(int i = 0;i < m;i++){
scanf("%s",str);
int q = quercy(str);
printf("%d\n",q);
}
}
return 0;
}

字典树 水题

判断一个是否为另一个的前缀。

注意 9113

　　911 的情况……

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;
const int MAXN = 10000000 + 10;
const int NextNum = 10;

int tt;
struct Node{
int next[NextNum];
bool flag;
void init(){
memset(next,-1,sizeof(next));
flag = 0;
}
};
Node trie[MAXN];
bool build_trie(char str[]){
int p = 0;
int len = strlen(str);
for(int i = 0;i < len;i++){
int x = str[i] - '0';
if(trie
.next[x] == -1){
trie[tt].init();
trie[p].next[x] = tt++;
}
p = trie[p].next[x];
if(trie[p].flag){
return 0;
}
}
for(int i = 0;i < NextNum;i++){
if(trie[p].next[i] != -1){
return 0;
}
}
trie[p].flag = 1;
return 1;
}

int main(){
//    freopen("input.txt","r",stdin);
int t;
scanf("%d",&t);
while(t--){
int n;
scanf("%d",&n);
tt = 0;
trie[tt++].init();
bool ok = 1;
char str[40+10];
for(int i = 0;i < n;i++){
scanf("%s",str);
if(!ok){
continue;
}
ok = build_trie(str);
}
printf("%s\n",ok?"YES":"NO");
}
return 0;
}