Submission #363548

#	Time	Username	Problem	Language	Result	Execution time	Memory
363548		silverfish	Gondola (IOI14_gondola)	C++14	55 / 100	48 ms	4736 KiB

gondola

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

using namespace std;
#define ar array
#define pb push_back

int valid(int n, int a[]){
	bool ok = 1, first = 1;
	int diff = -1;
	map<int, int> cnt;
	for(int i = 0; i < n; ++i){
		if(cnt[a[i]]) ok = 0;
		++cnt[a[i]];
		if(a[i] <= n){
			if(first) {
				diff = (a[i]-(i+1)+n)%n;
				first = 0;
				continue;
			}
			//cerr << i+1 << ' ' << (n+a[i]-diff)%n << endl;
			if(i != (a[i]-diff-1+n)%n) ok = 0;
		}
	}

	return ok;
}

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

int replacement(int n, int a[], int r[]){
	bool ok = 1, first = 1;
	int diff = -1;
	vector<ar<int,2>> replaced;

	for(int i = 0; i < n; ++i){
		if(a[i] <= n){
			diff = (a[i]-(i+1)+n)%n;
		}else{
			replaced.pb({a[i], i});
		}
	}
	if(diff == -1) diff = 0;
	sort(replaced.begin(), replaced.end());
	int cur = 0, curnum = n;

	for(auto x : replaced){
		int orig = (x[1]+1+diff);
		if(orig > n) orig -= n;
		r[cur++] = orig;
		++curnum;
		for(; curnum < x[0]; ++curnum) r[cur++] = curnum;
	}
	return cur;
}

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

int countReplacement(int n, int a[]){
	if(!valid(n, a)) return 0;

	return -1;
}

Compilation message (stderr)

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:32:7: warning: unused variable 'ok' [-Wunused-variable]
   32 |  bool ok = 1, first = 1;
      |       ^~
gondola.cpp:32:15: warning: unused variable 'first' [-Wunused-variable]
   32 |  bool ok = 1, first = 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: 0 / 5
• Subtask 8: 0 / 15
• Subtask 9: 0 / 15
• Subtask 10: 0 / 10