模板求调

P3372 【模板】线段树 1

huwanpeng @ 2023-09-22 21:16:24

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define ll long long
using namespace std;
int n,m;
ll a[100010],add[400010],sum[400010];//懒惰标记add 
void push_up(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    push_up(rt);
}
void pushdown(int rt,int ln,int rn)//ln:左子树元素个数  rn:右子树元素个数 
{
    if(add[rt])
    {
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        sum[rt<<1]+=add[rt]*ln;
        sum[rt<<1|1]+=add[rt]*rn;
        add[rt]=0;
    }
}
void update(int L,int R,ll k,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        sum[rt]+=k*(r-l+1);
        add[rt]+=k;
        return;
    }
    if(l==r) return;
    int mid=(l+r)>>1;
    pushdown(rt,mid-l+1,r-mid);
    if(L<=mid)  update(L,R,k,l,mid,rt<<1);
    if(R>mid) update(L,R,k,mid+1,R,rt<<1|1);
    push_up(rt);
}
ll query(int L,int R,int l,int r,int rt)
{
    ll ans=0;
//  printf("%d %d\n",l,r);
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    if(l==r) return 0;
    int mid=(l+r)>>1;
    pushdown(rt,mid-l+1,r-mid);
    if(L<=mid) ans+=query(L,R,l,mid,rt<<1);
    if(R>mid) ans+=query(L,R,mid+1,r,rt<<1|1);
    return ans;
}
int main()
{
//  freopen("P3372.in","r",stdin);
//  freopen("P3372.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    build(1,n,1);
    while(m--)
    {
        int op,x,y,k;
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d%d%lld",&x,&y,&k);
            update(x,y,k,1,n,1);
        }
        else if(op==2)
        {
            scanf("%d%d",&x,&y);
            printf("%lld\n",query(x,y,1,n,1));
        }
    }
    return 0;
}

线段树模板,学校老师教的什么鬼,答案查询都没讲


by One_JuRuo @ 2023-09-22 21:22:17

@huwanpeng

首先,你的定义有些用int有些用longlong,很多输入输出格式也不对应,建议同意改longlong


by One_JuRuo @ 2023-09-22 21:23:13

建议你改一下码风,看着有点难受


by huwanpeng @ 2023-09-22 21:33:33

@One_JuRuo 但是小样例没有用到longlong,但我小样例错了


by One_JuRuo @ 2023-09-22 21:36:11

@huwanpeng

你先改了试试啊。

int op,x,y,k;
scanf("%d",&op);
if(op==1)
{
    scanf("%d%d%lld",&x,&y,&k);
    update(x,y,k,1,n,1);
}

这段代码,你的 k 是整型,但是用longlong 的格式输入,小的负数都会出错,可能是你小样例错误的原因


by One_JuRuo @ 2023-09-22 21:44:13

@huwanpeng

update 函数中:

if(R>mid) update(L,R,k,mid+1,R,rt<<1|1);

你写错了


by huwanpeng @ 2023-09-22 21:46:00

update怎么改啊 @One_JuRuo


by One_JuRuo @ 2023-09-22 21:47:04

@huwanpeng

你是没有理解线段树的含义吗?

if(R>mid) update(L,R,k,mid+1,r,rt<<1|1);

by huwanpeng @ 2023-09-22 21:47:16

k改完后最后一个输出就是22


by huwanpeng @ 2023-09-22 21:48:29

@One_JuRuo 这段不就是说右区间有交集吗


by One_JuRuo @ 2023-09-22 21:48:59

@huwanpeng 你自己思考一下 r R 的含义?


| 下一页