제출 #1116155

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1116155		staszic_ojuz	Foehn Phenomena (JOI17_foehn_phenomena)	C++17	0 / 100	183 ms	7496 KiB

foehn_phenomena

#include <iostream>
using namespace std;

int tree[800'007];

void UpdateTree(int v, int l, int r, int vl, int vr, int val){
    if (vl>r||vr<l) return;
    if (vl>=l&&vr<=r){
        tree[v]+=val;
        return;
    }
    int mid=(vl+vr)/2;
    UpdateTree(2*v, l, r, vl, mid, val);
    UpdateTree(2*v+1, l, r, mid+1, vr, val);
}

int GetTree(int v){
    int out=0;
    while (v>0){
        out+=tree[v];
        v>>=1;
    }
    return out;
}

int temps[800'007];

void UpdateTemps(int v, int l, int r, int vl, int vr, int val){
    if (vl>r||vr<l) return;
    if (vl>=l&&vr<=r){
        temps[v]+=val;
        return;
    }
    int mid=(vl+vr)/2;
    UpdateTemps(2*v, l, r, vl, mid, val);
    UpdateTemps(2*v+1, l, r, mid+1, vr, val);
}

int GetTemps(int v){
    int out=0;
    while (v>0){
        out+=temps[v];
        v>>=1;
    }
    return out;
}

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    for (int i=0;i<400'007;i++) tree[i]=0;
    for (int i=0;i<400'007;i++) temps[i]=0;
    int n, q, s, t;
    cin>>n>>q>>s>>t;
    int trsize=1;
    while (trsize<n+1) trsize<<=1;
    for (int i=0;i<=n;i++){
        cin>>tree[trsize+i];
    }
    for (int i=1;i<n+1;i++){
        temps[trsize+i]=temps[trsize+i-1];
        if (tree[trsize+i]>tree[trsize+i-1]) temps[trsize+i]-=s*(tree[trsize+i]-tree[trsize+i-1]);
        else temps[trsize+i]+=t*(tree[trsize+i-1]-tree[trsize+i]);
    }
//    for (int i=trsize;i<2*trsize;i++) cout<<GetTemps(i)<<" ";
//    cout<<'\n';
    for (int qe=0;qe<q;qe++){
        int l, r, x;
        cin>>l>>r>>x;
        UpdateTree(1, trsize+l, trsize+r, trsize, 2*trsize-1, x);
        if (l>0){
            int preBorder=GetTree(trsize+l-1), postBorder=GetTree(trsize+l);
            if (preBorder>=postBorder&&preBorder>=postBorder-x){
                UpdateTemps(1, trsize+l, 2*trsize-1, trsize, 2*trsize-1, -t*x);
            }
            else if (preBorder<postBorder&&preBorder<postBorder-x){
                UpdateTemps(1, trsize+l, 2*trsize-1, trsize, 2*trsize-1, -s*x);
            }
            else if (preBorder<postBorder&&preBorder>=postBorder-x){
                UpdateTemps(1, trsize+l, 2*trsize-1, trsize, 2*trsize-1, -t*(preBorder-postBorder+x)+(-s)*(postBorder-preBorder));
            }
            else{
                UpdateTemps(1, trsize+l, 2*trsize-1, trsize, 2*trsize-1, s*(postBorder-x-preBorder)+t*(postBorder-preBorder));
            }
        }
        if (r<n){
            int preBorder=GetTree(trsize+r), postBorder=GetTree(trsize+r+1);
            if (preBorder>=postBorder&&preBorder-x>=postBorder){
                UpdateTemps(1, trsize+r+1, 2*trsize-1, trsize, 2*trsize-1, t*x);
            }
            else if (preBorder<postBorder&&preBorder-x<postBorder){
                UpdateTemps(1, trsize+r+1, 2*trsize-1, trsize, 2*trsize-1, s*x);
            }
            else if (preBorder<postBorder&&preBorder-x>=postBorder){
                UpdateTemps(1, trsize+r+1, 2*trsize-1, trsize, 2*trsize-1, -t*(preBorder-x-postBorder)+(-s)*(postBorder-preBorder));
            }
            else{
                UpdateTemps(1, trsize+r+1, 2*trsize-1, trsize, 2*trsize-1, s*(postBorder-preBorder+x)+t*(preBorder-postBorder));
            }
        }
//        for (int i=trsize;i<2*trsize;i++) cout<<GetTree(i)<<" ";
//        cout<<'\n';
//        for (int i=trsize;i<2*trsize;i++) cout<<GetTemps(i)<<" ";
//        cout<<'\n';
        cout<<GetTemps(2*trsize-1)<<'\n';
    }
    return 0;
}

• Subtask 1: 0 / 30
• Subtask 2: 0 / 10
• Subtask 3: 0 / 60