1. 程式人生 > >Repository HDU

Repository HDU

傳送門

題意:給出一組單詞,再給出一組字串,問這組字串在這組單詞出現過多少次?

題解:使用字典樹,對於每個單詞,從每一個位置開始都放入字典樹中,但是有一點要注意,如果字串是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;
}