蓝桥杯练习 算法提高 ADV-183 分苹果【线段树】

题目链接：http://lx.lanqiao.cn/problem.page?gpid=T345

线段树模板题，稍微改一下query函数就行了。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m,x,y,k,a[N],tr[4*N],add[4*N];
void pushup(int i)
{tr[i]=tr[2*i]+tr[2*i+1];}
void build(int i,int l,int r)
{
    add[i]=0;
    if(l==r){tr[i]=a[l];return;}
    int mid=(l+r)/2;
    build(2*i,l,mid);
    build(2*i+1,mid+1,r);
    pushup(i);
}
void Add(int i,int l,int r,int k)
{
    add[i]+=k;
    tr[i]+=(r-l+1)*k;
}
void pushdown(int i,int l,int r)//这里传参时不必传k
{
    if(add[i]==0)return;
    int mid=(l+r)/2;
    Add(2*i,l,mid,add[i]);
    Add(2*i+1,mid+1,r,add[i]);
    add[i]=0;
}
void update(int i,int l,int r,int x,int y,int k)
{
    if(r<x||l>y)return;
    if(l>=x&&r<=y)return Add(i,l,r,k);
    pushdown(i,l,r);
    int mid=(l+r)/2;
    update(2*i,l,mid,x,y,k);
    update(2*i+1,mid+1,r,x,y,k);
    pushup(i);
}
void query(int i,int l,int r,int x,int y)
{
    if(r<x||l>y)return;
    if(l==r){a[l]=tr[i];return;}
    pushdown(i,l,r);
    int mid=(l+r)/2;
    query(2*i,l,mid,x,y);
    query(2*i+1,mid+1,r,x,y);
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    build(1,1,n);
    while(m--)
    {
        cin>>x>>y>>k;
        update(1,1,n,x,y,k);
    }
    query(1,1,n,1,n);
    for(int i=1;i<=n;i++)
        i==n?printf("%d\n",a[i]):printf("%d ",a[i]);
    return 0;
}