| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 974361 | rahidilbayramli | 곤돌라 (IOI14_gondola) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "gondola.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll long long
#define ld long double
#define vl vector<ll>
#define vi vector<int>
#define pii pair<int, int>
#define pll pair<ll, ll>
#define all(v) v.begin(), v.end()
#define pb push_back
#define f first
#define s second
using namespace std;
using namespace __gnu_pbds;
typedef tree<pll, null_type, less<pll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;
const ll mod = 1e9+9;
ll binpow(ll a, ll b)
{
ll res = 1;
while(b)
{
if(b & 1)
res = res * a % mod;
a = a * a % mod;
b >>= 1;
}
return res;
}
//----------------------
int valid(int n, int inputSeq[])
{
int nxt[n+5], v[n+5];
set<int>st;
for(int i = 1; i <= n; i++)
{
v[i] = inputSeq[i-1];
st.insert(v[i]);
}
for(int i = 1; i <= n; i++)
nxt[i] = (i % n) + 1;
bool flag = false;
for(int i = 1; i <= n; i++)
{
if(v[i] <= n && v[(i%n)+1] <= n && nxt[v[i]] != v[(i%n)+1])
flag = true;
}
if(flag || st.size() < n)
return 0;
return 1;
}
//----------------------
int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
int v[n+5];
int flag = 0;
for(int i = 1; i <= n; i++){
v[i] = gondolaSeq[i-1];
if(v[i] <= n)
flag = 1;
}
if(!flag)
{
vector<pii>vect;
for(int i = 1; i <= n; i++)
vect.pb({v[i], i});
sort(all(vect));
int last = n + 1, idx = 0;
for(int i = 0; i < n; i++)
{
replacementSeq[idx] = vect[i].s;
idx++;
for(int j = last; j < vect[i].f; j++)
{
replacementSeq[idx] = j;
idx++;
}
last = vect[i].f + 1;
}
return idx;
}
else
{
vector<pii>vect;
int idx;
for(int i = 1; i <= n; i++)
{
if(v[i] <= n && v[i] >= 1){
idx = i;
break;
}
}
vect.pb({v[idx], v[idx]});
int idxt = v[idx];
for(int i = idx + 1; i <= n; i++)
{
idxt++;
if(idxt == n + 1)
idxt = 1;
vect.pb({v[i], idxt});
}
idxt = v[idx];
for(int i = idx - 1; i >= 1; i--)
{
idxt--;
if(idxt == 0)
idxt = n;
vect.pb({v[i], idxt});
}
sort(all(vect));
int last = n + 1;
idx = 0;
for(int i = 0; i < n; i++)
{
if(vect[i].f > n){
replacementSeq[idx] = vect[i].s;
idx++;
}
for(int j = last; j < vect[i].f; j++)
{
replacementSeq[idx] = j;
idx++;
}
last = max(last, vect[i].f + 1);
}
return idx;
}
}
//----------------------
int countReplacement(int n, int inputSeq[])
{
int k = valid(n, inputSeq);
if(k == 0)
return 0;
else
{
ll v[n+5], cnt = 0;
bool flag1 = false, flag2 = false;
vl vect;
for(ll i = 1; i <= n; i++)
v[i] = inputSeq[i-1];
for(ll i = 1; i <= n; i++)
{
if(v[i] > n)
{
vect.pb(v[i]);
flag1 = true;
cnt++;
}
if(v[i] <= n)
flag2 = true;
}
ll h = 1;
if(!flag2)
h = n;
sort(all(vect));
ll lst = n + 1;
for(ll i = 0; i < vect.size(); i++)
{
ll diff = vect[i] - lst;
ll p = binpow(cnt, diff);
h *= p;
h %= mod;
cnt--;
lst = vect[i] + 1;
}
int f = h;
return f;
}
}
int main()
{
int n;
cin >> n;
int v[n+5];
for(int i = 0; i < n; i++)
cin >> v[i];
cout << countReplacement(n, v);
}