Submission #160689

#	Time	Username	Problem	Language	Result	Execution time	Memory
160689		GioChkhaidze	Gondola (IOI14_gondola)	C++14	100 / 100	52 ms	2424 KiB

gondola

#include <bits/stdc++.h>
#include "gondola.h"
#define F first
#define S second
using namespace std;
bool f[1000006],F[1000006];
int Gondol[250003],b[250003];
pair < int , int > a[250003];
long long mod=1e9+9;
int valid(int n, int inputSeq[]) {
	int idx=-1;
	for (int i=0; i<n; i++)  
		b[i]=inputSeq[i];
	sort(b,b+n);
	for (int i=1; i<n; i++)
		if (b[i]==b[i-1]) 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;
}

long long int po(long long a,long long b) {
	long long int res=1;
	while (b>0) { if (b%2) res=(res*a)%mod; a=(a*a)%mod; b/=2; }
	return res;
}

int countReplacement(int n, int inputSeq[]) {
	if (!valid(n,inputSeq)) return 0;
	sort(inputSeq,inputSeq+n);
	long long int ANS=1,last=n;
	if (inputSeq[0]>n) ANS=(ANS*n)%mod;
	for (int i=0; i<n; i++) {
		if (inputSeq[i]<=n) continue;
		ANS=(ANS*po(n-i,inputSeq[i]-last-1))%mod;
		last=inputSeq[i];
	}
   return ANS;
}

Compilation message (stderr)

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:36: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: 15 / 15
• Subtask 9: 15 / 15
• Subtask 10: 10 / 10