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; }