Submission #1004855

#	Time	Username	Problem	Language	Result	Execution time	Memory
1004855		u_suck_o	Gondola (IOI14_gondola)	C++17	75 / 100	11 ms	2220 KiB

gondola

#include "bits/stdc++.h"
#include "gondola.h"
#define MOD 1000000009

using namespace std;

int valid(int n, int inputSeq[])
{
    bool seen[250001];
    memset(seen, 0, sizeof seen);
    bool found = false;
    int diff = -1;
    for (int i = 0; i < n; i++) {
        if (seen[inputSeq[i]]) return 0;
        seen[inputSeq[i]] = true;
        if (inputSeq[i] > n) continue;
        if (!found) {
            found = true;
            diff = (inputSeq[i] - i + n) % n;
        } else {
            if ((inputSeq[i] - i + n) % n != diff)
                return 0;
        }
    }
    return 1;
}

int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
    int diff = 0;
    for (int i = 0; i < n; i++) {
        if (gondolaSeq[i] <= n) {
            diff = (gondolaSeq[i] - i + n) % n;
            break;
        }
    }
    
    vector<pair<int, int>> v;
    for (int i = 0; i < n; i++) {
        if (gondolaSeq[i] > n) {
            int x = (i+diff) % n;
            if (x == 0) x = n;
            v.push_back({gondolaSeq[i], x});
        }
    }
    sort(v.begin(), v.end());
    int last = n; //last assigned gondola #
    vector<int> rep;
    for (int i = 0; i < v.size(); i++) {
        rep.push_back(v[i].second); //v[i].second replaced with last+1
        for (int j = last+1; j < v[i].first; j++)
            rep.push_back(j); //j replaced with j+1
        last = v[i].first;
    }
    for (int i = 0; i < rep.size(); i++)
        replacementSeq[i] = rep[i];
    return rep.size();
}

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

long long modpow(long long b, long long e) {
    if (e == 0)
        return 1;
    if (e % 2 == 1)
        return (b * modpow(b, e-1)) % MOD;
    long long r = modpow(b, e/2);
    return (r * r) % MOD;
}

int countReplacement(int n, int inputSeq[])
{
    if (valid(n, inputSeq) == 0) return 0;
    long long ans = 1;
    vector<int> v;
    for (int i = 0; i < n; i++) {
        if (inputSeq[i] > n)
            v.push_back(inputSeq[i]);
    }
    sort(v.begin(), v.end());
    int last = n;
    for (int i = 0; i < v.size(); i++) {
        //cout << modpow(v.size()-i, v[i] - last);
        ans = (ans * modpow(v.size()-i, v[i] - last - 1)) % MOD;
        last = v[i];
    }
    return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:49:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |     for (int i = 0; i < v.size(); i++) {
      |                     ~~^~~~~~~~~~
gondola.cpp:55:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |     for (int i = 0; i < rep.size(); i++)
      |                     ~~^~~~~~~~~~~~
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:82:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |     for (int i = 0; i < v.size(); i++) {
      |                     ~~^~~~~~~~~~

• 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