[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^'';
     while(isdigit(ch=getchar())) x=(((x<<)+x)<<)+(ch^'');
     return x;
 }
 class Trie {
     private:
         static const int maxnode=;
         int ch[maxnode][];
         int val[maxnode],cnt[maxnode];
         int sz;
     public:
         Trie() {
             sz=;
             memset(ch,,sizeof ch);
             memset(val,,sizeof val);
             memset(cnt,,sizeof cnt);
         }
         void insert(int *s) {
             int x=,n=s[];
             for(int i=;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=,n=s[],ans=;
             for(int i=;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[];
     while(m--) {
         s[]=getint();
         for(int i=;i<=s[];i++) s[i]=getint();
         t.insert(s);
     }
     while(n--) {
         s[]=getint();
         for(int i=;i<=s[];i++) s[i]=getint();
         printf("%d\n",t.query(s));
     }
     return ;
 }