Submission #1287782

#TimeUsernameProblemLanguageResultExecution timeMemory
1287782ghammazhassanReal Mountains (CCO23_day1problem2)C++20
0 / 25
1 ms572 KiB
// #include <bits/stdc++.h>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <map>
#include <unordered_map>
#include <vector>
#include <iomanip>
#include <string>
#include <queue>
#include <set>
#include <deque>
using namespace std;
#define int long long
#define endl "\n"
#define fi first
#define se second
const int M=1e6+3;
const int inf = 1e14;
const int LOG=18;
const int N=1e6+5;
int n , m , c , w , k , t=1 , q=1 , x , y , z , l , r;
int seg[4*N];
int a[N];
void update(int i,int s=0,int e=n,int v=1){
    if (i<s or i>=e)return;
    if (e-s==1){
        seg[v]=a[i];
        return;
    }
    int li=(2*v);
    int ri=(2*v+1);
    int m=(s+e)/2;
    update(i,s,m,li);
    update(i,m,e,ri);
    seg[v]=min(seg[li],seg[ri]);
}
int query(int l,int r,int s=0,int e=n,int v=1){
    if (r<=s or l>=e){
        return inf;
    }
    if (l<=s and e<=r){
        return seg[v];
    }
    int li=(2*v);
    int ri=(2*v+1);
    int m=(s+e)/2;
    return min(query(l,r,s,m,li),query(l,r,m,e,ri));
}
void solve()
{   
    cin >> n;
    for (int i=0;i<n;i++){
        cin >> a[i];
        update(i);
    }
    vector<pair<int,int>>p(n);
    for (int i=0;i<n;i++){
        p[i]={a[i],i};
    }
    sort(p.begin(),p.end());
    vector<int>l;
    l.push_back(p[0].fi);
    for (int i=1;i<n;i++){
        if(p[i].fi==p[i-1].fi){
            // l[l.size()-1].push_back(p[i].se);
        }
        else{
            l.push_back(p[i].fi);
        }
    }
    c=0;
    for (int x:l){
        bool f=0;
        int i1=-1,i2=-1;
        for (int i=0;i<n;i++){
            if (a[i]>x)f=1;
            if (a[i]==x and f){
                i1=i;
                break;
            }
        }
        f=0;
        for (int i=n-1;i>=0;i--){
            if (a[i]>x)f=1;
            if (a[i]==x and f){
                i2=i;
                break;
            }
        }
        if (i2<i1)continue;
        if (i1==-1 or i2==-1)continue;
        int u1=inf;
        for (int i=0;i<i1;i++){
            if (a[i]>x){
                u1=min(u1,a[i]);
            }
        }
        int u2=inf;
        for (int i=n-1;i>i2;i--){
            if (a[i]>x){
                u2=min(u2,a[i]);
            }
        }
        int u3=inf;
        for (int i=i1;i<i2;i++){
            if (a[i]>x){
                u3=min(u3,a[i]);
            }
        }
        int u=min({u1,u2,u3})-x;
        if (i1==i2){
            c+=(u1+u2+x)*u+u*(u-1)/2;
            continue;
        }
        c+=((min(u1+min(u2,u3),min(u1,u3)+u2))+x)*u+u*(u-1)/2;
        c+=(min(u1,u2)+x+x)*u+u*(u-1)/2+u*(u+1)/2;
        int k=0;
        for (int i=i1+1;i<i2;i++){
            k+=a[i]==x;
        }
        c+=k*(x+x+x+2*u*(u+1)/2+u*(u-1)/2);
        for (int i=i1;i<=i2;i++){
            if (a[i]==x){
                a[i]+=u;
            }
        }
    }
    // for (int i=0;i<n;i++){
    //     cout << a[i] << " ";
    // }
    cout << c << endl;
}
signed main()    
{   

    ios::sync_with_stdio(0);//DO NOT USE IN INTERACTIVE
    cin.tie(0), cout.tie(0);//DO NOT USE IN INTERACTIVE
    cout << fixed << setprecision(9);
    srand(time(0));
    // int t=1;
    // cin >> t;
    for (int _=1;_<=t;_++){
        solve();
        q++;
    }
} 
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...