hdu 1251 统计难题

题目：点击打开链接

分析：一道裸的字典树模板题。。。

代码：

///#include<bits/stdc++.h>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn = 1e5;
struct tre{
int next[26];
int val;
}tree[5*maxn];
int cnt=1;

int init(){///初始化节点
memset(&tree[cnt],0,sizeof(tre));///把next数组和val都赋值为0（结构体节点整体赋值）
return cnt++;
}

void build(char *s){
int tmp=0,len=strlen(s);
for(int i=0;i<len;i++){
int num=s[i]-'a';
if(!tree[tmp].next[num])
tree[tmp].next[num]=init();
tmp=tree[tmp].next[num];
tree[tmp].val++;///每个字符串的所有前缀标记+1
}
}

int Count(char *s){///记录前缀所含个数
int index=0,len=strlen(s);
for(int i=0;i<len;i++){
int num=s[i]-'a';
if(!tree[index].next[num]) return 0;
index=tree[index].next[num];
}
return tree[index].val;
}

int main(){
memset(&tree[0],0,sizeof(tre));
char word[11];

// while(gets(word)){
// if(word[0]==NULL) //空行。gets读入的回车符会自动转换为NULL。
// break;
// Insert(word);
// }
// while(gets(word)){
// if(word[0]==NULL) //空行。gets读入的回车符会自动转换为NULL。
// break;
// Insert(word);
// }

while(cin.getline(word,12)){ ///输入单词
if(strlen(word)==0 || word[0]==' ') ///如果读入字符串的长度为0或者是空格，说明读入的是空行
break;
build(word);
}
while(scanf("%s",word)!=EOF)
cout<<Count(word)<<endl;
}


这次被memset函数坑了。。。 memset是以字节为单位，初始化内存块，memset函数一共有三个参数，第一个是变量的地址，第二是所赋的值，int数组只能为0或-1，第三个为内存块的大小。memset对结构体内的元素赋值要特别注意！！！