hihoCoder 1107 Shortest Proper Prefix （字典樹的遍歷）
來源：程序員人生 發(fā)布時(shí)間：2015-05-06 08:33:09 閱讀次數(shù)：3021次
題目鏈接：
http://hihocoder.com/problemset/problem/1107?sid=406355
題意：
求頻率不超過5的前綴的數(shù)目。字典樹的遍歷
代碼以下：
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 10010;
const int max_size = 30;
int id(char c){
return c-'a';
}
struct Trie{
Trie *ch[max_size];
int num;
Trie(){
num = 0;
for(int i=0;i<max_size;i++)
ch[i]=NULL;
}
}*root;
void insert_str(char *s){
Trie *p = root;
p->num++;
for(int i=0; p&&s[i] ; i++){
int u = id(s[i]);
if(p->ch[u]==NULL)
p->ch[u] = new Trie;
p=p->ch[u];
p->num++;
}
}
int find_str(char *s){
Trie *p = root;
for(int i=0;p&&s[i];i++){
int u = id(s[i]);
if(p->ch[u]==NULL) return 0;
p=p->ch[u];
}
return p->num;
}
int ans;
void vis(Trie *nod){
Trie *p=nod;
if(p->num<=5&&p->num>0) {ans++;return;}
for(int i=0;i<max_size;i++)
if(p->ch[i]) vis(p->ch[i]);
}
char s[2000010];
int main()
{
int n, m;
while(~scanf("%d", &n)){
root = new Trie;
for(int i=0;i<n;i++){
scanf("%s", s);
insert_str(s);
}
ans = 0;
vis(root);
printf("%d
",ans);
}
return 0;
}
生活不易，碼農(nóng)辛苦
如果您覺得本網(wǎng)站對您的學(xué)習(xí)有所幫助,可以手機(jī)掃描二維碼進(jìn)行捐贈
