CodeForces 666A :Reberland Linguistics DP

传送门

题目描述

给一个单词，分为前缀和后缀两部分，前缀长度不得小于5，后缀长度只能2或3，请输出后缀的个数和所有的后缀，要求所有后缀均不重复。

分析

半夜睡不着，感谢猫猫送来的题，让我……更清醒了
我们用f[i][j]表示第i位能否接上长度为j的后缀，如果f[i + 2][3] == 1，那么i ~ i + 2的字符串就可以作为后缀，然后如果长度相同的话判断一下后缀是否相同就可以了，最后用一个set去重排序即可

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const ll mod= 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a){char c=getchar();T x=0,f=1;while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}a=f*x;}
int gcd(int a,int b){return (b>0)?gcd(b,a%b):a;}
string str;
int f[N][2];
set<string> ans;

int main(){
    cin >> str;
    int n = str.length();
    if(n <= 6){
        puts("0");
        return 0;
    }
    for(int i = n - 1;i >= 5;i--){
        if(i + 2 == n) {
            f[i][0] = 1;
            continue;
        }
        if(i + 3 == n) {
            f[i][1] = 1;
            continue;
        }
        if(i + 2 < n){
            if(f[i + 2][1] == 1) f[i][0] = 1;
            if(f[i + 2][0] == 1 && ((str[i] != str[i + 2]) || (str[i + 1] != str[i + 3]))) f[i][0] = 1;
        }
        if(i + 3 < n){
            if(f[i + 3][0] == 1) f[i][1] = 1;
            if(f[i + 3][1] == 1 && ((str[i] != str[i + 3]) || (str[i + 1] != str[i + 4]) || (str[i + 2] != str[i + 5]))) f[i][1] = 1;
        }
    }
    for(int i = 5;i < n;i++){
        if(f[i][0]) ans.insert(str.substr(i,2));
        if(f[i][1]) ans.insert(str.substr(i,3));
    }
    cout << ans.size() << '\n';
    for(auto t:ans){
        cout << t << '\n';
    }
    return 0;
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/