Repository HDU
阿新 • • 發佈:2018-12-19
題意:給出一組單詞,再給出一組字串,問這組字串在這組單詞出現過多少次?
題解:使用字典樹,對於每個單詞,從每一個位置開始都放入字典樹中,但是有一點要注意,如果字串是abab,這樣如果統計ab那麼豈不是多算了一個麼,所以使用一個數組統計當前到了哪個單詞(具體可見程式碼),最後跟據字典樹查詢即可。
附上程式碼:
#include<iostream> #include<cstdio> #include<cstring> using namespace std; const int maxnode=1e6+50; const int sigma_size=26; const int maxn=35; struct Trie{ int ch[maxnode][sigma_size]; int val[maxnode]; int id[maxnode]; int sz; void clear(){ sz=1; memset(ch[0],0,sizeof(ch[0])); memset(val,0,sizeof(val)); memset(id,0,sizeof(id)); } int idx(char c){ return c-'a'; } void insert(char *s,int v){ int u=0,n=strlen(s); for(int i=0;i<n;i++){ int c=idx(s[i]); if(!ch[u][c]){ memset(ch[sz],0,sizeof(ch[sz])); ch[u][c]=sz++; } u=ch[u][c]; if(id[u]!=v){ val[u]++; } id[u]=v; } } int find_prefixes(char *s,int len){ int u=0,ans=0; for(int i=0;i<len;i++){ if(s[i]=='\0'){ return 0; } int c=idx(s[i]); if(!ch[u][c]){ return 0; } u=ch[u][c]; } return val[u]; } }; Trie trie; int main() { char s[maxn]; int p,q; scanf("%d",&p); trie.clear(); for(int i=1;i<=p;i++){ scanf("%s",s); int len=strlen(s); for(int j=0;j<len;j++){ trie.insert(s+j,i); } } scanf("%d",&q); for(int i=0;i<q;i++){ scanf("%s",s); printf("%d\n",trie.find_prefixes(s,strlen(s))); } return 0; }