萌新刚学线段树,样例没过...
查看原帖
萌新刚学线段树,样例没过...
363669
hwwqy楼主2021/6/17 09:58
#include<bits/stdc++.h>
#define ll long long
#define bug cout<<"ok"<<endl
using namespace std;
const int dat_size=(1<<18)-1;
ll n,data[dat_size],datb[dat_size];
void init(ll n_){
    n=1;
    while(n<n_)n*=2;
    for(int i=0;i<2*n-1;i++)data[i]=2147483647;
}
void update(ll k,ll a)
{
    k+=n-1;
    data[k]=a;
    while(k>0){
        k=(k-1)/2;
        data[k]=min(data[k*2+1],data[k*2+2]);
    }
}
void add(int a,int b,int x,int k,int l,int r)
{
    if(a<=l&&r<=b){
        data[k]+=x;
    }
    else if(l<b&&a<r){
        datb[k]+=(min(b,r)-max(a,l))*x;
        add(a,b,x,k*2+1,l,(l+r)/2);
        add(a,b,x,k*2+2,(l+r)/2,r);
    }
}
ll sum(int a,int b,int k,int l,int r){
    if(b<=l||r<=a){
        return 0;
    }
    else if(a<=l&&r<=b){
        return data[k]*(r-1)+datb[k];
    }
    else{
        ll res=(min(b,r)-max(a,l))*data[k];
        res+=sum(a,b,k*2+1,l,(l+r)/2);
        res+=sum(a,b,k*2+2,(l+r)/2,r);
        return res;
    }
}
int main()
{
    ll a;
    int N,m,cmd,x,y;
    scanf("%d%d",&N,&m);
    init(N);
    for(int i=0;i<N;i++)
    {
        scanf("%lld",&a);
        update(i,a);
    }
    for(int i=0;i<N;i++)
    {
        printf("%d ",data[i]);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&cmd);
        if(cmd==1)
        {
            scanf("%d %d %d",&x,&y,&a);
            add(x,y+1,a,0,0,N);
        }
        else
        {
            scanf("%d %d",&x,&y);
            printf("%lld\n",sum(x,y+1,0,0,N));
        }
    }
    return 0;
}
2021/6/17 09:58
加载中...