bzoj 2962 序列操作——线段树(卷积?)

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2962

如果 _,_,_,…… 变成了 (_+k),(_+k),(_+k),…… ,计算就是在每个括号里选 _ 或 k ,乘起来求和。

为了算那个,枚举选了 j 个 k ;剩下那部分的乘积就是sm[cr][ i-j ]!j 和 k 可以在 len 里除了那 i-j 个位置里选,所以乘上 k^j 再乘上 C( len-(i-j) , j ) 。

调了2h+竟然只因组合数推导公式写错……

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=5e4+,Lm=,mod=;
int n,q,t[N],tot,c[N][Lm+],ans[Lm+],tmp[Lm+];
int ls[N<<],rs[N<<],add[N<<],sm[N<<][Lm+];
bool rev[N<<];
char ch;
int rdn()
{
int ret=;bool fx=;char ch=getchar();
while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
return fx?ret:-ret;
}
void upd(int &x){x-=(x>=mod?mod:);}
void init()
{
c[][]=;
for(int i=;i<=n;i++)//n!
for(int j=;j<=i&&j<=Lm;j++)
{
c[i][j]=c[i-][j];
if(j)c[i][j]+=c[i-][j-],upd(c[i][j]);//i-1
}
}
void pshp(int cr,int l,int mid,int r)
{
int lm=min(r-l+,Lm),Ls=ls[cr],Rs=rs[cr];
for(int i=;i<=lm;i++)
{
sm[cr][i]=;
for(int j=;j<=i;j++)
sm[cr][i]=(sm[cr][i]+(ll)sm[Ls][j]*sm[Rs][i-j])%mod;
}
// printf("pshp l=%d r=%d\n",l,r);
// for(int i=0;i<=lm;i++)
// printf(" sm[%d][%d]=%d\n",cr,i,sm[cr][i]);
}
void updt(int cr,int k,int len)
{
add[cr]+=k; upd(add[cr]);
int lm=min(Lm,len); for(int i=lm;i>=;i--)
for(int j=,ml=k;j<=i;j++,ml=(ll)ml*k%mod)//j=1 //j<=i not j<=lm
{
sm[cr][i]=(sm[cr][i]+(ll)sm[cr][i-j]*ml%mod*c[len-i+j][j])%mod;
// if(cr==3&&i==2)
// printf("smi=%d j=%d sm[i-j]=%d ml=%d c=[%d][%d]=%d\n",sm[cr][i],j,sm[cr][i-j],ml,len-i+j,j,c[len-i+j][j]);
} // printf("updt\n");
// for(int i=0;i<=lm;i++)
// printf(" sm[%d][%d]=%d\n",cr,i,sm[cr][i]);
}
void updr(int cr,int lm)
{
if(add[cr])add[cr]=mod-add[cr]; rev[cr]^=;//add
for(int i=;i<=lm;i+=)
if(sm[cr][i]) sm[cr][i]=mod-sm[cr][i]; // printf("updr\n");
// for(int i=0;i<=lm;i++)
// printf(" sm[%d][%d]=%d\n",cr,i,sm[cr][i]);
}
void pshd(int cr,int l,int mid,int r)
{
int Ls=ls[cr],Rs=rs[cr];
if(rev[cr])
{
rev[cr]=;
updr(Ls,min(mid-l+,Lm));
updr(Rs,min(r-mid,Lm));
}
if(add[cr])
{
int k=add[cr]; add[cr]=;
updt(Ls,k,mid-l+); updt(Rs,k,r-mid);
}
}
void build(int l,int r,int cr)
{
if(l==r){sm[cr][]=t[l];sm[cr][]=;return;}
int mid=l+r>>;
ls[cr]=++tot; build(l,mid,ls[cr]);
rs[cr]=++tot; build(mid+,r,rs[cr]);
pshp(cr,l,mid,r);
}
void mdfy(int l,int r,int cr,int L,int R,int v)
{
if(l>=L&&r<=R)
{
// printf("mdf: cr=%d l=%d r=%d\n",cr,l,r);
updt(cr,v,r-l+); return;
}
int mid=l+r>>; pshd(cr,l,mid,r);
if(L<=mid) mdfy(l,mid,ls[cr],L,R,v);
if(mid<R) mdfy(mid+,r,rs[cr],L,R,v);
pshp(cr,l,mid,r);
}
void revs(int l,int r,int cr,int L,int R)
{
if(l>=L&&r<=R)
{
// printf("rev: cr=%d l=%d r=%d\n",cr,l,r);
updr(cr,min(r-l+,Lm));
return;
}
int mid=l+r>>; pshd(cr,l,mid,r);
if(L<=mid) revs(l,mid,ls[cr],L,R);
if(mid<R) revs(mid+,r,rs[cr],L,R);
pshp(cr,l,mid,r);
}
void query(int l,int r,int cr,int L,int R,int k)
{
if(l>=L&&r<=R)
{
int lm=min(k,r-l+);
for(int i=k;i>=;i--)
for(int j=;j<=lm&&j<=i;j++)//j<=i
ans[i]=(ans[i]+(ll)sm[cr][j]*ans[i-j])%mod;
return;
} int mid=l+r>>; pshd(cr,l,mid,r);
if(L<=mid) query(l,mid,ls[cr],L,R,k);
if(mid<R) query(mid+,r,rs[cr],L,R,k);
}
int main()
{
// freopen("data.in","r",stdin);
n=rdn(); q=rdn(); init();
for(int i=;i<=n;i++) t[i]=rdn()%mod+mod,upd(t[i]);
tot=; build(,n,); ans[]=;
for(int i=,a,b,c;i<=q;i++)
{
cin>>ch;scanf("%d%d",&a,&b);
if(ch!='R')scanf("%d",&c);
if(ch=='I')
{
c=c%mod+mod; upd(c);
mdfy(,n,,a,b,c);
}
if(ch=='R') revs(,n,,a,b);
if(ch=='Q')
{
for(int i=;i<=c;i++)ans[i]=;
query(,n,,a,b,c);
printf("%d\n",ans[c]);
}
}
return ;
}
上一篇:dedecms 采集规则过滤与替换


下一篇:UNIX环境高级编程——线程与进程区别