Submission #160678

#	Time	Username	Problem	Language	Result	Execution time	Memory
160678		GioChkhaidze	Gondola (IOI14_gondola)	C++14	60 / 100	23 ms	2300 KiB

gondola

#include <bits/stdc++.h>
#include "gondola.h"

#define F first
#define S second

using namespace std;

bool f[250003],F[250003];
int Gondol[250003];
pair < int , int > a[250003];

int valid(int n, int inputSeq[])
{
	int idx=-1;
	for (int i=0; i<n; i++) 
		if (!f[inputSeq[i]]) f[inputSeq[i]]=1;
			else return 0;
	
	for (int i=0; i<n; i++) 
		if (inputSeq[i]<=n) { idx=i; break; }
	
	if (idx==-1) return 1;
	
	int x=inputSeq[idx];
	for (int i=idx+1; ; i++) {
		x++;
		if (x>n) x-=n;
		if (i==n) i-=n;
		if (i==idx) break;
		if (inputSeq[i]<=n && inputSeq[i]!=x) return 0;
	}
		
  return 1;
}

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

int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
	int l=0,M=0,idx=-1;
	for (int i=0; i<n; i++) 
		if (gondolaSeq[i]<=n) { idx=i; break; }
	
	for (int i=0; i<n; i++) 
		a[i].F=gondolaSeq[i],a[i].S=i;
	
	sort(a,a+n);
	
	if (idx==-1) { 
		for (int i=0; i<n; i++) 
			Gondol[i]=i+1;
	}
		else {
		int x=gondolaSeq[idx];
		Gondol[idx]=x;
		for (int i=idx+1; ; i++) {
			x++;
			if (x>n) x-=n;
			if (i==n) i-=n;
			if (i==idx) break;
			Gondol[i]=x;	
		}
	}
	
	int last=n;
	for (int i=0; i<n; i++) {
		if (a[i].F<=n) continue;
		int id=a[i].S;
		replacementSeq[l++]=Gondol[id];
		for (int j=last+1; j<a[i].F; j++) 
			replacementSeq[l++]=j;
		last=a[i].F;
	}
	return l;
}

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

int countReplacement(int n, int inputSeq[])
{
	if (!valid(n,inputSeq)) return 0;
	int k=0;
	
	for (int i=0; i<n; i++)
		if (inputSeq[i]>n) k++;
		
	if (k<=1) return 1;
	
	int idx=-1;
	for (int i=0; i<n; i++) 
		if (inputSeq[i]<=n) { idx=i; break; }
	
	sort(inputSeq,inputSeq+n);

	long long mod=1e9+9,ANS=1,last=n;
	for (int i=0; i<n; i++) {
		if (inputSeq[i]<=n) continue;
		ANS=(1LL*ANS*(inputSeq[i]-last))%mod;
		last=inputSeq[i];
	}
	
	if (idx==-1) ANS=(ANS*n)%mod;

   return ANS;
}

Compilation message (stderr)

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:41:10: warning: unused variable 'M' [-Wunused-variable]
  int l=0,M=0,idx=-1;
          ^

• 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: 5 / 5
• Subtask 8: 0 / 15
• Subtask 9: 0 / 15
• Subtask 10: 0 / 10