社区讨论

样例已过 0pts,求调

P3373【模板】线段树 2参与者 2已保存回复 2

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
2 条
当前快照
1 份
快照标识符
@m2ir5kpt
此快照首次捕获于
2024/10/21 16:26
去年
此快照最后确认于
2025/11/04 16:37
4 个月前
查看原帖
bro刚学线段树想尝试自己打一遍代码
CPP
#include <cstdio>
#define int long long
using namespace std;
const int N=1e5+10;
int mod;
int tree[N<<2];
int tag1[N<<2],tag2[N<<2];  
int a[N];
int n,q;
int LS(int x) {return x<<1;}
int RS(int x) {return x<<1|1;}
void addtag1(int l,int r,int i,int k)  //写的时候忘了
{
    if(!tag1[i]) tag1[i]=k;
    else tag1[i]*=k;
    tree[i]=(k%mod)*(tree[i]%mod);//相当于总和也乘K
}
void addtag2(int l,int r,int i,int k)
{
    tag2[i]+=k;
    tree[i]+=k*(r-l+1)%mod;
    tree[i]%=mod;
}
void push_up(int i)
{
    tree[i]=(tree[LS(i)]+tree[RS(i)])%mod;
}
void push_down(int l,int r,int i)
{
    int mid=l+r>>1;
    if(tag1[i])
    {
        addtag1(l,mid,LS(i),tag1[i]);  //第三个参数序号开始错了
        addtag1(mid+1,r,RS(i),tag1[i]);
        tag1[i]=0;  //别忘清零
    }
    if(tag2[i])
    {
        addtag2(l,mid,LS(i),tag2[i]);
        addtag2(mid+1,r,RS(i),tag2[i]);
        tag2[i]=0;
    }
}
void build(int l,int r,int i)
{
    if(l==r)
    {
        tree[i]=a[l];
        return;
    }
    int mid=l+r>>1;
    build(l,mid,LS(i));
    build(mid+1,r,RS(i));
    push_up(i);
}
int query(int L,int R,int pl,int pr,int p)
{
    if(L<=pl && pr<=R) return tree[p]%mod;
    push_down(pl,pr,p);
    int mid=pl+pr>>1;
    int ans=0;
    if(L<=mid) ans+=query(L,R,pl,mid,LS(p))%mod;
    if(R>mid) ans+=query(L,R,mid+1,pr,RS(p))%mod;
    return ans%mod;
}
void update1(int L,int R,int pl,int pr,int p,int k)
{
    if(L<=pl && pr<=R)
    {
        addtag1(pl,pr,p,k);
        return;
    }
    int mid=pl+pr>>1;
    push_down(pl,pr,p);//别忘了先传tag
    if(L<=mid) update1(L,R,pl,mid,LS(p),k);
    if(R>mid) update1(L,R,mid+1,pr,RS(p),k);
    push_up(p);//别忘了把值传上来!!
}
void update2(int L,int R,int pl,int pr,int p,int k)
{
    if(L<=pl && pr<=R)
    {
        addtag2(pl,pr,p,k);
        return;
    }
    int mid=pl+pr>>1;
    push_down(pl,pr,p);//别忘了先传tag
    if(L<=mid) update2(L,R,pl,mid,LS(p),k);
    if(R>mid) update2(L,R,mid+1,pr,RS(p),k);
    push_up(p);//别忘了把值传上来
}
signed main()
{
    scanf("%lld%lld%lld",&n,&q,&mod);
    for(int i=1;i<=n;i++) scanf("%lld",a+i);
    build(1,n,1);
    int op,x,y,k;
    while(q--)
    {
        scanf("%lld",&op);
        switch(op)
        {
            case 1:
            {
                scanf("%lld%lld%lld",&x,&y,&k);
                update1(x,y,1,n,1,k);
            }
            break;
            case 2:
            {
                scanf("%lld%lld%lld",&x,&y,&k);
                update2(x,y,1,n,1,k);
            }
            break;
            case 3:
            {
                scanf("%lld%lld",&x,&y);
                printf("%lld\n",query(x,y,1,n,1));
            }
        }
    }
    return 0;
}

回复

2 条回复,欢迎继续交流。

正在加载回复...