2018 Multi-University Training Contest 4 Problem B. Harvest of Apples

本文介绍了一种针对大量组合数求和问题的有效算法,并通过使用莫队算法进行优化,实现了快速计算C(n,0)+C(n,1)+...C(n,m)的方案。文章详细解释了算法原理,包括快速幂取模、逆元预处理等关键步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题意:

求T组 C(n,0) + C(n,1) +...C(n,m) . ( 1<=T,n,m<=1e5,)

题解:

代码:

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<math.h>
#define ls (o<<1)
#define rs ((o<<1)+1)
#define mid ((l+r)>>1)
using namespace std;
typedef long long int ll;
const int maxn = 100000;
const ll Mod = 1e9+7;
int n,m,T;
ll pow_mod(int x,int t)
{
    if(t==0) return 1;
    ll ans = pow_mod(x,t/2);
    ans = ans*ans%Mod;
    if(t%2) ans = ans*x%Mod;
    return ans;
}
ll fac[maxn+5],inv[maxn+5],ans[maxn+5];
int pos[maxn+5],chunk;
struct Node
{
    int n,k,i;
} Q[maxn+5];
ll C(int a,int b)
{
    return 1ll * fac[a] * inv[b] % Mod * inv[a - b] % Mod;
}
bool cmp(Node n1,Node n2)
{
    if(pos[n1.n]==pos[n2.n]) return n1.k < n2.k;
    return n1.n < n2.n;
}
int main()
{
    fac[0] = 1;
    for(int i=1; i<=maxn; i++) fac[i] = fac[i-1]*i%Mod;
    inv[maxn] = pow_mod(fac[maxn],Mod-2);
    for(int i=maxn-1; i>=0; i--) inv[i] = inv[i+1]*(i+1)%Mod;
    chunk = sqrt(maxn);
    for(int i=1; i<=maxn; i++) pos[i] = i/chunk + 1;
    int T;
    scanf("%d",&T);
    for(int _=1; _<=T; _++)
    {
        scanf("%d %d",&Q[_].n,&Q[_].k), Q[_].i=_;
    }
    sort(Q+1,Q+T+1,cmp);
    int l = Q[1].n,r = -1;
    ll val = 0;
    ll inv2 = pow_mod(2,Mod-2);
    ///莫队基本 在知道 (l,r) 的答案后 能够 向 (l+1,r),(l-1,r),(l,r+1),(l,r-1) O1 时间的转移。
    for(int i=1; i<=T; i++)
    {
        if(pos[Q[i].n]!=pos[Q[i-1].n]) l = Q[i].n , r = -1, val = 0;
        while(l<Q[i].n) val = (0ll + val + val + Mod - C(l++, r)) % Mod;
        while(l>Q[i].n) val = (val + C(--l,r))%Mod*inv2%Mod;
        while(r<Q[i].k) val = (val + C(l, ++ r)) % Mod;
        //while(r>Q[i].k) val = (val + Mod - C(l, r--)) % Mod;
        ans[Q[i].i] = val;
    }
    for(int i=1; i<=T; i++) printf("%lld\n",ans[i]);
    return 0;
}
/**
写了if(pos[Q[i].n]!=pos[Q[i-1].n]) l = Q[i].n , r = -1, val = 0;就可以不写 while(r>Q[i].k) val = (val + Mod - C(l, r--)) % Mod;
因为排序时 相同块 k 已经排好了(同理按k分块也是一样的)。
 for(int i=1; i<=T; i++)
    {
        if(pos[Q[i].n]!=pos[Q[i-1].n]) l = Q[i].n , r = -1, val = 0;
        while(l<Q[i].n) val = (0ll + val + val + Mod - C(l++, r)) % Mod;
        while(l>Q[i].n) val = (val + C(--l,r))%Mod*inv2%Mod;
        while(r<Q[i].k) val = (val + C(l, ++ r)) % Mod;
        //while(r>Q[i].k) val = (val + Mod - C(l, r--)) % Mod;
        ans[Q[i].i] = val;
    }
**/

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值