14. Juju and Binary String
题目链接:Juju and Binary String
感觉这题的思路和代码实现都比 E 题简单,不知道怎么评到的 2700 分。
给你一个长为 \(n\) 的 01 串,定义其 cuteness 为 1 的个数比上总长度。让你找 \(k\) 个子串,使它们的长度之和为 \(m\),且拼接后的 cuteness 与原串的相等。最小化 \(k\) 并输出方案。
设总共有 \(x\) 个 1,那么子串中需要 \(y=\dfrac{xm}{n}\) 个 1。显然,如果 \(y\) 不是整数,那么一定构造不出来。否则肯定能构造出来,因为直接拿 \(y\) 个 1 和 \(m-y\) 个 0 就是一种方案了。
然后就是最小化 \(k\),最理想的情况当然是存在一个长为 \(m\) 的子串正好有 \(y\) 个 1。第二个样例提示我们这种情况并不一定存在,但我们可以把原串连成一个环,如果我们选择的区间正好跨过了连起来的地方,\(k\) 就为 \(2\)。可以证明,在环上总是可以找到这样的一个子串。
先证明一个引理:环上两个相邻的长为 \(m\) 的子串内 1 的个数相差不会超过 \(1\)。
这是显然的,若第一个子串从 \(i\) 处开始,则当且仅当 \(s[i]\neq s[i+m]\) 时两个子串内 \(1\) 的个数才会发生变化,两者差的绝对值必为 \(1\)。
这就意味着各子串之间 \(1\) 的个数的变化是连续的,即如果存在两个子串,其中一个含有的 \(1\) 的个数小于 \(y\),另一个大于 \(y\),就必然存在一个子串中 \(1\) 的个数刚好是 \(y\)。
然后可以用反证法证明不可能所有子串内 \(1\) 的个数均小于或均大于 \(y\)。求和一下就行了。
那么将原串复制一遍后求个前缀和就可以找到这个子串了。
#include
using namespace std;
#define endl '\n'
using ll = long long;
const int maxn = 4e5 + 5;
int n, m;
char str[maxn];
int pre[maxn];
void solve() {
cin >> n >> m;
cin >> (str + 1);
int cnt = 0;
for (int i = 1; i <= n; ++i) {
if (str[i] == '1')
cnt++;
str[i + n] = str[i];
}
for (int i = 1; i <= 2 * n; ++i) {
pre[i] = pre[i - 1] + (str[i] == '1');
}
ll tot = 1LL * cnt * m;
if (tot % n != 0) {
cout << -1 << endl;
return;
}
tot /= n;
for (int i = m; i < n + m; ++i) {
if (pre[i] - pre[i - m] == tot) {
if (i <= n) {
cout << 1 << endl;
cout << i - m + 1 << ' ' << i << endl;
} else {
cout << 2 << endl;
cout << 1 << ' ' << i - n << endl;
cout << n - (m - (i - n)) + 1 << ' ' << n << endl;
}
break;
}
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T = 1;
cin >> T;
while (T--) {
solve();
}
}