Submission #1066597

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
1066597	2024-08-20T03:01:40 Z	Charizard2021	Gondola (IOI14_gondola)	C++17	75 / 100	28 ms	6852 KB

gondola

#include "gondola.h"
#include<bits/stdc++.h>
using namespace std;
int valid(int n, int inputSeq[]){
    set<int> s;
    bool works = true;
    int idx = 0;
    for(int i = 0; i < n; i++){
        s.insert(inputSeq[i]);
        if(inputSeq[i] <= n){
            idx = i;
            works = false;
        }
    }
    if((int)s.size() != n){
        return 0;
    }
    if(works){
        return 1;
    }
    int a[n];
    int val = inputSeq[idx];
    for(int i = idx; i < n; i++){
        a[i] = val++;
        val %= n;
        if(val == 0) val = n;
    }
    for(int i = 0; i < idx; i++){
        a[i] = val++;
        val %= n;
        if(val == 0) val = n;
    }
    bool works2 = true;
    for(int i = 0; i < n; i++){
        if(inputSeq[i] <= n){
            if(inputSeq[i] != a[i]){
                works2 = false;
            }
        }
    }
    if(works2){
        return 1;
    }
    return 0;
}

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

int replacement(int n, int gondolaSeq[], int replacementSeq[]){
    //3, 8, 5, 6, 2
    int maxVal = 0;
    int minVal = INT_MAX;
    for(int x = 0; x < n; x++){
        maxVal = max(maxVal, gondolaSeq[x]);
        minVal = min(minVal, gondolaSeq[x]);
    }
    if(maxVal <= n){
        return 0;
    }
    if(minVal > n){
        int x[maxVal + 1];
        for(int a = n + 1; a <= maxVal; a++){
            x[a] = 0;
        }
        int idx = 0;
        for(int i = 0; i < n; i++){
            if(gondolaSeq[i] <= n){
                idx = i;
            }
        }
        int a[n];
        for(int i = 0; i < n; i++){
            a[i] = i + 1;
        }
        for(int i = 0; i < n; i++){
            if(gondolaSeq[i] > n){
                x[gondolaSeq[i]] = a[i];
            }
        }
        for(int i = maxVal; i >= n + 1; i--){
            if(x[i] == 0){
                x[i] = x[i + 1];
            }
        }
        map<int, vector<int> > mp;
        for(int i = n + 1; i <= maxVal; i++){
            mp[x[i]].push_back(i);
        }
        int val9 = 0;
        int idx9 = 0;
        vector<pair<int, vector<int> > > v9;
        for(auto it : mp){
            val9 += (int)it.second.size();
            v9.push_back(make_pair(it.second[0], (it.second)));
        }
        sort(v9.begin(), v9.end());
        //6, 7, 8, 9
        for(pair<int, vector<int> > c : v9){
            replacementSeq[idx9++] = x[c.first];
            for(int a : c.second){
                if(a == c.second.back()){
                    break;
                }
                replacementSeq[idx9++] = a;
            }
        }
        return val9;
    }
    int x[maxVal + 1];
    for(int a = n + 1; a <= maxVal; a++){
        x[a] = 0;
    }
    int idx = 0;
    for(int i = 0; i < n; i++){
        if(gondolaSeq[i] <= n){
            idx = i;
        }
    }
    int a[n];
    int val = gondolaSeq[idx];
    for(int i = idx; i < n; i++){
        a[i] = val++;
        val %= n;
        if(val == 0) val = n;
    }
    for(int i = 0; i < idx; i++){
        a[i] = val++;
        val %= n;
        if(val == 0) val = n;
    }
    for(int i = 0; i < n; i++){
        if(gondolaSeq[i] > n){
            x[gondolaSeq[i]] = a[i];
        }
    }
    for(int i = maxVal; i >= n + 1; i--){
        if(x[i] == 0){
            x[i] = x[i + 1];
        }
    }
    map<int, vector<int> > mp;
    for(int i = n + 1; i <= maxVal; i++){
        mp[x[i]].push_back(i);
    }
    int val9 = 0;
    int idx9 = 0;
    vector<pair<int, vector<int> > > v9;
    for(auto it : mp){
        val9 += (int)it.second.size();
        v9.push_back(make_pair(it.second[0], (it.second)));
    }
    sort(v9.begin(), v9.end());
    //6, 7, 8, 9
    for(pair<int, vector<int> > c : v9){
        replacementSeq[idx9++] = x[c.first];
        for(int a : c.second){
            if(a == c.second.back()){
                break;
            }
            replacementSeq[idx9++] = a;
        }
    }
    return val9;
}

//----------------------
const long long MOD = 1e9 + 9;
long long binaryExponention(long long a, long long b){
    a %= MOD;
    long long ans = 1;
    while(b > 0){
        if(b % 2 == 1){
            ans = (ans * a) % MOD;
        }
        a = (a * a) % MOD;
        b /= 2LL;
    }
    return ans;
}
int countReplacement(int n, int inputSeq[]){
    if(valid(n, inputSeq) == 0) return 0;
    bool works = true;
    vector<long long> broke;
    for(long long i = 0; i < n; i++){
        if(inputSeq[i] < n){
            works = false;
        }
    }
    for(int i = 0; i < n; i++){
        if(inputSeq[i] >= n){
            broke.push_back(inputSeq[i]);
        }
    }
    sort(broke.begin(), broke.end());
    reverse(broke.begin(), broke.end());
    long long ans = 1;
    for(long long i = 0; i < (long long)broke.size(); i++){
        if(i != (long long)broke.size() - 1){
            long long val = broke[i] - broke[i + 1] - 1;
            long long val2 = binaryExponention(i + 1, val);
            ans = (ans * val2) % MOD;
        }
        else{
            long long val = broke[i] - n;
            long long val2 = binaryExponention((long long)broke.size(), val);
            ans = (ans * val2) % MOD;
        }
    }
    if(works){
        ans = (ans * (long long)n) % MOD;
    }
    return ans;
}

Compilation message

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:65:13: warning: variable 'idx' set but not used [-Wunused-but-set-variable]
   65 |         int idx = 0;
      |             ^~~

Subtask #1
5.0 / 5.0

#	Verdict	Memory	Grader output
1	Correct	348 KB	Output is correct
2	Correct	348 KB	Output is correct
3	Correct	348 KB	Output is correct
4	Correct	440 KB	Output is correct
5	Correct	344 KB	Output is correct

Subtask #2
5.0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	1 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	7 ms	2536 KB	Output is correct
7	Correct	20 ms	4176 KB	Output is correct
8	Correct	16 ms	4700 KB	Output is correct
9	Correct	5 ms	1628 KB	Output is correct
10	Correct	19 ms	5468 KB	Output is correct

Subtask #3
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	440 KB	Output is correct
4	Correct	0 ms	444 KB	Output is correct
5	Correct	0 ms	344 KB	Output is correct
6	Correct	8 ms	2652 KB	Output is correct
7	Correct	19 ms	4232 KB	Output is correct
8	Correct	13 ms	4700 KB	Output is correct
9	Correct	5 ms	1592 KB	Output is correct
10	Correct	18 ms	5464 KB	Output is correct
11	Correct	1 ms	348 KB	Output is correct
12	Correct	0 ms	348 KB	Output is correct
13	Correct	10 ms	2396 KB	Output is correct
14	Correct	0 ms	448 KB	Output is correct
15	Correct	26 ms	5652 KB	Output is correct

Subtask #4
5.0 / 5.0

#	Verdict	Memory	Grader output
1	Correct	344 KB	Output is correct
2	Correct	348 KB	Output is correct
3	Correct	348 KB	Output is correct
4	Correct	348 KB	Output is correct
5	Correct	600 KB	Output is correct

Subtask #5
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	444 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	1 ms	344 KB	Output is correct
8	Correct	1 ms	348 KB	Output is correct
9	Correct	1 ms	344 KB	Output is correct
10	Correct	1 ms	348 KB	Output is correct

Subtask #6
20.0 / 20.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	1 ms	348 KB	Output is correct
9	Correct	1 ms	348 KB	Output is correct
10	Correct	1 ms	348 KB	Output is correct
11	Correct	5 ms	1116 KB	Output is correct
12	Correct	6 ms	1256 KB	Output is correct
13	Correct	28 ms	6852 KB	Output is correct
14	Correct	5 ms	1116 KB	Output is correct
15	Correct	26 ms	6776 KB	Output is correct

Subtask #7
5.0 / 5.0

#	Verdict	Memory	Grader output
1	Correct	348 KB	Output is correct
2	Correct	440 KB	Output is correct
3	Correct	348 KB	Output is correct

Subtask #8
15.0 / 15.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	344 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	344 KB	Output is correct
5	Correct	1 ms	600 KB	Output is correct
6	Correct	1 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	1 ms	348 KB	Output is correct

Subtask #9
0 / 15.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Incorrect	26 ms	4932 KB	Output isn't correct
10	Halted	0 ms	0 KB	-

Subtask #10
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	444 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Incorrect	26 ms	4696 KB	Output isn't correct
10	Halted	0 ms	0 KB	-