Submission #1243490

#	Time	Username	Problem	Language	Result	Execution time	Memory
1243490		inkvizytor	곤돌라 (IOI14_gondola)	C++17	75 / 100	11 ms	2112 KiB

gondola

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

#define ll long long

int valid(int n, int inputSeq[])
{
    int p = 0;
    for (int i = 0; i < n; i++) {
        if (inputSeq[i] <= n) {
            p = (inputSeq[i]+n-i)%n;
            break;
        }
    }
    if (p == 0) {
        sort(inputSeq, inputSeq+n);
        for (int i = 1; i < n; i++) {
            if (inputSeq[i] == inputSeq[i-1]) {
                return 0;
            }
        }
        return 1;
    }
    for (int i = 0; i < n; i++) {
        if (inputSeq[i] <= n && inputSeq[i]%n != p%n) {
            return 0;
        }
        p++;
    }
    sort(inputSeq, inputSeq+n);
    for (int i = 1; i < n; i++) {
        if (inputSeq[i] == inputSeq[i-1]) {
            return 0;
        }
    }
    return 1;
}

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

int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
    int p = 0;
    for (int i = 0; i < n; i++) {
        if (gondolaSeq[i] <= n) {
            p = i-(gondolaSeq[i]-1);
            if (p < 0) {
                p+=n;
            }
            break;
        }
    }
    vector<pair<int, int>> x;
    for (int i = 0; i < n; i++) {
        if (gondolaSeq[i] > n) {
            x.push_back({gondolaSeq[i], i});
        }
    }
    sort(x.begin(), x.end());
    int w = n, in = 0;
    for (auto i : x) {
        int k = i.second-p+1;
        if (k <= 0) {
            k += n;
        }
        replacementSeq[in] = k;
        in++;
        w++;
        while (w < i.first) {
            replacementSeq[in] = w;
            in++;
            w++;
        }
    }
    return in;
}

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

int M = 1e9+9;

ll pow(ll a, ll b) {
    if (b == 0) {
        return 1;
    }
    ll x = pow(a, b/2);
    x = (x*x)%M;
    if (b%2) {
        x *= a;
    }
    return x%M;
}

int countReplacement(int n, int inputSeq[])
{
    if (!valid(n, inputSeq)) {
        return 0;
    }
    ll w = 1, l = 0;
    vector<ll> x;
    for (int i = 0; i < n; i++) {
        if (inputSeq[i] > n) {
            x.push_back(inputSeq[i]);
            l++;
        }
    }
    sort(x.begin(), x.end());
    int p = n;
    for (ll i : x) {
        w *= pow(l, i-p-1);
        w %= M;
        l--;
        p = i;
    }
    return (int)w;
}

• 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: 5 / 5
• Subtask 8: 15 / 15
• Subtask 9: 0 / 15
• Subtask 10: 0 / 10