[BZOJ3449] [Usaco2014 Feb]Secret Code
Description
Farmer John has secret message that he wants to hide from his cows; the message is a string of length at least 2 containing only the characters A..Z. To encrypt his message, FJ applies a sequence of "operations" to it, where an operation applied to a string S first shortens S by removing either some (but not all) of the initial characters or some (but not all) of the final characters from S, after which the original string S is attached either at the beginning or end. For example, a single operation to the string ABC could result in eight possible strings: AABC ABABC BCABC CABC ABCA ABCAB ABCBC ABCC Given the final encrypted string, please count the number of possible ways FJ could have produced this string using one or more repeated operations applied to some source string. Operations are treated as being distinct even if they give the same encryption of FJ's message. For example, there are four distinct separate ways to obtain AAA from AA. Print your answer out modulo 2014.
Input
* Line 1: A single encrypted string of length at most 100.
Output
* Line 1: The number of ways FJ could have produced this string with one or more successive operations applied to some initial string of length at least 2, written out modulo 2014. If there are no such ways, output zero.
Sample Input
Sample Output
OUTPUT DETAILS: Here are the different ways FJ could have produced ABABA:
1. Start with ABA -> AB+ABA
2. Start with ABA -> ABA+BA
3. Start with AB -> AB+A -> AB+ABA
4. Start with AB -> AB+A -> ABA+BA
5. Start with BA -> A+BA -> AB+ABA
6. Start with BA -> A+BA -> ABA+BA
7. Start with ABAB -> ABAB+A
8. Start with BABA -> A+BABA
#include <iostream> #include <cstdio> #include <map> #include <string> #include <cstring> using namespace std; #define reg register #define mod 2014 string str; int n; map <string, int> f; int dp(string s) { if (f.find(s) != f.end()) return f[s]; int L = s.length(); int res = 1; for (reg int l = 1 ; l * 2 < L ; l ++) { if (s.substr(0, l) == s.substr(l, l)) res = (res + dp(s.substr(l, L - l))) % mod; if (s.substr(0, l) == s.substr(L - l, l)) res = (res + dp(s.substr(l, L - l))) % mod; if (s.substr(L - l, l) == s.substr(0, l)) res = (res + dp(s.substr(0, L - l))) % mod; if (s.substr(L - l, l) == s.substr(L - l - l, l)) res = (res + dp(s.substr(0, L - l))) % mod; } return f[s] = res; } int main() { cin >> str; printf("%d\n", (dp(str) - 1 + mod) % mod); return 0; }