# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
996569 | hasan2006 | Gondola (IOI14_gondola) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
//#include "gondola.h"
using namespace std;
#define TL ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define rall(s) s.rbegin(),s.rend()
#define all(s) s.begin(),s.end()
#define pb push_back
#define se second
#define fi first
#define ll long long
#define ld long double
#define YES cout<<"YES\n"
#define Yes cout<<"Yes\n"
#define yes cout<<"yes\n"
#define NO cout<<"NO\n"
#define No cout<<"No\n"
#define no cout<<"no\n"
const int N = 5e5 + 9 , mod = 1e9 + 9;
ll a[N] , b[N] , d[N] , c[N] , dp[N] ;
int valid(int n , int inputSeq[]){
int x = -1 , y , k;
set<int>st;
for(int i = 0; i < n; i++){
st.insert(inputSeq[i]);
if(inputSeq[i] <= n)
x = k = i , y = inputSeq[i];
}
if(st.size() != n){
return 0;
}
if(x == -1)
return 1;
while((x + 1) % n != k){
x = (x + 1) % n;
y++;
y = (y - 1) % n + 1;
if(inputSeq[x] <= n && inputSeq[x] != y)
return 0;
}
return 1;
}
int replacement(int n, int gondolaSeq[], int replacementSeq[]){
int x = 0 , y = 1 , k = 0;
for(int i = 0; i < n; i++)
if(gondolaSeq[i] <= n)
x = k = i , y = gondolaSeq[i];
for(int i = 1; i <= 3e5; i++)
c[i] = 0;
int mx = 0 ,s = 0 , f;
c[gondolaSeq[x]] = y;
if(gondolaSeq[x] > mx)
mx = gondolaSeq[x],f = y;
while((x + 1) % n != k){
x = (x + 1) % n;
y++;
y = (y - 1) % n + 1;
c[gondolaSeq[x]] = y;
if(gondolaSeq[x] > mx)
mx = gondolaSeq[x],f = y;
}
x = n + 1;
while(x <= mx){
if(x == mx){
replacementSeq[s++] = f;
break;
}
if(c[x]){
replacementSeq[s++] = c[x];
}else {
replacementSeq[s++] = f;
f = x;
}
x++;
}
return s;
}
ll binpow(ll x , int y){
ll ans = 1;
while(y > 0){
if(y % 2) ans = ans * x % mod, y--;
else x = x * x % mod , y/=2;
}
return ans;
}
int countReplacement(int n, int inputSeq[]){
if(!valid(n , inputSeq)){
return 0;
}
vector<int>v;
for(int i = 0; i < n; i++)
if(inputSeq[i] > n)
v.pb(inputSeq[i]);
if(!v.size())
return 1;
v.pb(n);
sort(all(v));
ll ans = 1;
for(int i = 1; i < v.size(); i++){
ans = ans * binpow(v.size() - i , v[i] - v[i - 1] - 1) % mod;
}
if(v.size() == n + 1)
ans = ans * n % mod;
return ans;
}
// Author : حسن