CF - 712D 差值dp + 前缀和优化

题意：

A和B玩游戏，游戏一共分为t局，两个人各有一个初始分数a和b，每一局，两个人需要在[-k，+k]这个区间中任意选择一个整数，并且加到自己的分数中去，问游戏结束后，有多少种情况，A的分数可以严格大于B？（只要任意一局有至少一个人选择的分数不同两种，就算作不同情况）

思路：

dp进行计数，看到这种差值不是很大的关系，可以考虑dp的其中一维设置成两人分数的差值
设dp[i][j]表示到第i局为止，a-b的差值为j的不同情况数
因为每一局两个人选择的数都在[-k，+k]中，所以对于一局进行之后两个人差值的变化量不会超过[-2*k，+2*k]，dp[i][j]只能从dp[i-1][t]中转移过来，其中t的范围是[j-2*k，j+2*k]当然这里的范围要首先属于总的可能到达的范围。
通过观察可以发现，
变化量为-2k 到 2k时的方案数为 1， 2， 3，…2k+1，…3，2，1
所以状态转移方程为：dp[i][j] = dp[i-1][j-2*k] + dp[i-1][j-2*k+1]*2 + dp[i-1][j-2*k+2]*3 + … + (2*k+1)*dp[i-1][j] + … +dp[i-1][j+2*k-1]*2 + dp[i-1][j+2*k];
但是这样复杂度为O(t^2*k^2),会T
可以注意到我们这个方程有对称的特点，dp[i][j]等于dp[i][j-1]减掉前一半，加上后一半 。注意到方程中每项的系数，

dp[i][j] = 1 2 3 2 1 0 …
dp[i][j+1] = 0 1 2 3 2 1 …
可以观察到前半部分系数-1，后半部分系数+1
我们可以维护一个前缀和来进行计算，这样通过j到j+1只需要O(1)的时间，
代码有很多细节需要注意，调了很久。。
这样的优化套路在2015年南阳CCPC的H题（hdu5550）也出现过。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 5e5 + 10;
const ll MOD = 1e9 + 7;

ll dp[105][MAXN], pre[MAXN];

int main() {
    int a, b, k, t;
    scanf("%d%d%d%d", &a, &b, &k, &t);
    int mid = 2 * k * t, sum = 0;
    dp[0][mid] = 1;
    for (int i = 1; i <= 4 * k * t; i++)
        pre[i] = (pre[i - 1] + dp[0][i]) % MOD;
    for (int i = 1; i <= t; i++) {
        sum += 2 * k;
        for (int j = max(mid - sum - 2 * k, 0); j <= min(mid - sum + 2 * k, 4 * k * t); j++) {
            int tmp = 2 * k + 1 - abs(j - (mid - sum));
            dp[i][mid - sum] = (dp[i][mid - sum] + tmp * dp[i - 1][j] % MOD) % MOD;
        }
        for (int j = mid - sum + 1; j <= mid + sum; j++) {
            int t1 = max(j - 2 * k - 2, 0), t2 = min(j + 2 * k, 4 * k * t);
            dp[i][j] = (dp[i][j - 1] - (pre[j - 1] - pre[t1]) + MOD) % MOD;
            dp[i][j] = (dp[i][j] + (pre[t2] - pre[j - 1])) % MOD;
        }
        pre[0] = dp[i][0];
        for (int j = 1; j <= 4 * k * t; j++)
            pre[j] = (pre[j - 1] + dp[i][j]) % MOD;
    }
    ll ans = 0;
    for (int i = b - a + 1 + mid; i <= mid + sum; i++)
        ans = (ans + dp[t][i]) % MOD;
    printf("%I64d\n", ans);
    return 0;
}