Submission #307127

#	Time	Username	Problem	Language	Result	Execution time	Memory
307127		NintsiChkhaidze	Exercise Deadlines (CCO20_day1problem2)	C++14	0 / 25	6 ms	5248 KiB

Main

#include <bits/stdc++.h>
#define f first
#define s second
#define pb push_back
#define left (node<<1),l,((l+r)>>1)
#define right ((node<<1)|1),((l+r)>>1) + 1,r
#define N 200005
using namespace std;
priority_queue <int> q;
int fen[200005];
vector <int> v[200005];
void upd(int in,int val){
    while (in <= N){
        fen[in]+=val;
        in += (in&(-in));
    }
}
int get(int in){
    int sum=0;
    while (in > 0){
        sum+=fen[in];
        in -= (in&(-in));
    }
    return sum;
}
int main(){
    ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int n;
    cin>>n;
    
    for (int i=1;i<=n;i++){
        int a;
        cin>>a;
        v[a].pb(i);
    }
    for (int i = 0; i < v[n].size(); i++)
        q.push(v[n][i]);

    int ind = n,ans = 0;
    while(q.size()){
        int x = q.top();
        q.pop();
        //cout<<x<<" "<<ind<<endl;
        ans+=ind - x;
        upd(x,1),upd(ind + 1,-1);
        if (ind > 1){
            ind--;
            for (int i = 0; i < v[ind].size(); i++){
                q.push(v[ind][i] - get(v[ind][i] - 1));
                //cout<<v[ind][i]<<" - "<<get(v[ind][i] - 1)<<endl;
            }
        }
    }
    if (ind != 1 && q.size() == 0) cout<<-1;
    else cout<<ans;
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:36:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     for (int i = 0; i < v[n].size(); i++)
      |                     ~~^~~~~~~~~~~~~
Main.cpp:48:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |             for (int i = 0; i < v[ind].size(); i++){
      |                             ~~^~~~~~~~~~~~~~~

• Subtask 1: 0 / 17
• Subtask 2: 0 / 8