[USACO08DEC]Secret Message

OJ题号：

洛谷2922

思路：

字典树，每个结点记录经过该节点的字符串数cnt和以该结点结尾的字符串数量val。
每次询问时累加经过节点的val值和结尾结点的cnt值。

#include<cstdio>
#include<cctype>
#include<cstring>
inline int getint() {
    char ch;
    while(!isdigit(ch=getchar()));
    int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
class Trie {
    private:
        static const int maxnode=500001;
        int ch[maxnode][2];
        int val[maxnode],cnt[maxnode];
        int sz;
    public:
        Trie() {
            sz=0;
            memset(ch,0,sizeof ch);
            memset(val,0,sizeof val);
            memset(cnt,0,sizeof cnt);
        }
        void insert(int *s) {
            int x=0,n=s[0];
            for(int i=1;i<=n;i++) {
                if(!ch[x][s[i]]) {
                    ch[x][s[i]]=++sz;
                }
                cnt[x]++;
                x=ch[x][s[i]];
            }
            val[x]++;
        }
        int query(int *s) {
            int x=0,n=s[0],ans=0;
            for(int i=1;i<=n;i++) {
                ans+=val[x];
                if(!ch[x][s[i]]) {
                    return ans;
                }
                x=ch[x][s[i]];
            }
            return ans+val[x]+cnt[x];
        }
};
Trie t;
int main() {
    int m=getint(),n=getint();
    int s[10001];
    while(m--) {
        s[0]=getint();
        for(int i=1;i<=s[0];i++) s[i]=getint();
        t.insert(s);
    }
    while(n--) {
        s[0]=getint();
        for(int i=1;i<=s[0];i++) s[i]=getint();
        printf("%d\n",t.query(s));
    }
    return 0;
}