Submission #1020580

#	Time	Username	Problem	Language	Result	Execution time	Memory
1020580		lamagril	Mountains (NOI20_mountains)	C++14	100 / 100	172 ms	29484 KiB

Mountains

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int N=3e5+5;
vector<int> v(N);
vector<int> f[N];

struct fenwick{
    int d[N];
    void update(int idx,int vl){
        while(idx<N){
            d[idx]+=vl;
            idx+=idx&(-idx);
        }
    }
    int query(int idx){
        int sm=0;
        while(idx>0){
            sm+=d[idx];
            idx-=(idx&-idx);
        }
        return sm;
    }
} fe;

int32_t main(){
    cin.tie(NULL)->sync_with_stdio(false);
    int ans=0;
    int n; cin >> n;
    vector<int> a;
    for(int i=1 ; i<=n ; i++){
        fe.update(i,1);
    }
    for(int i=1 ; i<=n ; i++) cin >> v[i],a.push_back(v[i]);
    sort(a.begin(),a.end()); 
    for(int i=1 ; i<=n ; i++){
        v[i]=lower_bound(a.begin(),a.end(),v[i])-a.begin()+1;
    }
    for(int i=1 ; i<=n ; i++){
        f[v[i]].push_back(i);
    }
    for(int i=n ; i>=1 ; i--){
        if(f[i].empty()) continue;
        for(int j:f[i]){
            fe.update(j,-1);
        }
        for(int j:f[i]){
            ans+=fe.query(j)*(fe.query(n)-fe.query(j));
        }
    }
    cout << ans << '\n';
}

• Subtask 1: 2 / 2
• Subtask 2: 4 / 4
• Subtask 3: 9 / 9
• Subtask 4: 36 / 36
• Subtask 5: 28 / 28
• Subtask 6: 9 / 9
• Subtask 7: 12 / 12