T525784 悬关求调
  • 板块题目总版
  • 楼主chenyuyang
  • 当前回复0
  • 已保存回复0
  • 发布时间2024/12/16 20:13
  • 上次更新2024/12/17 00:11:50
查看原帖
T525784 悬关求调
428150
chenyuyang楼主2024/12/16 20:13

跳转题面

我写的是线段树,维护每组磁铁的 x+x/3x+ \lfloor x/3 \rfloor ,区间修改、区间和查询,但是全WA。有没有大佬帮忙看看是哪里错了。

代码

#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[500005];

struct node
{
    long long s,tag;
    int s0,s1,s2;
}tree[2000005];
void add(int p,int l,int r,long long v)
{
    if(v==0)
    {
        return;
    }
    if(v>0)
    {
        tree[p].tag+=v;
        tree[p].s+=(v+v/3)*(r-l+1);
        if(v%3==1)
        {
            tree[p].s+=tree[p].s2;
            swap(tree[p].s1,tree[p].s2);
            swap(tree[p].s0,tree[p].s1);
        }
        if(v%3==2)
        {
            tree[p].s+=tree[p].s1+tree[p].s2;
            swap(tree[p].s0,tree[p].s1);
            swap(tree[p].s1,tree[p].s2);
        }
    }
    else
    {
        v=-v;
        tree[p].tag-=v;
        tree[p].s-=(v+v/3)*(r-l+1);
        if(v%3==1)
        {
            tree[p].s-=tree[p].s0;
            swap(tree[p].s0,tree[p].s1);
            swap(tree[p].s1,tree[p].s2);
        }
        if(v%3==2)
        {
            tree[p].s-=tree[p].s0+tree[p].s1;
            swap(tree[p].s1,tree[p].s2);
            swap(tree[p].s0,tree[p].s1);
        }
    }
}
void pushup(int p)
{
    tree[p].s=tree[p*2].s+tree[p*2+1].s;
    tree[p].s0=tree[p*2].s0+tree[p*2+1].s0;
    tree[p].s1=tree[p*2].s1+tree[p*2+1].s1;
    tree[p].s2=tree[p*2].s2+tree[p*2+1].s2;
}
void pushdown(int p,int l,int r)
{
    if(tree[p].tag)
    {
        int mid=(l+r)>>1;
        add(p*2,l,mid,tree[p].tag);
        add(p*2+1,mid+1,r,tree[p].tag);
        tree[p].tag=0;
    }
}
void build(int p,int l,int r)
{
    if(l==r)
    {
        tree[p].s=tree[p].s0=tree[p].s1=tree[p].s2=0;
        tree[p].s=a[l]+a[l]/3;
        if(a[l]%3==0)
        {
            tree[p].s0++;
        }
        if(a[l]%3==1)
        {
            tree[p].s1++;
        }
        if(a[l]%3==2)
        {
            tree[p].s2++;
        }
        return;
    }
    int mid=(l+r)>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    pushup(p);
}
void change(int p,int l,int r,int x,int y,int z)
{
    if((x<=l)&&(r<=y))
    {
        add(p,l,r,z);
        return;
    }
    if((y<l)||(r<x))
    {
        return;
    }
    int mid=(l+r)>>1;
    change(p*2,l,mid,x,y,z);
    change(p*2+1,mid+1,r,x,y,z);
    pushup(p);
}
long long query(int p,int l,int r,int x,int y)
{
    if((x<=l)&&(r<=y))
    {
        return tree[p].s;
    }
    if((y<l)||(r<x))
    {
        return 0;
    }
    pushdown(p,l,r);
    int mid=(l+r)>>1;
    return query(p*2,l,mid,x,y)+query(p*2+1,mid+1,r,x,y);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    build(1,1,n);
    while(m--)
    {
        int op;
        scanf("%d",&op);
        if(op==1)
        {
            int l,r,v;
            scanf("%d%d%d",&l,&r,&v);
            change(1,1,n,l,r,v);
        }
        else
        {
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%lld\n",query(1,1,n,l,r));
        }
    }
}
2024/12/16 20:13
加载中...