hdu3724Encoded Barcodes(Trie tree)

题目请戳这里

题目大意：给n个字符串，给m个询问，每个询问给k个条形码。每个条形码由8个小码组成，每个小码有相应的宽度，已知一个条形码的宽度只有2种，宽的表示1，窄的表示0。并且宽的宽度是窄的宽度的2倍。由于扫描的时候有误差，每个小码的宽度为一个浮点型数据，保证每个数据的误差在5%内。所以一个条形码可以对应一个ASCC码，表示一个小写字母。k个条形码表示一个字符串s，每个询问表示给定的m个字符串中以s为前缀的字符串个数。

题目分析：将n个字符串插入到字典树中，并记录下每个前缀有多少个字符串。即每插入一个字符串，在相应路径上+1，然后就是模拟出字符串s，在字典树中查询即可。

关于条形码数据的处理：输入的时候记录下8个数据中的最大值，然后对于所有的数据，满足fabs(mx - data[i])/mx < 0.1的，那么第i为为1，否则0。

详情请见代码：

 

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
using namespace std;
const int N = 10005;
const int M = 2005;
const int NM = 10000005;
const double eps = 1e-6;
typedef __int64 ll;

struct node
{
    int cnt;
    int next[26];
}trie[NM];
int num,n,m;
double dt[9];
char str[M];
void init(int x)
{
    memset(trie[x].next,0,sizeof(trie[x].next));
    trie[x].cnt = 0;
}
void insert(int cur,int dp,int len)
{
    trie[cur].cnt ++;
    if(dp == len)
        return;
    if(trie[cur].next[str[dp] - 'a'] == 0)
    {
        trie[cur].next[str[dp] - 'a'] = num;
        init(num);
        num ++;
    }
    insert(trie[cur].next[str[dp] - 'a'],dp + 1,len);
}
int query(int cur,int dp,int len)
{
    if(dp == len)
        return trie[cur].cnt;
    if(trie[cur].next[str[dp] - 'a'])
        return query(trie[cur].next[str[dp] - 'a'],dp + 1,len);
    else
        return 0;
}
int main()
{
    int i,j,k;
    while(scanf("%d%d",&n,&m) != EOF)
    {
        num = 1;
        ll ans = 0;
        init(0);
        while(n --)
        {
            scanf("%s",str);
            int len = strlen(str);
            insert(0,0,len);
        }
        while(m --)
        {
            scanf("%d",&k);
            for(i = 0;i < k ;i ++)
            {
                double mx = 0;
                int code = 0;
                for(j = 0;j < 8;j ++)
                {
                    scanf("%lf",&dt[j]);
                    if(dt[j] - mx > eps)
                        mx = dt[j];
                }
                for(j = 0;j < 8;j ++)
                {
                    if(fabs(mx - dt[j])/mx - 0.1 < eps)
                        code += (1<<(7 - j));
                }
                str[i] = code;
            }
            str[k] = '\0';
            ans += query(0,0,k);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}