网易cc题3

这道题困扰了我两个月，终于还是用trie图的方法做出来了。

题目3 : giftscount
时间限制:8000ms
单点时限:1000ms
内存限制:256MB
描述
网易CC里，有很多礼物（总数<1000种礼物），每一种礼物都有一个编号ID，系统里使用1到10位的数字串（其实是字符串，只不过刚好是数字而已）来表示这个ID，如3个礼物ID：{44，
45， 470}。
其中礼物又分为付费礼物与免费礼物。现在我们将一小时内，当前主播收到的所有礼物（付费+免费）的ID串起来，就形成了一个很大的数字串V（总长度<20000）。
现在给定一个全部由礼物ID合成的串V，计算里面最多有多少次付费礼物的消费。
如付费礼物ID表：S = {12，3，4，44，45，470}
全部由礼物ID合成的数字串V：V = 1434512412744
最多可以由 8
个礼物组成。分别为： 4, 3, 45, 12, 4, 12，4，4这8次付费礼物的消费。
输入
第1行输入付费礼物个数，N
接下来N行（第2到第N+1行）输入每个付费礼物的ID。1<=
ID的长度 <10位。
最后一行输入礼物ID串V
输出
最多送了多少个付费礼物。
样例输入

6
12
3
4
44
45
470
1434512412744

样例输出

8

#include<iostream>
#include<string>
#include<queue>
using namespace std;

struct trie{
int flag;
struct trie *next[10];
struct trie *suffix;
trie(){
flag=0;
memset(next,NULL,sizeof(next));
suffix=NULL;
}
};

void build(trie *root,string &s)
{
trie *cur=root;
int len=s.length();
for(int i=0;i<len;++i)
{
int id=s[i]-'0';
if(!cur->next[id]){
trie *nn=new trie();
cur->next[id]=nn;
}
cur=cur->next[id];
}
cur->flag=1;
}

int main()
{
int n;
cin>>n;
string s;
trie *root=new trie();
while(n--)
{
cin>>s;
build(root,s);
}
root->suffix=root;
queue<trie *> qt;
for(int i=0;i<10;++i){
if(!root->next[i])
root->next[i]=root;
else{
root->next[i]->suffix=root;
qt.push(root->next[i]);
}

}

trie *cur,*suf;
while(!qt.empty())
{
cur=qt.front();
suf=cur->suffix;
qt.pop();
for(int i=0;i<10;++i)
{
if(!cur->next[i])
cur->next[i]=suf->next[i];
else{
cur->next[i]->suffix=suf->next[i];
qt.push(cur->next[i]);
}
}
}
cin>>s;
int len=s.length(),count=0;
cur=root;
for(int i=0;i<len;++i)
{
int id=s[i]-'0';
cur=cur->next[id];
if(cur->flag){
count++;
cur=root; //加上这个条件是因为不能重复计算(例如：45只能算一种，不能算4和45两种)
}
}
cout<<count<<endl;

return 0;
}