题目链接:
[POJ 3007]Organize Your Train part II[Hash][Trie]
题意分析:
将字符串任意分割两半。可进行下列操作:1.互换两半的位置;2.将其中一个反转。每次操作次数不限。
问:给出的字符串,一共能产生多少个不同的字符串?
解题思路:
搞搞搞~先看看分成两半都有什么情况,发现总共有八种组合,我们用s串记录没有前后交换过位置的字符串,t串记录交换后的字符串,对八种情况一一hash即可。
这题用map惨遭TLE。所以使用Trie树进行映射,每次出现新的字符串的时候,++ans。
个人感受:
WA->TLE->RE->AC。曲折啊。WA是因为T串没有初始化= =我天,debug了40分钟之余。RE是Trie树开小了。
具体代码如下:
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<sstream>
#include<stack>
#include<string>
#define ll long long
#define pr(x) cout << #x << " = " << (x) << '\n';
using namespace std;
const int MAXN = 111;
char s[MAXN], t[MAXN];
int ans = 0;
struct Trie {
int next[50000][26], end[50000];
int root, L;
int newnode() {
for (int i = 0; i < 26; ++i) next[L][i] = -1;
end[L++] = 0;
return L - 1;
}
void init() {
L = 0;
root = newnode();
}
void insert(char buf[]) {
int len = strlen(buf);
int now = root;
for (int i = 0; i < len; ++i) {
int &x = next[now][buf[i] - 'a'];
if (x == -1)
x = newnode();
now = x;
}
end[now]++;
if (end[now] == 1) ++ans;
}
};
Trie ac;
int main()
{
int kase; scanf("%d", &kase);
while (kase --) {
scanf("%s", s);
ac.init();
ans = 0;
memset(t, 0, sizeof t);
int len = strlen(s);
for (int pos = 1; pos < len; ++pos) {
reverse(s, s + pos);
ac.insert(s); // 转+不转
reverse(s + pos, s + len);
ac.insert(s); // 转+转
reverse(s, s + pos);
ac.insert(s); // 不转+转
reverse(s + pos, s + len);
int cnt = 0;
for (int i = pos; i < len; ++i) t[cnt++] = s[i];
for (int i = 0; i < pos; ++i) t[cnt++] = s[i];
ac.insert(t); // 交换后啥都不干
reverse(t, t + len - pos);
ac.insert(t); // 转+不转
reverse(t + len - pos, t + len);
ac.insert(t); // 转+转
reverse(t, t + len - pos);
ac.insert(t); // 不转+转
}
printf("%d\n", ans);
}
return 0;
}