题解 AtCoder abc154 F

Posted 2021-03-10 whx1003

题目大意 令 (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}^rsum_{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}^rsum_{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);
}