hdu-2846 Repository

来源：互联网发布：金融it 编辑：程序博客网时间：2024/06/05 06:31

思路：

好题，必须记一笔。

这题有两个之前没有接触过的字典树新变形。

第一个是insert的方式，不再只是单单的将一整个单词都插进去，而是从单词的第一个字母开始往后遍历，依次插入第i个字母到末尾的单词，这样做就可以满足题目中“substring”的要求，即将“substring”的问题转换为我们熟悉的字典树可以解决的“前缀”问题。

第二个是标记每个char* s所属的原始字符串，这种做法可以避免一个单词有相同部分的问题。

AC代码：

#include <iostream>#include <cstring>using namespace std;struct node{    int num;    int id;    struct node* next[26];    node()    {        num = 0;        id = -1;        for(int i = 0;i < 26;i++)            next[i] = NULL;    }};node* root;char str[10007][27];void Insert(char* s,int mark){    node* p = root;    for(;*s!='\0';s++)    {        int n = *s-'a';        if(p->next[n] == NULL) {            p->next[n] = new node();            p = p->next[n];            p->id = mark;            p->num = 1;        }        else {            p = p->next[n];            if(p->id != mark) {                p->id = mark;                p->num++;            }        }    }}int search(char* s){    node* p = root;    for(;*s!='\0';s++)    {        int n = *s-'a';        if(p->next[n] == NULL) return 0;        p = p->next[n];    }    return p->num;}int main(){    root = new node();    int P;    cin>>P;    for(int i = 0;i < P;i++)    {        cin>>str[i];        char* tmp = str[i];        for(;*tmp!='\0';tmp++)            Insert(tmp,i);    }    int tc;    cin>>tc;    char t[27];    while(tc--)    {        cin>>t;        cout<<search(t)<<endl;    }    return 0;}

0 0