# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
250665 | hhh07 | 곤돌라 (IOI14_gondola) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <utility>
#include <set>
#include <cmath>
#include <climits>
#include <cstring>
#include <stdio.h>
#include <assert.h>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef pair<ll, ll> ii;
ll MOD = 1e9 + 9;
ll stepen(int x, int st){
ll res = 1;
for (int i = 1; i <= st; i++){
res *= x;
res %= MOD;
if (i*2 <= st){
i *= 2;
res *= res;
res %= MOD;
}
}
return res;
}
int valid(int n, int x[]){
int j = -1;
for (int i = 0; i < n; i++){
if (x[i] > n)
continue;
j = i;
break;
}
if (j == -1)
return true;
vector<bool> visited(250001, 0);
for (int i = 0; i < n; i++){
if (visited[x[i]])
return false;
visited[x[i]] = true;
if (j > i){
int moze = x[j] - (j - i);
if (moze < 1)
moze += n;
if (x[i] <= n && moze != x[i]){
return false;
}
}
else{
int moze = x[j] + (i - j);
if (moze > n)
moze -= n;
if (x[i] <= n && moze != x[i]){
return false;
}
}
}
return true;
}
int replacement(int n, int x[], int y[]){
vector<ii> t;
vi visited(250001, false);
int j = 0;
for (int i = 0; i < n; i++){
visited[x[i]] = true;
if (x[i] > n)
t.push_back(make_pair(x[i], i));
else
j = i;
}
sort(t.begin(), t.end());
for (int i = 0; i < n; i++){
if (j > i){
int moze = x[j] - (j - i);
if (moze < 1)
moze += n;
x[i] = moze;
}
else{
int moze = x[j] + (i - j);
if (moze > n)
moze -= n;
x[i] = moze;
}
}
vi res;
for (int i = 0; i < t.size(); i++){
int curr = t[i].first, idx = t[i].second;
for (int j = x[idx]; j <= curr; j++){
if (!visited[j])
res.push_back(j);
visited[j] = true;
}
visited[curr] = true;
}
for (int i = 0; i < res.size(); i++)
y[i] = res[i];
return res.size();
}
int countReplacement(int n, int x[]){
bool k = valid(n, x);
if (!k)
return 0;
vector<ii> t;
int j = 0;
for (int i = 0; i < n; i++){
if (x[i] > n)
t.push_back(make_pair(x[i], i));
else
j = i;
}
for (int i = 0; i < n; i++){
if (j > i){
int moze = x[j] - (j - i);
if (moze < 1)
moze += n;
x[i] = moze;
}
else{
int moze = x[j] + (j - i);
if (moze > n)
moze -= n;
x[i] = moze;
}
}
sort(t.begin(), t.end());
ll res = 1;
int curr = t.size();
int prev = n;
for (int i = 0; i < t.size(); i++){
res *= stepen(t.size() - i, t[i].first - prev - 1);
res %= MOD;
prev = t[i].first;
}
return res;
}
int gondolaSequence[100001];
int replacementSequence[250001];
int main()
{
int i, n, tag;
int nr;
assert(scanf("%d", &tag)==1);
assert(scanf("%d", &n)==1);
for(i=0;i< n;i++)
assert( scanf("%d", &gondolaSequence[i]) ==1);
switch (tag){
case 1: case 2: case 3:
printf("%d\n", valid(n, gondolaSequence));
break;
case 4: case 5: case 6:
nr = replacement(n, gondolaSequence, replacementSequence);
printf("%d ", nr);
if (nr > 0)
{
for (i=0; i<nr-1; i++)
printf("%d ", replacementSequence[i]);
printf("%d\n", replacementSequence[nr-1]);
}
else printf("\n");
break;
case 7: case 8: case 9: case 10:
printf("%d\n", countReplacement(n, gondolaSequence));
break;
}
return 0;
}