Submission #138799

#	Time	Username	Problem	Language	Result	Execution time	Memory
138799		Mahmoud_Adel	Gondola (IOI14_gondola)	C++14	70 / 100	93 ms	8180 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6+5, mod = 1e9+9;
int mep[N];
map<ll, ll> mark;
ll power(ll a, ll b)
{
	if(!b) return 1;
	ll temp = power(a, b/2);
	temp = (temp*temp)%mod;
	if(b%2) temp *= a, temp %= mod;
	return temp;
}
int valid(int n, int inputSeq[])
{
	for(int i=0; i<n; i++)
	{
		if(mark[inputSeq[i]]) return 0;
		else mark[inputSeq[i]] = 1;
	}
	int j = -1;
	for(int i=0; i<n; i++) if(inputSeq[i] > 0 && inputSeq[i] <= n) j = i;
	if(j == -1) return 1;
	for(int i=0; i<n; i++)
	{
		int id = (i+j)%n, jd = (i+j+1)%n;
		if(inputSeq[jd] > n) inputSeq[jd] = inputSeq[id]%n + 1;
		if(inputSeq[jd] != inputSeq[id]%n + 1) return 0;
	}
	return 1;
}

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

int replacement(int n, int inputSeq[], int replacementSeq[])
{
	memset(mep, 0, sizeof mep);
	vector<int> vec;
	int j = -1;
	int r = *max_element(inputSeq, inputSeq+n);
	for(int i=0; i<n; i++) if(inputSeq[i] <= n) j = i;
	if(j == -1) for(int i=0; i<n; i++) vec.push_back(inputSeq[i]), 
	mep[inputSeq[i]] = i+1;
	else
	{
		for(int i=0; i<n; i++)
		{
			int id = (i+j)%n, jd = (i+j+1)%n;
			if(inputSeq[jd] > n)
			{
				vec.push_back(inputSeq[jd]);
				mep[inputSeq[jd]] = inputSeq[id]%n + 1;
				inputSeq[jd] = inputSeq[id]%n + 1;
			}
		}
	}
	sort(vec.begin(), vec.end());
	int cur = n+1, c = 0;
	for(int i : vec)
	{
		int last = mep[i];
		while(cur < i) replacementSeq[c++] = last, last = cur, cur++;
		replacementSeq[c++] = last;
	}
	return r-n;
}

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

int countReplacement(int n, int inputSeq[])
{
	int tmp[n];
	for(int i=0; i<n; i++) tmp[i] = inputSeq[i];
	if(!valid(n, inputSeq)) return 0;
	for(int i=0; i<n; i++) inputSeq[i] = tmp[i];
	ll j = -1, c = n;
	vector<ll> vec;
	for(int i=0; i<n; i++) if(inputSeq[i] <= n) j  = i;
	else vec.push_back(inputSeq[i]);
	ll ans = 1;
	vec.push_back(n);
	sort(vec.begin(), vec.end());
	for(int i=0; i<vec.size()-1; i++)
	{
		ll tmp = power((ll)vec.size()-i-1, vec[i+1]-vec[i]-1);
		ans = (ans%mod * tmp%mod)%mod;
	}
	return (ans*(j==-1? n:1))%mod;
}

Compilation message (stderr)

gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:85:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0; i<vec.size()-1; i++)
               ~^~~~~~~~~~~~~
gondola.cpp:78:13: warning: unused variable 'c' [-Wunused-variable]
  ll j = -1, c = n;
             ^

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