Codeforces 667C Reberland Linguistics【DFS】

一道卡题意的题。

题目链接：

http://codeforces.com/problemset/problem/667/C

题意：

一个串可以看成一个长度大于4的根，加上其后面的若干个相邻(in a row)不同的长度为2或3的子串组成。给定字符串，问你有多少种不同的子串，并按字典序输出。

分析：

很容易想到dfs的做法，注意剪枝的地方，只有在合理的情况下才标记这种情况。对于每个子串直接放到set里面就好了。

代码：

#include<cstring>
#include<set>
#include<iostream>
using namespace std;
const int maxn = 1e4 + 5;
string s;
set<string>res;
int tot = 0;
int vis[4][maxn];
void dfs(int cnt, int endd, int pa)
{
if(endd - cnt < 4||vis[cnt][endd]) return;
string t;
bool flg = true;
for(int i = 0; i < cnt ; i++){
t += s[endd - cnt + i + 1];
if(pa != cnt || (pa == cnt && s[endd - cnt + i + 1] != s[endd + i + 1])) flg = false;
}
if(!flg){
vis[cnt][endd] = true;
res.insert(t);
dfs(2, endd - cnt, cnt);
dfs(3, endd - cnt, cnt);
}
}
int main (void)
{
ios::sync_with_stdio(false);
cin>>s;
int len = s.length() ;
dfs(2, len - 1, 0);
dfs(3, len - 1, 0);
cout<<res.size()<<endl;
set<string>::iterator i;
for(i = res.begin(); i != res.end(); i++){
cout<<*i<<endl;
}
return 0;
}