Longest Y 字符串，货仓选址模型（600）

题意 ：

• 给一个Y.串，每次操作可以交换相邻的两个字符，问在操作若干次$([0,k])$后，字符串中连续的Y最多是多少

思路 ：

• 设定a数组表示原字符串中每个y出现的下标，从左到右
• 设定b数组为$b[i]=a[i]-i$，b数组代表将字符串中所有的y全部按顺序交换到原字符串的前面，形成一段完整的y字符串，每个原字符串中的y需要交换的次数
• 那么题意就变换为 ：$(1)$每次操作可以对b数组某个元素+1或者-1，最多k次操作，问b数组中最多有几个相同的元素
• 由于这个答案有单调函数性质$monotonicity$，我们可以使用二分
• 考虑下列猜想 ：$(2)$每次操作可以对b数组某个元素+1或者-1，最多k次操作，是否可以得到b数组中k个相同的元素？
• $(2)$，可以转换为我们需要多少次操作才可以使得所有的$b_i,b_{i+1},...,b_{i+m-1}$相等，如果这个操作数小于等于k，说明这一段可以，根据这个来更新二分区间
• 也就是说，找到下列柿子的最小值（其中x是变成的相同元素） ：$(3)$ $$f(x)=\sum _{k=i}^{i+m-1}|B_k-x|$$
• 这样就变成了一个经典模型 - 货仓选址，结论是当x取$B_{i+\lfloor \frac{m}{2}\rfloor }$时，$f(x)$取得最小值

#include <bits/stdc++.h>
using namespace std;
const int maxn = 200005;
typedef long long ll;
int n; ll m;
char s[maxn];
int pos[maxn], N;
ll pre[maxn], suf[maxn];
bool check(int Len) {
    ll val = -1ll * (Len / 2) * (Len / 2 + 1);
    if (Len & 1 ^ 1) val += Len / 2;
    for (int i = 1; i + Len - 1 <= N; i++)
        if (val + pre[i + Len / 2 - 1] - pre[i - 1] + suf[i + Len / 2 + (Len & 1)] - suf[i + Len] <= m)
            return 1;
    return 0;
}
int main() {
    #ifdef LOCAL
    freopen("data.in", "r", stdin);
    freopen("pro.out", "w", stdout);
    #endif
    scanf("%s%lld", s + 1, &m);
    n = strlen(s + 1);
    for (int i = 1; i <= n; i++)
        if (s[i] == 'Y') pos[++N] = i;
    for (int i = N; i; i--) suf[i] = suf[i + 1] + pos[i];
    for (int i = 1; i <= N; i++) pre[i] = pre[i - 1] - pos[i];
    int l = 0, r = N + 1, mid;
    while (l + 1 < r) {
        mid = (l + r) >> 1;
        if (check(mid)) l = mid;
        else r = mid;
    }
    printf("%d\n", l);
    return 0;
}