hdu 1247 Hat’s Words (trie树)
题意：给出n个字符串的单词表，输出所有的字符串a，满足字符串a是由n中另外两个字符串拼接成的。
思路：字典树。。其实我一开始想出了正解。。。。就是分割一个单词然后分别在trie上查找。。。但是由于题目坑爹得没给单词的长度这个数据范围。。并不是很敢写2333。。。看了下题解发现就是这么做。。。然后写了下1A。。。
不给数据范围玩个鸟啊。。。
1/* ***********************************************
2Author :111qqz
3Created Time :2016年08月16日 星期二 00时08分01秒
4File Name :code/hdu/1247.cpp
5************************************************ */
6#include <cstdio>
7#include <cstring>
8#include <iostream>
9#include <algorithm>
10#include <vector>
11#include <queue>
12#include <stack>
13#include <set>
14#include <map>
15#include <string>
16#include <cmath>
17#include <cstdlib>
18#include <deque>
19#include <ctime>
20#define fst first
21#define sec second
22#define lson l,m,rt<<1
23#define rson m+1,r,rt<<1|1
24#define ms(a,x) memset(a,x,sizeof(a))
25typedef long long LL;
26#define pi pair < int ,int >
27#define MP make_pair
28using namespace std;
29const double eps = 1E-8;
30const int dx4[4]={1,0,0,-1};
31const int dy4[4]={0,-1,1,0};
32const int inf = 0x3f3f3f3f;
33const int N = 5E4+7;
34char s[N][200];
35int n ;
36struct Trie
37{
38 struct Node
39 {
40 Node *nxt[26];
41 bool ok; //标记单词的结尾
42 Node()
43 {
44 for ( int i = 0 ; i < 26 ; i++) nxt[i] = NULL;
45 ok = false;
46 }
47 };
48 Node *root;
49 void init()
50 {
51 root = new Node();
52 }
53 Trie()
54 {
55 root = new Node();
56 }
57 void Insert(char *s)
58 {
59 Node *u = root;
60 int len = strlen(s);
61 for ( int i = 0 ; i < len ; i++)
62 {
63 int v = s[i]-'a';
64 if (u->nxt[v]==NULL) u->nxt[v] = new Node();
65 u = u->nxt[v];
66 }
67 u->ok = true;
68 }
69 bool Find(char *s)
70 {
71 Node *u = root;
72 int len = strlen(s);
73 for ( int i = 0 ; i < len ; i++)
74 {
75 int v = s[i]-'a';
76 if (u->nxt[v]==NULL) return 0;
77 u=u->nxt[v];
78 }
79 return u->ok;
80 }
81}trie;
82int main()
83{
84 #ifndef ONLINE_JUDGE
85 freopen("code/in.txt","r",stdin);
86 #endif
87 n = 0 ;
88 trie.init();
89 while (~scanf("%s",s[++n]))
90 {
91 trie.Insert(s[n]);
92 }
93 n--;
94 char a[200],b[200];
95 for ( int i = 1 ; i <= n ; i++)
96 {
97 int len = strlen(s[i]);
98 // cout<<i<<" s:"<<s[i]<<endl;
99 for ( int j = 1 ; j < len ; j++) //枚举切割的位置
100 {
101 ms(a,0);
102 ms(b,0);
103 strncpy(a,s[i],j);
104 strncpy(b,s[i]+j,len);
105 // cout<<"a:"<<a<<" b:"<<b<<endl;
106 if (trie.Find(a)&&trie.Find(b))
107 {
108 printf("%s\n",s[i]);
109 break;
110 }
111 }
112 }
113 #ifndef ONLINE_JUDGE
114 fclose(stdin);
115 #endif
116 return 0;
117}