CodeForces 666A. Reberland Linguistics【DP】

题目链接

http://codeforces.com/problemset/problem/666/A

思路

题意就是给你一个字符串，删掉最前面5个字符，问剩下的字符串，从右往左不停拿掉长度为2或3的字符串，且不能连续两次拿相同的字符串，可以不拿完，问所有的拿法中，拿掉的字符串组成的集合是什么，字典序输出。

这题虽说代码形式不太像DP，但却用到了DP的思维。

从右往左推，设valid[i]表示存不存在一种方案，使得能将s[i]（包括）以后的字符全部拿掉，这个就可以递推了，每往左推一格，看valid[i+2]和valid[i+3]，如果为真，再往后搜一下相应的字符串，看有没有连续出现。要注意的是，就算连续出现了，如果存在一种方案使得长度不一样的话，也是合法的，例如pptpptt，搜ppt的时候，是连续出现的，但可以这么分：ppt|pp|tt，所以需要额外判断下valid[i+5]。一旦合法，就把valid[i]置为1，以此类推。

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <set>
using namespace std;

bool valid[10000 + 100];
set<string>ans;
int main()
{
    string s;
    cin >> s;
    valid[s.length()] = 1;
    for (int i = s.length() - 1; i>4; --i)
    {
        if (valid[i + 2])
        {
            string t = s.substr(i, 2);
            if (s.find(t, i + 2) != i + 2 || valid[i + 5])
            {
                valid[i] = 1;
                ans.insert(t);
            }
        }
        if (valid[i + 3])
        {
            string t = s.substr(i, 3);
            if (s.find(t, i + 3) != i + 3 || valid[i + 5])
            {
                valid[i] = 1;
                ans.insert(t);
            }
        }
    }
    cout << ans.size() << "\n";
    for (auto &ite : ans)
    {
        cout << ite << "\n";
    }
    return 0;
}