Submission #896790

#	Time	Username	Problem	Language	Result	Execution time	Memory
896790		LCJLY	Gondola (IOI14_gondola)	C++14	55 / 100	17 ms	6252 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>
using namespace std;

int valid(int n, int arr[]){
	unordered_set<int>se;
	bool amos=true;
	deque<int>d;
	int mini=INT_MAX;
	int small=0;
	int big=0;
	for(int x=0;x<n;x++){
		if(se.find(arr[x])!=se.end()) amos=false;
		se.insert(arr[x]);
		if(arr[x]<=n){
			d.push_back(arr[x]);
			mini=min(mini,arr[x]);
			small++;
		}	
		else big++;
	}
	
	int ptr=1;
	for(int x=1;x<=n;x++){
		if(se.find(x)==se.end()){
			if(se.find(ptr+n)==se.end()) amos=false;
			ptr++;
		}
	}
	//if(n-small!=big) amos=false;
	
	if(!d.empty()){
		while(d.front()!=mini){
			d.push_front(d.back());
			d.pop_back();
		}
		
		for(int x=1;x<(int)d.size();x++){
			if(d[x]<d[x-1]) amos=false;
		}
	}
	
	return amos;
}

int replacement(int n, int arr[], int ans[]){
	int maxi=0;
	int shift=-1;
	deque<int>d;
	for(int x=0;x<n;x++){
		maxi=max(maxi,arr[x]);
		if(arr[x]<=n){
			if(arr[x]>=x+1){
				shift=arr[x]-(x+1);
			}
			else{
				shift=arr[x]+n-(x+1);
			}
		}
		d.push_back(arr[x]);
	}
	
	while(shift>0){
		shift--;
		d.push_front(d.back());
		d.pop_back();
	}
	
	vector<pair<int,int>>v;
	for(int x=0;x<n;x++){
		if(d[x]>n){
			v.push_back({d[x],x});
		}
	}
	
	sort(v.begin(),v.end());
	
	int index=0;
	int cur=n;
	for(auto it:v){
		bool done=true;
		while(cur<it.first){
			if(done){
				ans[index]=it.second+1;
				done=false;
				index++;
			}
			else{
				ans[index]=cur;
				index++;
			}
			cur++;
		}
	}
	
	return maxi-n;
}

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

int countReplacement(int n, int arr[]){
	int arr2[n];
	for(int x=0;x<n;x++) arr2[x]=arr[x];
	//if(valid(n,arr2)==0) return 0;
	int maxi=0;
	vector<int>v;
	for(int x=0;x<n;x++){
		maxi=max(maxi,arr[x]);
		if(arr[x]>n)v.push_back(arr[x]);
	}
	if(maxi==n) return 1;
	int mod=1e9+7;
	int cur=n;
	int counter=1;
	sort(v.begin(),v.end());
	int sz=v.size();
	//show(sz,sz);
	for(auto it:v){
		int gap=max(1,it-cur-1);
		//show3(gap,gap,it,it,cur,cur);
		counter=(counter*gap)%mod;
		counter=(counter*sz)%mod;
		cur=it;
		sz--;
	}
	//show(check,1);
	return counter;
}
//

Compilation message (stderr)

gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:102:6: warning: variable 'arr2' set but not used [-Wunused-but-set-variable]
  102 |  int arr2[n];
      |      ^~~~

• 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