HDU3724 ( Encoded Barcodes ) 字典树

字典树模板，v记录当前节点下单词数，注意与当前节点相同（到当前节点为止）的单词的统计p->v-sumall_vnext

#include <iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<bitset>
using namespace std;
#define zero {0}
typedef struct Trie
{
    int v;
    Trie *next[26];
} Trie;
char s[100100][100];
Trie root;

void createTrie(char *str)
{
    int len = strlen(str);
    Trie *p = &root, *q;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'a';
        if(p->next[id] == NULL)
        {
            q = (Trie *)malloc(sizeof(root));
            q->v = 1;
            for(int j=0; j<26; ++j)
                q->next[j] = NULL;
            p->next[id] = q;
            p = p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p = p->next[id];
        }
    }
}

int findTrie(char *str)
{
    int len = strlen(str);
    int ret=0,x,y;
    Trie *p = &root;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'a';
        if (i!=0) x=p->v;
        else x=-1;
        p = p->next[id];
        y=p->v;
        if(i!=0&&x==y)
            ret++;


    }
    return len-ret;
}
int findnum(char *str)
{
    int len=strlen(str);
    Trie *p = &root;
    for(int i=0; i<len; i++)
    {
        int id=str[i]-'a';
        if(p->next[id])
        {
            p=p->next[id];
        }
        else
        {
            return 0;
        }
    }
    int sum=0;
   
    for(int i=0; i<26; i++)
    {
        if(p->next[i])
            sum+=p->next[i]->v;
    }
     sum=sum+p->v-sum;
    return sum;
}
int main()
{
    int i,n,sum,m;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=0; i<26; ++i)
            root.next[i] = NULL;
        getchar();
        for (i=1; i<=n; i++)
        {
            gets(s[i]);
            createTrie(s[i]);
        }
        sum=0;

        for(i=0;i<m;i++)
        {
            int zimushu;
            scanf("%d",&zimushu);
            char str[1005]=zero;
            double bit[10]=zero;
            for(int j=0;j<zimushu;j++)
            {
                double ave=0;
                bitset<32>b;
                for(int k=7;k>=0;k--)
                {
                    scanf("%lf",&bit[k]);
                    ave+=bit[k];
                }
                ave/=8;
                for(int k=7;k>=0;k--)
                {
                    if(bit[k]>ave)
                    {
                        b.set(k,1);
                    }
                    else
                    {
                        b.set(k,0);
                    }
                }
                str[j]=b.to_ulong();
              //  cout<<b.to_string()<<endl;
            }
          //  cout<<str<<endl;
            sum+=findnum(str);
        }
       // scanf("%s",str);
        // cout<<findnum(str)<<endl;
        printf("%d
",sum);
    }
    return 0;
}