Submission #389348

#	Time	Username	Problem	Language	Result	Execution time	Memory
389348		Pichon5	Gondola (IOI14_gondola)	C++17	55 / 100	40 ms	4612 KiB

gondola

#include "gondola.h"
#include<bits/stdc++.h>
#define vi vector<int>
#define pb push_back
#define F first
#define S second
#define ll long long int
using namespace std;
const int MOD=1e9+7;
int valid(int n, int v[]){
	set<int>st;
	for(int i=0;i<n;i++){
        st.insert(v[i]);
	}
	if(st.size()!=n)return 0;
	int pos=0;
	for(int i=0;i<n;i++){
        if(v[i]<=n){
            pos=i-(v[i]-1);
            if(pos<0)pos+=n;
            break;
        }
	}
	int now=1;
	for(int i=pos;i<n;i++){
        if(v[i]<=n){
            if(v[i]!=now){
                return 0;
            }
        }
        now++;
	}
	for(int i=0;i<pos;i++){
        if(v[i]<=n){
            if(v[i]!=now){
                return 0;
            }
        }
        now++;
	}
	return 1;
}
//----------------------

int replacement(int n, int v[], int R[])
{
    int pos1=0;
    for(int i=0;i<n;i++){
        if(v[i]<=n){
            int B=v[i]-1;
            if(i-B>=0){
                pos1=i-B;
            }else{
                B-=i;
                pos1=n-B;
            }
            break;
        }
    }
    int pos=pos1;
    vector<pair<int,int> >E;//gondola final y gondola inicial
    for(int i=1;i<=n;i++){
        E.pb({v[pos],i});
        pos++;
        if(pos==n)pos=0;
    }
    sort(E.begin(),E.end());
    int next=n+1;
    int res=0;
    for(int i=0;i<n;i++){
        if(E[i].F==E[i].S)continue;
        R[res]=E[i].S;next++;res++;
        //cout<<"pongo  "<<E[i].S<<endl;
        int curr=next-1;
        while(curr!=E[i].F){
            //cout<<"pongo   "<<curr<<endl;
            R[res]=curr;
            curr++;next++;res++;
        }
    }
    //cout<<"respuesta   "<<res<<endl;
    //cout<<endl;

    return res;
}

//----------------------

int countReplacement(int n, int v[]){
    if(!valid(n,v))return 0;
    ll res=1;
    vi sum(250001,0);
    ll m=n;
    bool ok=0;
    for(int i=0;i<n;i++){
        if(v[i]<=n)ok=1;
        sum[v[i]]++;
    }
    for(int i=250000;i>n;i--)sum[i]+=sum[i+1];
    for(int i=n+1;i<250000;i++){
        if(sum[i]!=sum[i+1])continue;
        res*=sum[i];
        res*=MOD;
    }
    if(ok)res*=m;
    res%=MOD;
    return res;
}
//number of subtask
//n
//3 1 4
//5 1 2 3 4
//2 3 4 9 6 7 1

Compilation message (stderr)

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:15:14: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   15 |  if(st.size()!=n)return 0;
      |     ~~~~~~~~~^~~

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