[bzoj4303]数列

传送门

这个出题人啊,真是想问候他全家!
不得了啊,模数是合数,诶,挺正常的!
没错挺正常的!
这个题算是思路比较简单的kd-tree了,打标记也和线段树差不多。
\(a_i\)是标号为\(i\)的权值,\(p_i\)为标号为\(i\)的标识符
然后转化为二维平面上的点\((i,p_i)\),然后上kd-tree就好了
(没有什么不正常的,会不会T就看你的常数了)
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=5e4+1,mod=(1<<29)-1;
int n,m,w,rt,ans;
struct oo{int d[2],mx[2],mn[2],sum,l,r,size,tag,pow,v;}s[4*maxn];
inline bool cmp(const oo&a,const oo&b){return a.d[w]<b.d[w];}
inline void update(int x)
{
    if(s[x].l)
    {
        s[x].mn[0]=min(s[s[x].l].mn[0],s[x].mn[0]);
        s[x].mn[1]=min(s[s[x].l].mn[1],s[x].mn[1]);
        s[x].mx[0]=max(s[s[x].l].mx[0],s[x].mx[0]);
        s[x].mx[1]=max(s[s[x].l].mx[1],s[x].mx[1]);
    }
    if(s[x].r)
    {
        s[x].mn[0]=min(s[s[x].r].mn[0],s[x].mn[0]);
        s[x].mn[1]=min(s[s[x].r].mn[1],s[x].mn[1]);
        s[x].mx[0]=max(s[s[x].r].mx[0],s[x].mx[0]);
        s[x].mx[1]=max(s[s[x].r].mx[1],s[x].mx[1]);
    }
    s[x].size=s[s[x].l].size+s[s[x].r].size+1;
}
inline void pushup(int x){s[x].sum=(s[x].v+s[s[x].l].sum+s[s[x].r].sum)&mod;}
inline int build(int l,int r,int d)
{
    int mid=(l+r)>>1;w=d;
    nth_element(s+l,s+mid,s+r+1,cmp);
    s[mid].mn[0]=s[mid].mx[0]=s[mid].d[0];
    s[mid].mn[1]=s[mid].mx[1]=s[mid].d[1];
    s[mid].pow=s[mid].size=1;
    if(l!=mid)s[mid].l=build(l,mid-1,!d);
    if(r!=mid)s[mid].r=build(mid+1,r,!d);
    update(mid);return mid;
}
inline void pushdown(int x)
{
    int ls=s[x].l,rs=s[x].r,a=s[x].pow,b=s[x].tag;
    if(ls)
    {
        s[ls].v=(1ll*s[ls].v*a+b)&mod,s[ls].sum=(1ll*s[ls].sum*a+1ll*s[ls].size*b)&mod;
        s[ls].tag=(1ll*s[ls].tag*a+b)&mod,s[ls].pow=(1ll*s[ls].pow*a)&mod;
    }
    if(rs)
    {
        s[rs].v=(1ll*s[rs].v*a+b)&mod,s[rs].sum=(1ll*s[rs].sum*a+1ll*s[rs].size*b)&mod;
        s[rs].tag=(1ll*s[rs].tag*a+b)&mod,s[rs].pow=(1ll*s[rs].pow*a)&mod;
    }
    s[x].pow=1,s[x].tag=0;
}
inline void change(int x,int l,int r,int ll,int rr,int a,int b)
{
    if(l<=s[x].mn[0]&&r>=s[x].mx[0]&&ll<=s[x].mn[1]&&rr>=s[x].mx[1])
    {
        s[x].v=(1ll*s[x].v*a+b)&mod,s[x].sum=(1ll*s[x].sum*a+1ll*s[x].size*b)&mod;
        s[x].tag=(1ll*s[x].tag*a+b)&mod,s[x].pow=(1ll*s[x].pow*a)&mod;
        return ;
    }
    else if(l<=s[x].d[0]&&r>=s[x].d[0]&&ll<=s[x].d[1]&&rr>=s[x].d[1])s[x].v=(1ll*s[x].v*a+b)&mod;
    if(s[x].tag)pushdown(x);
    if(s[x].l&&l<=s[s[x].l].mx[0]&&ll<=s[s[x].l].mx[1])change(s[x].l,l,r,ll,rr,a,b);
    if(s[x].r&&r>=s[s[x].r].mn[0]&&rr>=s[s[x].r].mn[1])change(s[x].r,l,r,ll,rr,a,b);
    pushup(x);
}
inline int get(int x,int l,int r,int ll,int rr)
{
    int ans=0;
    if(l<=s[x].mn[0]&&r>=s[x].mx[0]&&ll<=s[x].mn[1]&&rr>=s[x].mx[1])return s[x].sum;
    else if(l<=s[x].d[0]&&r>=s[x].d[0]&&ll<=s[x].d[1]&&rr>=s[x].d[1])ans=(1ll*ans+s[x].v)&mod;
    if(s[x].tag)pushdown(x);
    if(s[x].l&&l<=s[s[x].l].mx[0]&&ll<=s[s[x].l].mx[1])ans=(1ll*ans+get(s[x].l,l,r,ll,rr))&mod;
    if(s[x].r&&r>=s[s[x].r].mn[0]&&rr>=s[s[x].r].mn[1])ans=(1ll*ans+get(s[x].r,l,r,ll,rr))&mod;
    pushup(x);return ans;
}
signed main()
{
    read(n),read(m);
    for(rg int i=1;i<=n;i++)read(s[i].d[0]),s[i].d[1]=i;
    rt=build(1,n,0);
    for(rg int i=1,op,x,y,l,r;i<=m;i++)
    {
        read(op);
        if(!op)
        {
            read(l),read(r),read(x),read(y);
            change(rt,1,n,l,r,x,y);
        }
        if(op==1)
        {
            read(l),read(r),read(x),read(y);
            change(rt,l,r,1,n,x,y);
        }
        if(op==2)read(l),read(r),printf("%d\n",get(rt,1,n,l,r));
        if(op==3)read(l),read(r),printf("%d\n",get(rt,l,r,1,n));
    }
}

猜你喜欢

转载自www.cnblogs.com/lcxer/p/10262454.html