H 小w的糖果 题解
两个简单结论
给定单位操作的差分数组1 0 0 0 … 0.
结论1:对差分位及以后所有数加1的操作,可以由一次前缀和得到1 1 1 1 … 1.
结论2:对差分位及以后各数依次加1,2,…,n的操作,可以由两次前缀和得到1 2 3 4 … n.
两个进阶结论
对于单位操作的差分数组1 0 0 0 … 0.
结论3:该差分数组的三次前缀和数组的通项公式为.
证明:由等差数列的求和公式显然可得该结论.
结论4:该差分数组的四次前缀和数组的通项公式为.
证明:由结论3,该差分数组的四次前缀和数组的通项应为
.
一个重要结论
结论5:给定单位操作的差分数组1 1 0 0 … 0,对差分位及以后各数依次加的操作,可以对差分数组1 1 0 0 … 0求三次前缀和得到.
证明:设对某一差分数组做次前缀和可得目标数列
,则目标数列的通项公式为
.比对结论3和结论4,发现做三次前缀和得到二阶通项公式,做四次前缀和得到三阶通项公式,可知
.由结论3,标准差分数组做三次前缀和得到的数列通项公式为
,因此考虑构造令一通项公式
,使得
.而
,
,故数列
的原始差分数组为0 1 0 0 … 0,将数列
与数列
的差分数组叠加,得原始差分数组为1 1 0 0 … 0.
AC代码
#include<bits/stdc++.h> using namespace std; #define mem(x) memset(x,0,sizeof(x)) const int mod=1e9+7; const int maxn=1e5+5; int d1[maxn],d2[maxn],d3[maxn]; int main(){ int T; cin>>T; ios::sync_with_stdio(0); cin.tie(0),cout.tie(0); while(T--){ int n,m; cin>>n>>m; int type,pos; mem(d1),mem(d2),mem(d3); for(int i=1;i<=m;i++){ cin>>type>>pos; if(type==1)d1[pos]++; if(type==2)d2[pos]++; if(type==3)d3[pos]++,d3[pos+1]++; } for(int i=1;i<=n;i++){ d1[i]=(d1[i]+d1[i-1])%mod; d2[i]=(d2[i]+d2[i-1])%mod; d3[i]=(d3[i]+d3[i-1])%mod; } for(int i=1;i<=n;i++){ d2[i]=(d2[i]+d2[i-1])%mod; d3[i]=(d3[i]+d3[i-1])%mod; } for(int i=1;i<=n;i++){ d3[i]=(d3[i]+d3[i-1])%mod; } for(int i=1;i<=n;i++){ cout<<(d1[i]+d2[i]+d3[i])%mod<<" "; } cout<<endl; } return 0; }