Submission #921180

#	Time	Username	Problem	Language	Result	Execution time	Memory
921180		Macker	곤돌라 (IOI14_gondola)	C++17	100 / 100	19 ms	2396 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>
 
using namespace std;
typedef long long ll;
#define all(v) v.begin(), v.end()

ll mod = 1000000009;

ll power(ll n, ll k){
    ll res = 1;
    n %= mod;
    while(k > 0){
        if(k & 1) res = (res * n) % mod;
        n = (n * n) % mod;
        k /= 2;
    }
    return res;
}

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

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

int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
    vector<pair<int, int>> v(n);
    for (int i = 0; i < n; i++) {
        v[i] = {gondolaSeq[i], i};
    }
    sort(all(v));

    ll strt = -1;
    for (auto &i : v) {
        if(i.first <= n){
            strt = (i.second - i.first + 1 + n) % n;
            break;
        }
    }
    if(strt == -1) strt = 0;
    for (int i = 0; i < n; i++) {
        gondolaSeq[i] = (i - strt + n) % n + 1;
    }
    
    
    int ls = n;
    int j = 0;
    for (auto &i : v) {
        while(ls < i.first){
            replacementSeq[j++] = gondolaSeq[i.second];
            ls++;
            gondolaSeq[i.second] = ls;
        }
    }
    return j;
}

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


int countReplacement(int n, int inputSeq[])
{
    vector<pair<int, int>> v(n);
    for (int i = 0; i < n; i++) {
        v[i] = {inputSeq[i], i};
    }
    sort(all(v));
    for (int i = 1; i < v.size(); i++) {
        if(v[i].first == v[i - 1].first) return 0;
    }
    
    ll strt = -1;
    ll fin = 0;
    bool m = true;
    ll res = 1;
    for (int j = 0; j < v.size(); j++) {
        auto i = v[j];
        if(i.first <= n){
            if(strt == -1){
                strt = (i.second - i.first + 1 + n) % n;
            }
            else{
                if((i.second - i.first + 1 + n) % n != strt) return 0;
            }
            m = false;
        }
        else{
            ll ls;
            if(j == 0 || v[j - 1].first <= n){
                ls = n + 1;
            }
            else ls = v[j - 1].first + 1;
            res = (res * power(n - fin, i.first - ls)) % mod;
        }
        fin++;
    }
    if(m) res = (res * n) % mod;
    return res;
}

Compilation message (stderr)

gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:71: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]
   71 |     for (int i = 1; i < v.size(); i++) {
      |                     ~~^~~~~~~~~~
gondola.cpp:79: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]
   79 |     for (int j = 0; j < v.size(); j++) {
      |                     ~~^~~~~~~~~~

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