多校4 hdu 6333 Problem B. Harvest of Apples （莫队算法+组合数）

题目链接：哆啦A梦传送门

题意：有T组测试样例，每组为n,m，现在让你求  的值。

题解：刚开始看到这道题发现好像是个类似二项式公式，细看，不会。题解说是莫队算法。

参考博客：https://blog.csdn.net/qq_40731186/article/details/81359283

此博主写的很好。

我们能得到这两条式子，然后我们就开始莫队就行了。

莫队算法参考论文：百度百科

S(n,m)=S(n,m−1)+C(​​n m​​),

S(n,m)=2S(n−1,m)−C(​​n−1 m​​)

代码选择c++编译器，T，选择G++就不会。玄。

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>

using namespace std;
typedef long long LL;
const int maxn = 100005;
const int mod = 1000000007;

struct node
{
    int l,r,id;
    int ans;
}q[maxn];
int unit;
LL ans[maxn],sum;
LL fac[maxn],inv[maxn];
LL inv2;

bool cmp(const struct node& a,const struct node& b)
{
    if(a.l / unit == b.l / unit) return a.r < b.r;
    else return a.l / unit < b.l / unit;
}

LL fast_pow(LL a,int n)
{
    LL res = 1,base = a;
    while(n) {
        if(n & 1) res = res * base % mod;
        base = base * base % mod;
        n >>= 1;
    }
    return res;
}

LL Comb(int n,int k)  /// c(n, m) = n!/(m! * (n-m)!)
{
    return fac[n] * inv[k] % mod * inv[n - k] % mod;
}
void init()
{
    inv2 = fast_pow(2,mod - 2);
    fac[0] = fac[1] = 1;
    for(int i = 2; i < maxn; i++) {
        fac[i] = i * fac[i - 1] % mod;
    }
    inv[maxn - 1] = fast_pow(fac[maxn - 1],mod - 2);///费马小定理预处理阶乘逆元
    for(int i = maxn - 2; i >= 0; i--) {
        inv[i] = inv[i + 1] * (i + 1) % mod;
    }
}
void add_n(int L,int R)   /// S(n,m) = 2 * S(n-1,m) - C(n-1,m)
{
    sum = (2 * sum % mod - Comb(L - 1,R) + mod) % mod;
}
void sub_n(int L,int R)  /// S(n-1,m) = (S(n,m) + C(n-1,m)) / 2
{
    sum = (sum + Comb(L - 1,R)) % mod * inv2 % mod;
}
void add_m(int L,int R)  ///S(n,m)=S(n,m-1)+C(n,m);
{
    sum = (sum + Comb(L,R)) % mod;

}
void sub_m(int L,int R)  ///S(n,m-1)=S(n,m)-C(n,m);
{
    sum = (sum - Comb(L,R) + mod) % mod;
}
int main(void)
{
    int T;
    init();
    unit = sqrt(maxn * 1.0);///分块

    scanf("%d",&T);
    for(int i = 1; i <= T; i++) {
        scanf("%d %d",&q[i].l,&q[i].r);
        q[i].id = i;
    }
    sort(q + 1,q + T + 1,cmp);

    sum = 2;///C(1,0)+C(1,1)
    int L = 1,R = 1;

    for(int i = 1; i <= T; i++) {
        while(q[i].l > L) add_n(++L,R);
        while(q[i].l < L) sub_n(L--,R);
        while(q[i].r > R) add_m(L,++R);
        while(q[i].r < R) sub_m(L,R--);
        ans[q[i].id] = sum;
    }
    for(int i = 1; i <= T; i++) {
        printf("%lld\n",ans[i]);
    }
    return 0;
}