Educational Codeforces Round 170 (Rated for Div. 2) D题题解记录

大致题意:
有$a,b$两种属性，给定一个数组$c$，长度为$n$，其中有$m$个为$0$。$1\le m\le 5000;m<n\le 2\cdot 10^6$。如果$c_i=0$，那么可以选择一种属性，使其值加1，如果$c_i>0$，那么会有一次检查，如果$a>c_i$，那么通过检查。$c_i<0$的时候检查$b$属性，如果$b>\left|c_i\right|$，那么通过检查。
现在询问，在某种分配策略情况下，最多可以通过多少检查。
解
设$s$为截至目前为止遇到的总的分配点数。我们首先可以想到一个很朴素的$dp$：设$f[i][j]$表示前$i$个记录中有$j$个属性点分配给了$a$的前提下，通过的最多的检查数。那么可以得到转移方程：
$$c_i=0:f[i][j]=max(f[i-1][j],f[i-1][j-1])$$
$$c_i>0:f[i][j]=f[i-1][j]+1(j>=c_i),f[i][j]=f[i-1][j](j<c_i)$$
$$c_i<0:f[i][j]=f[i-1][j]+1(s+c_i>=j),f[i][j]=f[i-1][j](s+c_i<j)$$
但是这样的时间复杂度比较高。
考虑优化：我们发现，多数操作花在了第二种和第三种情况中的加法。而第一种情况只有$m$次，非常少。那么我们可以考虑用差分数组来代替原先的加法，而对于第一种情况，把差分数组的信息暴力传给$f$数组更新以后再转移方程。
我们维护一个$f[i]$的差分数组$d$，用来实现$O(1)$的区间加法操作。
每次遇到$c_i=0$的时候，我们把$d$的信息传给当前状态下的$f[i]$，然后再更新$f[i]$。

code

void solve() {
    int n, m;
    cin >> n >> m;
    int s = 0;
    vector<int> dp(m + 1);
    vector<int> cf(m + 2);
    for (int i = 1; i <= n; i++) {
        int x;
        cin >> x;
        if (x == 0) {
            //还原
            int temp = 0;
            for (int j = 0; j <= m; j++) {
                temp += cf[j];
                cf[j] = 0;
                if (j <= s)
                    dp[j] += temp;
            }
            for (int j = m; j >= 1; j--) {
                dp[j] = max(dp[j], dp[j - 1]);
            }
            s++;
        } else if (x > 0) {
            cf[x]++;
            cf[m + 1]--;

        } else {
            if (s + x < 0) continue;
            cf[0]++;
            cf[s + x + 1]--;
        }
    }
    int temp = 0;//注意这里，如果c数组最后全是检查，那么我们就没机会把差分数组信息传回
    //来了 所以必须在结束以后再更新一次
    for (int j = 0; j <= m; j++) {
        temp += cf[j];
        if (j <= s)
            dp[j] += temp;
    }

    int mi = 0;
    for (int i = 0; i <= m; i++)
        mi = max(mi, dp[i]);
    cout << mi << endl;

}