Submission #288405

#	Time	Username	Problem	Language	Result	Execution time	Memory
288405		Atill83	곤돌라 (IOI14_gondola)	C++14	45 / 100	22 ms	2364 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>
using namespace std;
const int MAXN = (int)3e5 + 5;
const int mod = (int) 1e9 + 9;
int yer[MAXN];
int valid(int n, int inputSeq[])
{
    for (int i = 1; i <= n; i++)
    {
        yer[i] = -1;
    }
    vector<int> ot;
    for (int i = 0; i < n; i++)
    {
        if (inputSeq[i] <= n)
        {
            if (yer[inputSeq[i]] != -1)
                return 0;
            yer[inputSeq[i]] = i;
        }
        else
        {
            ot.push_back(inputSeq[i]);
        }
    }
    sort(ot.begin(), ot.end());
    for (int i = 1; i < ot.size(); i++)
        if(ot[i] == ot[i - 1]) return 0;
    
    int last = -1, deg = 0;
    int bas = n;
    for (int i = 1; i <= n; i++){
        if (yer[i] == -1)
            continue;
        if(bas == n){
            bas = yer[i];
        }else if(yer[i] > bas && deg){
            return 0;
        }
        if (yer[i] < last){
            last = yer[i];
            if (deg || yer[i] > bas)
                return 0;
            deg = 1;
        }
        last = yer[i];
    }

    return 1;
}

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

int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
    int *p = gondolaSeq, * q = replacementSeq;
    for (int i = 1; i <= n; i++)
    {
        yer[i] = -1;
    }
    vector<int> ot;
    for (int i = 0; i < n; i++)
    {
        if (gondolaSeq[i] <= n)
        {
            if (yer[gondolaSeq[i]] != -1)
                return 0;
            yer[gondolaSeq[i]] = i;
        }
        else
        {
            ot.push_back(gondolaSeq[i]);
        }
    }
    int curi = 0;
    if((int)ot.size() == n){
        int mx = 0;
        int last = -1;
        for(int i = 0; i < n; i++){
            int cur = gondolaSeq[i];
            if(cur - n > mx){
                mx = cur - n;
                last = i + 1;
            }
            q[cur - n - 1] = i + 1;
        }
        q[mx - 1] = 0;
        for(int j = 0; j < mx; j++){
            if(q[j] == 0){
                q[j] = last;
                last = n + j + 1;
            }
        }
        return mx;
    }else{
        int piv = 0;
        for(int i = 1; i <= n; i++){
            if(yer[i] != -1){
                piv = i;
                break;
            }
        }
        
        int last = piv;
        last++;
        if(last == n + 1) last = 1;
        int mx = 0;
        int ono = -1;
        for(int j = 1; j < n; j++){
            int idx = yer[piv] + j;
            idx %= n;
            int cur = gondolaSeq[idx];
            if(cur > n){
                assert(last != 0);
                if(mx < cur - n){
                    mx = cur - n;
                    ono = last;
                }
                replacementSeq[cur - n - 1] = last;
            }else{
                assert(cur == last);
            }
            last++;
            if(last == n + 1) last = 1;
        }
        replacementSeq[mx - 1] = 0;

        for(int i = 0; i < mx; i++){
            if(replacementSeq[i] == 0){
                replacementSeq[i] = ono;
                ono = n + i + 1;
            }
        }
        return mx;
    }
}

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

long long bp(long long a, long long b){
    long long res = 1;
    while(b){
        if(b % 2) res = (res * a) % mod;
        a = (a * a) % mod;
        b = b / 2;
    }
    return res;
}

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

    sort(p, p + n);

    long long ans = 1;
    cerr<<"ali"<<endl;
    for(int i = 0; i < n - 1; i++){
        ans = ans*bp(n - i - 2, (p[i + 1] - p[i] - 1));
        ans %= mod;
    }
    return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:28:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |     for (int i = 1; i < ot.size(); i++)
      |                     ~~^~~~~~~~~~~
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:57:10: warning: unused variable 'p' [-Wunused-variable]
   57 |     int *p = gondolaSeq, * q = replacementSeq;
      |          ^
gondola.cpp:76:9: warning: unused variable 'curi' [-Wunused-variable]
   76 |     int curi = 0;
      |         ^~~~

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 0 / 10
• Subtask 4: 5 / 5
• Subtask 5: 10 / 10
• Subtask 6: 20 / 20
• Subtask 7: 0 / 5
• Subtask 8: 0 / 15
• Subtask 9: 0 / 15
• Subtask 10: 0 / 10