题解 AtCoder abc154 F

题目大意 令 \(f(r,c)=C_{r+c}^r\)，要求输出 \(\sum_{i=r_1}^{r_2}\sum_{j=c_1}^{c_2} f(i,j)\)，其中 \(r_2,r_2,c_1,c_2\) 为给定值。

分析 令

\[g(r,c)=\sum_{i=0}^r\sum_{j=0}^c f(i,j)\]

则题目所求为

\[\sum_{i=r_1}^{r_2}\sum_{j=c_1}^{c_2} f(i,j)=g(r_2,c_2)-g(r_2,c_1-1)-g(r_1-1,c_2)+g(r_1-1,c_1-1)\]

如果我们能快速计算 \(g\) 则就解决了问题。

现在我们考虑如何用 \(f(r,\cdots)\) 求出 \(f(r+1,c)\)，我们有

\[f(r+1,c)-f(r,c)=C_{r+1+c}^{r+1}-C_{r+c}^r=\frac{(r+c+1)!}{(r+1)!c!}-\frac{(r+c)!}{r!c!}=\frac{(r+c+1)!-(r+c)!(r+1)}{(r+1)!c!}=\frac{(r+c)![(r+c+1)-(r+1)]}{(r+1)!c!}=\frac{(r+c)!}{(r+1)!(c-1)!}=f(r+1,c-1)\]

也就是说，我们有递归式

\[f(r+1,c)=f(r,c)+f(r+1,c-1)=f(r,c)+f(r,c-1)+f(r+1,c-2)=\cdots=f(r,c)+f(r,c-1)+f(r,c-2)+\cdots+f(r,0)=\sum_{i=0}^cf(r,i)\]

那么

\[g(r,c)=\sum_{i=0}^r\sum_{j=0}^c f(i,j)=\sum_{i=0}^rf(i+1,c)=\sum_{i=0}^rf(c,i+1)=f(c+1,r+1)-1\]

那么代码就很简单了。

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int mod = 1e+9 + 7;

ll r1, r2, c1, c2;

ll qPow(ll a, ll b)
{
    a %= mod;
    
    ll res = 1;
    while(b) {
        if(b & 1) res = res * a % mod;
        a = a * a % mod, b >>= 1;
    }
    return res;
}

ll C(ll n, ll m)
{
    if(n < m) return 0;
    
    ll res = 1;
    for(int i = 1; i <= m; ++i)
        res = res * (n - i + 1) % mod * qPow(i, mod - 2) % mod;
    return res;
}

int main()
{
    scanf("%lld%lld%lld%lld", &r1, &c1, &r2, &c2);
    
    ++r2, ++c2;
    printf("%lld", ((C(r2 + c2, c2) - C(r1 + c2, c2) - C(r2 + c1, c1) + C(r1 + c1, c1)) % mod + mod) % mod);
}