Submission #316189

#	Time	Username	Problem	Language	Result	Execution time	Memory
316189		qpwoeirut	Gondola (IOI14_gondola)	C++17	60 / 100	24 ms	2816 KiB

gondola

#include <bits/stdc++.h>
#include "gondola.h"

using namespace std;

typedef long long ll;

const int MN = 250001;
const ll MOD = 1e9 + 9;

bool normalize(int n, int seq[]) {
    for (int i=0; i<n; ++i) {
        if (seq[i] <= n) {
            int val = seq[i];
            int idx = (n + i - val + 1) % n;
            rotate(seq, seq+idx, seq+n);
            return true;
        }
    }
    return false;
}

bool seen[MN];
int valid(int n, int inputSeq[]) {
    fill(seen, seen+n, false);
    normalize(n, inputSeq);
    //for (int i=0; i<n; ++i) { cerr << inputSeq[i] << ' '; } cerr << endl;

    for (int i=0; i<n; ++i) {
        if (inputSeq[i] != i+1 && inputSeq[i] <= n) {
            return 0;
        }
        if (seen[inputSeq[i]]) {
            return 0;
        }
        seen[inputSeq[i]] = true;
    }

    return 1;
}

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

int pos[MN];
pair<int,int> process(int n, int seq[]) {
    int mxv = 0;
    int mxi = -1;
    for (int i=0; i<n; ++i) {
        pos[seq[i]] = i;
        seen[seq[i]] = true;
        if (mxv < seq[i]) {
            mxv = seq[i];
            mxi = i;
        }
    }
    return pair<int,int>(mxv, mxi);
}

int replacement(int n, int gondolaSeq[], int replacementSeq[]) {
    fill(seen, seen+n, false);
    normalize(n, gondolaSeq);

    auto mx = process(n, gondolaSeq);
    int mxv = mx.first;
    int cur = mx.second + 1;

    int idx = 0;
    for (int i=n+1; i<=mxv; ++i) {
        if (!seen[i] || i == mxv) {
            replacementSeq[idx++] = cur;
            cur = i;
        } else {
            replacementSeq[idx++] = pos[i] + 1;
        }
    }

    return idx;
}

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

int countReplacement(int n, int inputSeq[]) {
    bool rotated = normalize(n, inputSeq);
    if (!valid(n, inputSeq)) {
        return 0;
    }

    int mxv = process(n, inputSeq).first;
    set<int> nums;
    for (int i=0; i<n; ++i) {
        if (inputSeq[i] > n) {
            nums.insert(inputSeq[i]);
        }
    }

    ll ans = 1;
    for (int i=n+1; i<=mxv; ++i) {
        if (seen[i]) {
            nums.erase(i);
            continue;
        }
        ll more = n - (ll)nums.size();
        ans = (ans * more) % MOD;
    }

    return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:83:10: warning: unused variable 'rotated' [-Wunused-variable]
   83 |     bool rotated = normalize(n, inputSeq);
      |          ^~~~~~~

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