CF 526D Om Nom and Necklace

Om Nom and Necklace

题目背景：

CF 526D

分析：还是next数组的妙用，若n % (n - next[n])
== 0 则表示n为循环串，每一串都是长度为n
– next[n]，但是这个题每一位都要判断，具体的判定条件和判定思路，详见代码注释。

Source：

#include
#include
#include
#include
#include
#include
#include

using namespace std;

int p[1001000], n, k;
string s, t;
int cas;

void pre(string &s)
{
int len = s.size();
memset(p, 0, sizeof(p));
p[0] = -1;
int j = -1;
for(int i = 1; i < len; ++i)
{
while(j != -1 && s[i] != s[j + 1])
j = p[j];
if(s[i] == s[j + 1])
j++;
p[i] = j;
}
}

int main()
{
cin >> n >> k;
cin >> s;
pre(s);
cout << s << '\n';
for(int i = 0; i < s.size(); ++i) cout << p[i] << endl;
int lens = s.size();
for(int i = 0; i < lens; ++i)
{
int len = i + 1;
int x = len / (i - p[i]);
if(len % (i - p[i]) == 0) {
/*若每一段都是循环节*/
if(x / k >= x % k) cout << "1";
/*判定条件，若多余的，即不被分成k段的
部分(x % k)的个数小于分成k的段数(x / k)，则可以由
a = x / k, b = x % k ==>
x = a * k + b <=> x = b * (k + 1) + (a - b) * k;
从而有满足条件的A = b段循环节, B = (a - b)段循环节 */
else cout << "0";
} else {
/*若每一段不全是循环节，说明最后存在几个多余字母*/
if(x / k > x % k) cout << "1";
/*先同上判定然后考虑一个问题，我们可以把原来的每个A
后面直接加上多余字符，B去掉那些字符，即可，但是这就要保证B
不能为空，(多余字符少于一个循环节)，所以由上面推导可得*/
else cout << "0";
}
}
cout << '\n';
return 0;
}