Submission #542047

#	Time	Username	Problem	Language	Result	Execution time	Memory
542047		Sho10	Gondola (IOI14_gondola)	C++17	25 / 100	86 ms	6924 KiB

gondola

#include <bits/stdc++.h> //Andrei Alexandru a.k.a Sho
#include "gondola.h"
using ll=long long;
using ld=long double;
int const INF=1000000005;
ll const LINF=1000000000000000005;
ll const mod=1000000007;
ld const PI=3.14159265359;
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define endl '\n'
#define CODE_START  ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
int valid(int n,int a[]){
int mn=INF,pos=-1;
map<ll,ll>viz;
for(int i=0;i<n;i++)
{
    viz[a[i]]++;
    if(a[i]<mn){
        mn=a[i];
        pos=i;
    }
}
for(ll i=0;i<n;i++)
{
    if(viz[a[i]]>=2){
        return 0;
    }
}
if(mn>n){
    return 1;
}
int b[200005];
for(int i=0;i<n;i++)
{
    b[i]=a[i];
}
a[0]=b[pos];
ll val=0;
for(ll i=pos+1;i<n;i++)
{
    a[i-pos]=b[i];
    val=i-pos;
}
val++;
for(ll i=0;i<pos;i++)
{
a[val]=b[i];
val++;
}
ll x=a[0];
x++;
b[0]=a[0];
for(ll i=1;i<n;i++)
{
    x%=n;
    if(x==0){
        x=n;
    }
b[i]=x;
x++;
}
for(ll i=1;i<n;i++)
{
    if(b[i]!=a[i]){
        if(a[i]<=n){
            return 0;
        }
    }
}
return 1;
}
int replacement(int n,int a[],int g[]){
int mn=INF,pos=-1;
map<ll,ll>viz;
for(int i=0;i<n;i++)
{
    viz[a[i]]++;
    if(a[i]<mn){
        mn=a[i];
        pos=i;
    }
}
int b[200005],val;
for(int i=0;i<n;i++)
{
    b[i]=a[i];
}
if(mn>n){
    goto Next;
}
a[0]=b[pos];
val=0;
for(ll i=pos+1;i<n;i++)
{
    a[i-pos]=b[i];
    val=i-pos;
}
val++;
for(ll i=0;i<pos;i++)
{
a[val]=b[i];
val++;
}
for(int i=0;i<n;i++)
{
    b[i]=a[i];
}
val=mn;
for(ll i=0;i<n;i++)
{
    ll x=i-mn+1;
    if(x<0){
        x+=n;
    }
    a[i]=b[x];

}
Next:
vector<int>ans;
multiset<pair<ll,ll>>s;
int mx=0;
ll check=0;
for(ll i=0;i<n;i++)
{
    mx=max(mx,a[i]);
    if(a[i]==n+1){
        check=1;
    }
    if(a[i]>n){
        s.insert(mp(a[i],i+1));
    }
}
if(check==0){
    for(ll i=0;i<n;i++)
    {
        if(a[i]>n){
            s.insert(mp(n+1,i+1));
            s.erase(s.find(mp(a[i],i+1)));
            break;
        }
    }
}
while(!s.empty()){
    auto it=s.begin();
    pair<ll,ll>p=*(it);
    if(p.f>mx){
        break;
    }
    s.erase(s.begin());
    it=s.begin();
    pair<ll,ll>x=*(it);
    ans.pb(p.s);
    if(x.f==p.f+1){
        continue;
    }
    s.insert(mp(p.f+1,p.f));
}
ll res=ans.size();
for(int i=0;i<res;i++)
{
    g[i]=ans[i];
}
return res;
}
int countReplacement(int n,int a[]){
return 0;
}
/*
int32_t main(){
int n,a[100005],idk[100005];
cin>>n;
for(int i=0;i<n;i++)
{
    cin>>a[i];
}
cout<<replacement(n,a,idk)<<endl;
}
*/

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 10 / 10
• Subtask 4: 5 / 5
• Subtask 5: 0 / 10
• Subtask 6: 0 / 20
• Subtask 7: 0 / 5
• Subtask 8: 0 / 15
• Subtask 9: 0 / 15
• Subtask 10: 0 / 10