Submission #247631

# Submission time Handle Problem Language Result Execution time Memory
247631 2020-07-11T17:26:10 Z Evirir Gondola (IOI14_gondola) C++17
75 / 100
26 ms 3576 KB
#include "gondola.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

#define watch(x) cout<<(#x)<<"="<<(x)<<'\n'
#define mset(d,val) memset(d,val,sizeof(d))
#define setp(x) cout<<fixed<<setprecision(x)
#define forn(i,a,b) for(int i=(a);i<(b);i++)
#define fore(i,a,b) for(int i=(a);i<=(b);i++)
#define pb push_back
#define F first
#define S second
#define pqueue priority_queue
#define fbo find_by_order
#define ook order_of_key
typedef long long ll;
typedef pair<ll,ll> ii;
typedef vector<ll> vi;
typedef vector<ii> vii;
typedef long double ld;
typedef tree<ll, null_type, less<ll>, rb_tree_tag, tree_order_statistics_node_update> pbds;
void amin(ll &a, ll b){ a=min(a,b); }
void amax(ll &a, ll b){ a=max(a,b); }
void YES(){cout<<"YES\n";} void NO(){cout<<"NO\n";}
void SD(int t=0){ cout<<"PASSED "<<t<<endl; }
const ll INF = ll(1e18);
const int MOD = 1e9+9;

bool DEBUG = 0;
const int MAXN = 100005;
const int MAXA = 250005;

ll add(ll a,ll b)
{
	a+=b; a%=MOD;
	if(a<0) a+=MOD;
	return a;
}
ll mult(ll a, ll b)
{
	if(a>MOD) a%=MOD;
	if(b>MOD) b%=MOD;
	ll ans=(a*b)%MOD;
	if(ans<0) ans+=MOD;
	return ans;
}
ll pw(ll a, ll b)
{
	ll r=1;
	while(b){
		if(b&1) r=mult(r,a);
		a=mult(a,a);
		b>>=1;
	}
	return r;
}
ll inverse(ll a)
{
	return pw(a,MOD-2);
}

int valid(int n, int inputSeq[])
{
	int minid=0;
	forn(i,0,n){
		if(inputSeq[minid]>inputSeq[i]){
			minid=i;
		}
	}
	
	int tmp[n];
	forn(i,0,n){
		tmp[i]=inputSeq[(minid+i)%n];
	}
	forn(i,0,n) inputSeq[i]=tmp[i];
	
	bool used[MAXA]{};
	int last=-1;
	for(int i=0;i<n;i++){
		if(used[inputSeq[i]]) return 0;
		used[inputSeq[i]]=1;
		
		if(inputSeq[i]<=n){
			if(last!=-1 && inputSeq[i]-inputSeq[last]!=i-last) return 0;
			last=i;
		}
	}
	
	return 1;
}

int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
	assert(valid(n, gondolaSeq));
	
	int loc[MAXA];
	mset(loc,-1);
	
	int large=0;
	for(int i=0;i<n;i++){
		loc[gondolaSeq[i]]=i;
		if(gondolaSeq[i]>gondolaSeq[large]){
			large=i;
		}
	}
	
	int a[n];
	int start=0,val=1;
	for(int i=0;i<n;i++){
		if(gondolaSeq[i]<=n){
			start=i;
			val=gondolaSeq[i];
			break;
		}
	}
	for(int i=start;;){
		a[i]=val++;
		if(val==n+1) val=1;
		
		i=(i+1)%n;
		if(i==start) break;
	}
	
	int ptr=0;
	for(int i=n+1;i<=gondolaSeq[large];i++){
		if(loc[i] == -1){
			//cout<<"Not found\n";
			replacementSeq[ptr++]=a[large];
			a[large]=i;
		}
		else{
			//cout<<"Found\n";
			replacementSeq[ptr++] = a[loc[i]];
			a[loc[i]]=i;
		}
	}
	
	return ptr;
}

int countReplacement(int n, int inputSeq[])
{
	if(!valid(n, inputSeq)) return 0;
	
	vi v = {n};
	for(int i=0;i<n;i++){
		v.pb(inputSeq[i]);
	}
	sort(v.begin(), v.end());
	
	ll cur=0;
	forn(i,0,v.size()){
		if(v[i]>n) cur++;
	}
	
	ll ans=1;
	for(int i=1;i<v.size();i++){
		if(v[i] <= n) continue;
		
		ans=mult(ans, pw(cur, v[i]-v[i-1]-1));
		cur--;
	}
	
	return ans;
}

Compilation message

gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:11:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define forn(i,a,b) for(int i=(a);i<(b);i++)
                                    ^
gondola.cpp:155:2: note: in expansion of macro 'forn'
  forn(i,0,v.size()){
  ^~~~
gondola.cpp:160:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=1;i<v.size();i++){
              ~^~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 5 ms 512 KB Output is correct
2 Correct 5 ms 512 KB Output is correct
3 Correct 5 ms 640 KB Output is correct
4 Correct 5 ms 640 KB Output is correct
5 Correct 5 ms 640 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 640 KB Output is correct
2 Correct 5 ms 512 KB Output is correct
3 Correct 5 ms 512 KB Output is correct
4 Correct 5 ms 512 KB Output is correct
5 Correct 5 ms 512 KB Output is correct
6 Correct 10 ms 1152 KB Output is correct
7 Correct 17 ms 1792 KB Output is correct
8 Correct 14 ms 1536 KB Output is correct
9 Correct 8 ms 896 KB Output is correct
10 Correct 16 ms 1664 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 512 KB Output is correct
2 Correct 5 ms 640 KB Output is correct
3 Correct 5 ms 512 KB Output is correct
4 Correct 5 ms 512 KB Output is correct
5 Correct 5 ms 512 KB Output is correct
6 Correct 10 ms 1024 KB Output is correct
7 Correct 17 ms 1792 KB Output is correct
8 Correct 14 ms 1536 KB Output is correct
9 Correct 8 ms 896 KB Output is correct
10 Correct 16 ms 1664 KB Output is correct
11 Correct 5 ms 512 KB Output is correct
12 Correct 5 ms 512 KB Output is correct
13 Correct 10 ms 1024 KB Output is correct
14 Correct 5 ms 512 KB Output is correct
15 Correct 19 ms 1792 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 1536 KB Output is correct
2 Correct 5 ms 1536 KB Output is correct
3 Correct 5 ms 1536 KB Output is correct
4 Correct 5 ms 1536 KB Output is correct
5 Correct 5 ms 1536 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 1536 KB Output is correct
2 Correct 5 ms 1536 KB Output is correct
3 Correct 5 ms 1536 KB Output is correct
4 Correct 5 ms 1536 KB Output is correct
5 Correct 5 ms 1536 KB Output is correct
6 Correct 5 ms 1536 KB Output is correct
7 Correct 5 ms 1536 KB Output is correct
8 Correct 6 ms 1664 KB Output is correct
9 Correct 6 ms 1536 KB Output is correct
10 Correct 6 ms 1528 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 1536 KB Output is correct
2 Correct 5 ms 1536 KB Output is correct
3 Correct 6 ms 1536 KB Output is correct
4 Correct 5 ms 1536 KB Output is correct
5 Correct 5 ms 1536 KB Output is correct
6 Correct 5 ms 1536 KB Output is correct
7 Correct 5 ms 1536 KB Output is correct
8 Correct 6 ms 1536 KB Output is correct
9 Correct 6 ms 1664 KB Output is correct
10 Correct 6 ms 1536 KB Output is correct
11 Correct 17 ms 2560 KB Output is correct
12 Correct 18 ms 2868 KB Output is correct
13 Correct 18 ms 2688 KB Output is correct
14 Correct 17 ms 2560 KB Output is correct
15 Correct 26 ms 3576 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 640 KB Output is correct
2 Correct 5 ms 512 KB Output is correct
3 Correct 5 ms 512 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 512 KB Output is correct
2 Correct 5 ms 512 KB Output is correct
3 Correct 5 ms 512 KB Output is correct
4 Correct 5 ms 512 KB Output is correct
5 Correct 5 ms 512 KB Output is correct
6 Correct 5 ms 640 KB Output is correct
7 Correct 5 ms 640 KB Output is correct
8 Correct 5 ms 512 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 512 KB Output is correct
2 Correct 5 ms 512 KB Output is correct
3 Correct 5 ms 640 KB Output is correct
4 Correct 5 ms 512 KB Output is correct
5 Correct 5 ms 512 KB Output is correct
6 Correct 5 ms 512 KB Output is correct
7 Correct 5 ms 512 KB Output is correct
8 Correct 5 ms 512 KB Output is correct
9 Correct 24 ms 2808 KB Output is correct
10 Correct 20 ms 2044 KB Output is correct
11 Correct 10 ms 1280 KB Output is correct
12 Correct 12 ms 1280 KB Output is correct
13 Incorrect 6 ms 768 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 512 KB Output is correct
2 Correct 5 ms 512 KB Output is correct
3 Correct 5 ms 512 KB Output is correct
4 Correct 5 ms 512 KB Output is correct
5 Correct 5 ms 512 KB Output is correct
6 Correct 5 ms 512 KB Output is correct
7 Correct 5 ms 640 KB Output is correct
8 Correct 5 ms 512 KB Output is correct
9 Correct 24 ms 2808 KB Output is correct
10 Correct 21 ms 2044 KB Output is correct
11 Correct 10 ms 1280 KB Output is correct
12 Correct 12 ms 1280 KB Output is correct
13 Incorrect 6 ms 768 KB Output isn't correct
14 Halted 0 ms 0 KB -