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。

详情请见代码：

[cpp]  view plain copy

1. #include <iostream>  
2. #include<cstdio>  
3. #include<cstring>  
4. #include<cmath>  
5. #include<algorithm>  
6. #include<string>  
7. using namespace std;  
8. const int N = 10005;  
9. const int M = 2005;  
10. const int NM = 10000005;  
11. const double eps = 1e-6;  
12. typedef __int64 ll;  
13.   
14. struct node  
15. {  
16.     int cnt;  
17.     int next[26];  
18. }trie[NM];  
19. int num,n,m;  
20. double dt[9];  
21. char str[M];  
22. void init(int x)  
23. {  
24.     memset(trie[x].next,0,sizeof(trie[x].next));  
25.     trie[x].cnt = 0;  
26. }  
27. void insert(int cur,int dp,int len)  
28. {  
29.     trie[cur].cnt ++;  
30.     if(dp == len)  
31.         return;  
32.     if(trie[cur].next[str[dp] - 'a'] == 0)  
33.     {  
34.         trie[cur].next[str[dp] - 'a'] = num;  
35.         init(num);  
36.         num ++;  
37.     }  
38.     insert(trie[cur].next[str[dp] - 'a'],dp + 1,len);  
39. }  
40. int query(int cur,int dp,int len)  
41. {  
42.     if(dp == len)  
43.         return trie[cur].cnt;  
44.     if(trie[cur].next[str[dp] - 'a'])  
45.         return query(trie[cur].next[str[dp] - 'a'],dp + 1,len);  
46.     else  
47.         return 0;  
48. }  
49. int main()  
50. {  
51.     int i,j,k;  
52.     while(scanf("%d%d",&n,&m) != EOF)  
53.     {  
54.         num = 1;  
55.         ll ans = 0;  
56.         init(0);  
57.         while(n --)  
58.         {  
59.             scanf("%s",str);  
60.             int len = strlen(str);  
61.             insert(0,0,len);  
62.         }  
63.         while(m --)  
64.         {  
65.             scanf("%d",&k);  
66.             for(i = 0;i < k ;i ++)  
67.             {  
68.                 double mx = 0;  
69.                 int code = 0;  
70.                 for(j = 0;j < 8;j ++)  
71.                 {  
72.                     scanf("%lf",&dt[j]);  
73.                     if(dt[j] - mx > eps)  
74.                         mx = dt[j];  
75.                 }  
76.                 for(j = 0;j < 8;j ++)  
77.                 {  
78.                     if(fabs(mx - dt[j])/mx - 0.1 < eps)  
79.                         code += (1<<(7 - j));  
80.                 }  
81.                 str[i] = code;  
82.             }  
83.             str[k] = '\0';  
84.             ans += query(0,0,k);  
85.         }  
86.         printf("%I64d\n",ans);  
87.     }  
88.     return 0;  
89. }