BZOJ2962: 序列操作

14 篇文章 0 订阅
14 篇文章 0 订阅

题目大意:给定n个数,要求支持区间加,区间取相反数,区间查询任意选c(c<=20)个数的所有方案中乘积的和

和维护k次方的和很像,想要维护选c个数,就要把选1~c个数的方案全部维护出来
这样当合并两个区间的时候(pushup),只需要枚举左右区间分别取了几个数即可

现在考虑两种修改操作:
1.区间取相反数
发现这种操作只会影响区间选奇数个数的情况,把这些答案全部变成相反数即可
2.区间加
这个就比较厉害了,首先我们考虑c=2的情况
设总个数为len,这些数分别为 a1,a2...alen ,则
i=1lenj=1i1(ai+v)(aj+v)=i=1lenj=1i1(aiaj+v2+(ai+aj)v)=i=1lenj=1i1aiaj+C2lenv2+C1len1i=1lenaiv
同理可求得c=3时的式子,所以只需要预处理出组合数,就可以递推了
总时间复杂度 O(MlogN202)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100010
using namespace std;
struct ppp{int a[21],len;};
int a[N];
int C[N][21];
int mod=19940417;
int l[N<<2],r[N<<2],t[N<<2];
bool rev[N<<2];
ppp w[N<<2];
ppp ret;
ppp operator +(const ppp &x,const ppp &y)
{
    ret.len=x.len+y.len;
    memset(ret.a,0,sizeof(ret.a));
    int i,j;
    for(i=0;i<=20;i++)
    for(j=0;j<=20-i;j++)
    ret.a[i+j]=((long long)x.a[i]*y.a[j]+ret.a[i+j])%mod;
    return ret;
}
void pup(int x){w[x]=w[x<<1]+w[x<<1|1];}
void pudrev(int x)
{
    int i;
    rev[x]^=1;
    t[x]=(mod-t[x])%mod;
    for(i=1;i<=19;i+=2)
    w[x].a[i]=(mod-w[x].a[i])%mod;
}
int ksm(int d,int c)
{
    int ret=1;
    while(c)
    {
        if(c&1) ret=(long long)ret*d%mod;
        d=(long long)d*d%mod;c/=2;
    }
    return ret;
}
void pudadd(int x,int v)
{
    t[x]=(t[x]+v)%mod;
    int i,j,l=w[x].len;
    for(i=min(20,l);i>=1;i--)
    for(j=i-1;j>=0;j--)
    w[x].a[i]=((long long)ksm(v,i-j)*C[l-j][i-j]%mod*w[x].a[j]+w[x].a[i])%mod;
}
void pud(int x)
{
    int i,j;
    if(rev[x])
    {
        pudrev(x<<1);
        pudrev(x<<1|1);
        rev[x]=0;
    }
    if(t[x])
    {
        pudadd(x<<1,t[x]);
        pudadd(x<<1|1,t[x]);
        t[x]=0;
    }
}
void build(int now,int ll,int rr)
{
    l[now]=ll;r[now]=rr;
    if(ll==rr)
    {
        w[now].len=1;
        w[now].a[0]=1;
        w[now].a[1]=a[ll];
        return;
    }
    int mid=(ll+rr)>>1;
    build(now<<1,ll,mid);
    build(now<<1|1,mid+1,rr);
    pup(now);
}
void change(int now,int ll,int rr,int v)
{
    if(l[now]==ll&&r[now]==rr)
    {
        pudadd(now,v);
        return;
    }
    pud(now);
    int mid=(l[now]+r[now])>>1;
    if(rr<=mid) change(now<<1,ll,rr,v);
    else if(ll>mid) change(now<<1|1,ll,rr,v);
    else change(now<<1,ll,mid,v),change(now<<1|1,mid+1,rr,v);
    pup(now);
}
void reverse(int now,int ll,int rr)
{
    if(l[now]==ll&&r[now]==rr)
    {
        pudrev(now);
        return;
    }
    pud(now);
    int mid=(l[now]+r[now])>>1;
    if(rr<=mid) reverse(now<<1,ll,rr);
    else if(ll>mid) reverse(now<<1|1,ll,rr);
    else reverse(now<<1,ll,mid),reverse(now<<1|1,mid+1,rr);
    pup(now);
}
ppp check(int now,int ll,int rr)
{
    if(ll==l[now]&&rr==r[now]) return w[now];
    pud(now);
    int mid=(l[now]+r[now])>>1;
    if(rr<=mid) return check(now<<1,ll,rr);
    else if(ll>mid) return check(now<<1|1,ll,rr);
    else return check(now<<1,ll,mid)+check(now<<1|1,mid+1,rr);
}
int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    int i,j,x,y,z;
    C[0][0]=1;
    for(i=1;i<=n;i++)
    {
        C[i][0]=1;
        for(j=1;j<=20;j++)
        C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
    }
    for(i=1;i<=n;i++)
    scanf("%d",&a[i]);
    build(1,1,n);
    char s[10];
    while(q--)
    {
        scanf("%s%d%d",s,&x,&y);
        if(s[0]=='I')
        {
            scanf("%d",&z);
            change(1,x,y,(z%mod+mod)%mod);
        }
        else if(s[0]=='R') reverse(1,x,y);
        else
        {
            scanf("%d",&z);
            printf("%d\n",check(1,x,y).a[z]);
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值