hdu3724Encoded Barcodes(Trie tree)

 

hdu3724Encoded Barcodes(Trie tree)

分类： Trie tree 2013-10-07 21:47  83人阅读  评论(0)  收藏  举报

数据结构

题目请戳这里

题目大意：给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;
 }