codeforces 712D

题目大意

两个人玩游戏,第一个人最开始分数是 a ,第二个人最开始分数是 b ,总共玩 t 轮，每一轮两人会随机的在区间 [−k,k] 共 2k+1 个数中随机拿一个，并加到自己的分数中。
问第一个人最后赢的概率是多少。
将概率乘 (2k+1)t 以保证输出的是一个整数，并将整数 mod109+7 .

解题思路

设 fi,j 表示已经玩了 i 轮，第一个人比第二个人多 j 分的概率。
易得，两个人都是从 [−k,k] 取数，所以相当于游戏玩 2t 轮，每一轮中 fi,j 都会对 fi+1,j′,−k≤j−j′≤k 贡献。
那么我们枚举 j ,并在 fi+1,j−k 处打+1标记，在 fi+1,j+k+1 处打-1标记，即可前缀和求出 fi+1 .

参考代码

#include
#include
#include
#include
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define maxk 1005
#define maxt 105
#define maxn (maxk * maxt * 4+505)
#define add 200200
#define mx (add * 2)
#define mo 1000000007
#define max(a,b) (((a) > (b)) ? a : b)
#define min(a,b) (((a) < (b)) ? a : b)
using namespace std;

int f[2][maxn],sum[maxn];

int a,b,k,t;

int now;

int main(){
    cin>>a>>b>>k>>t;
    f[now][a-b+add]=1;
    f[now][a-b+add+1]=mo-1;
    fo(i,1,2*t) {
        int ad=0;
        fo(j,0,mx) {
            ad=(ad+f[now][j]) % mo;
            f[now][j]=0;
            f[now ^ 1][max(j-k,0)]=(f[now ^ 1][max(j-k,0)]+ad) % mo;
            f[now ^ 1][min(j+k+1,mx)]=(f[now ^ 1][min(j+k+1,mx)]-ad+mo) % mo;
        }
        now^=1;
    }
    int ad=0,ans=0;
    fo(i,0,mx) {
        ad=(ad+f[now][i]) % mo;
        if (i>add) ans=(ans+ad) % mo;
    }
    cout<return 0;
}