这个题一开始我想的是最暴力的方法,先把所有单词建成字典树。。
然后两层循环查找两个字符串是否在字典树里。。结果可想而知。。TLE
后来还是没有想出来,最后看了一下DISCUSS。倒着来。。
枚举每个字符串,把它拆成两个串,复杂度就降低了。。
值得学习的思路。。
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <iostream>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <assert.h>
#include <time.h>
typedef long long LL;
const int INF = 500000001;
const double EPS = 1e-9;
const double PI = acos(-1.0);
using namespace std;
char ch[50001][200];
struct Trie
{
Trie *next[26];
bool flag;
};
Trie *H;
int idx(char c)
{
return c - 'a';
}
Trie *newnode()
{
Trie *p = new Trie;
for(int i = 0; i < 26; i++)
{
p->next[i] = NULL;
}
p->flag = false;
}
void insert(char *ss)
{
int len = strlen(ss);
Trie *p = H;
for(int i = 0; i < len; i++)
{
int c = idx(ss[i]);
if(p->next[c] == NULL)
{
p->next[c] = newnode();
}
p = p->next[c];
}
p->flag = true;
}
bool query(char *ss)
{
int len = strlen(ss);
Trie *p = H;
for(int i = 0; i < len; i++)
{
int c = idx(ss[i]);
if(p->next[c] == NULL)
{
return false;
}
p = p->next[c];
}
if(p->flag)
{
return true;
}
return false;
}
int main()
{
char cc[1000], dd[1000];
int cnt = 0;
H = newnode();
while(~scanf("%s", ch[cnt]))
{
insert(ch[cnt]);
cnt++;
}
for(int i = 0; i < cnt; i++)
{
int len = strlen(ch[i]);
for(int j = 1; j < len; j++)
{
strcpy(cc, ch[i]);
cc[j] = '\0';
strcpy(dd, ch[i]+j);
if(query(cc) && query(dd))
{
puts(ch[i]);
break;
}
}
}
return 0;
}