[CSP-S模拟测试]:sum(数学+莫队)
题目传送门(内部题63)
输入格式
第一行有一个整数$id$,表示测试点编号。
第一行有一个整数$q$,表示询问组数。
然后有$q$行,每行有两个整数$n_i,m_i$。
输出格式
一共有$q$行,每行一个整数表示每组询问的答案$S_{n_i,m_i}$对$10^9+7$取模的结果。
样例
样例输入:
1
5
1 1
2 1
3 2
4 3
5 5
样例输出:
2
3
7
15
32
数据范围与提示
对于所有数据,$1\leqslant q,n_i,m_i\leqslant 10^5$。
题解
考场上把$80$分部分分都水全了,愣是没想到莫队……
先来考虑所有询问的$n_i$相等应该怎么办,预处理即可,考虑$S_{n,m-1}$如何转移到$S_{n,m}$,无非就是加上$C_n^m$即可,不再赘述。
现在考虑所有询问的$m_i$相等应该怎么办,显然预处理没有那么简单,考虑$S_{n-1,m}$如何转移到$S_{n,m}$,既然组合数可以用杨辉三角推得,不妨画个杨辉三角。
为方便,我现在只画出杨辉三角中的其中两行为例
设$1$号点为$n-1$行的行首,$4$号点为$n$行的行首,利用杨辉三角的性质,编号为$4$的点等于编号为$1$的点,编号为$5$的点等于编号为$1$的点和编号为$2$的点的加和,编号为$6$的点等于编号为$2$的点和编号为$3$的点的加和。
还可以发现,在从$n-1$行向$n$行转移的时候除了$3$号点以外其它点都被加了$2$次,只有$3$号点只加了$1$次,那么我们可以得出$S_{n,m}=S_{n-1,m}*2-C_{n-1}^m$,同理$S_{n-1,m}=\frac{S_{n,m}+C_{n-1}^m}{2}$。
利用这个性质我们就可以解决这个子问题了。
得出了这些性质,我们可以考虑莫队算法,$m$相当于$l$,$n$相当于$r$,这道题就解决了。
时间复杂度:$\Theta(n\sqrt{n})$。
期望得分:$100$分。
实际得分:$100$分。
代码时刻
#include<bits/stdc++.h>
using namespace std;
struct rec{int n,m,id,pos;}e[100001];
const int mod=1000000007;
const int inx=500000004;
int q;
long long ans[100001];
long long jc[100001],inv[100001];
long long qpow(long long x,long long y)
{
long long res=1;
while(y)
{
if(y%2)res=res*x%mod;
y>>=1;
x=x*x%mod;
}
return res;
}
void pre_work()
{
jc[0]=1;
for(long long i=1;i<=100000;i++)
jc[i]=jc[i-1]*i%mod;
inv[100000]=qpow(jc[100000],mod-2);
for(int i=100000;i;i--)
inv[i-1]=inv[i]*i%mod;
}
long long get_C(long long x,long long y){return jc[x]*inv[y]%mod*inv[x-y]%mod;}
long long lucas(long long x,long long y)
{
if(!y)return 1;
return get_C(x%mod,y%mod)*lucas(x/mod,y/mod)%mod;
}
bool cmp(rec a,rec b){return (a.pos)^(b.pos)?a.m<b.m:(((a.pos)&1)?a.n<b.n:a.n>b.n);}
int main()
{
pre_work();int mxn=0;
scanf("%d%d",&q,&q);
for(int i=1;i<=q;i++)
{
scanf("%d%d",&e[i].n,&e[i].m);
mxn=max(mxn,e[i].n);e[i].id=i;
}
int t=sqrt(mxn);
for(int i=1;i<=q;i++)e[i].pos=(e[i].m-1)/t+1;
sort(e+1,e+q+1,cmp);
int m=0,n=0;
long long now=1;
for(int i=1;i<=q;i++)
{
while(n<e[i].n)now=(now*2%mod-lucas(n++,m)+mod)%mod;
while(m<e[i].m)now=(now+lucas(n,++m))%mod;
while(m>e[i].m)now=(now-lucas(n,m--)+mod)%mod;
while(n>e[i].n)now=(now+lucas(--n,m))*inx%mod;
ans[e[i].id]=now;
}
for(int i=1;i<=q;i++)printf("%lld\n",ans[i]);
return 0;
}
rp++