【题解】洛谷P2418 yyy loves OI IV

感觉很是妙啊……这题数次误入歧途...最开始想的二维dp，单调队列优化；无果，卒。于是没忍住看了下标签：暴力枚举？搜索？于是开始想记忆化搜索。以为会有什么很强的剪枝之类的；30分，卒。最后终于回到正道上：50 0000的数据，只可能有O(n) & O(nlogn)两种复杂度吧？在这样的思想+标签线段树的指引下，总算是走向了光明。

暴力，正解的开端。首先考虑最开始的二维dp，转移方程为：dp[i] = min(dp[k] + 1) (k ∈ 1 ~ i - 1) , 且 i ~ k + 1为合法区间。大部分的时间消耗都在于枚举找最值+判断是否合法上。对于这部分的优化，我们先考虑一段合法的区间：要么相差 <= m, 要么都是一个人的粉丝。第二种情况明显特判就行，可以做到O(n), 暂时撇去不谈。再看第一种情况并列出式子：1. abs (a[i] - a[j - 1] - b[i] + b[j - 1]) <= m; 2. a[i] - b[i] - m <= a[j - 1] - b[j - 1] <= a[i] - b[i] + m. 到这里发现，可以用线段树维护区间的最值，将线段树建成 a[i] - b[i]的权值线段树，每次查询在满足条件的范围内的dp最小值就好了。注意要防止爆负数，加上一个大一点的数。

#include <bits/stdc++.h>
using namespace std;
#define INF 1061109567
#define maxn 600000
#define ADD 10000
int n, m, a[maxn], c[maxn], cont = INF, b[maxn], dp[maxn], ans = INF;
int N = ;

struct tree
{
    int l, r, num;
}T[maxn * ];

int read()
{
    int x = ;
    char c;
    c = getchar();
    while(c < '' || c > '') c = getchar();
    while(c >= '' && c <= '') x = x *  + c - '', c = getchar();
    return x;
}

void Build(int p, int l, int r)
{
    T[p].l = l, T[p].r = r, T[p].num = INF;
    if(l == r) return;
    int mid = (l + r) >> ;
    Build(p << , l, mid), Build(p <<  | , mid + , r);
}

void Getmin(int &x, int y)
{
    if(x > y) x = y;
}

void update(int p, int x, int num)
{
    if(T[p].l == T[p].r)
    {
        Getmin(T[p].num, num);
        return;
    }
    int mid = (T[p].l + T[p].r) >> ;
    if(x <= mid) update(p << , x, num);
    else update(p <<  | , x, num);
    T[p].num = min(T[p << ].num, T[p <<  | ].num);
}

int query(int p, int l, int r)
{
    int L = T[p].l, R = T[p].r;
    if(R < l || L > r) return INF;
    if(l <= L && r >= R) return T[p].num;
    return min(query(p << , l, r), query(p <<  | , l, r));
}

int main()
{
    n = read(), m = read();
    memset(dp, 0x3f3f3f, sizeof(dp));
    Build(, , N);
    for(int i = ; i <= n; i ++)
    {
        c[i] = read();
        a[i] = a[i - ] + (c[i] == );
        b[i] = b[i - ] + (c[i] == );
    }
    dp[] = ;
    update(, ADD, dp[]);
    for(int i = ; i <= n; i ++)
    {
        bool flag = false;
        if(c[i] == c[i - ]) dp[i] = cont + ;
        else flag = true;
        Getmin(dp[i], dp[i - ] + );
        int tem = query(, a[i] - b[i] - m + ADD, a[i] - b[i] + m + ADD);
        Getmin(dp[i], tem + );
        if(flag) cont = min(dp[i - ], dp[i]);
        else Getmin(cont, dp[i]);
        update(, a[i] - b[i] + ADD, dp[i]);
    }
    printf("%d\n", dp[n]);
    return ;
}