Submission #1064952

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
1064952	2024-08-18T20:01:50 Z	kkzyr	Gondola (IOI14_gondola)	C++17	Compilation error	0 ms	0 KB

gondola

#include <iostream>
#include <algorithm>
using namespace std;

const int mod = 1e9 + 7;

int input[100000];

bool has[100001];

int original[100001];
int index[250001];

int use_for_replacement[100000];

long long binary_exponentiation(int a, int b){
    if (b == 0){
        return 1;
    }
    long long cache = binary_exponentiation(a, b/2);
    long long ans = cache * cache;
    ans %= mod;
    if (b % 2 == 1){
        ans *= a;
        ans %= mod;
    }
    return ans;
}

int valid(int n, int inputSeq[]){
    for (int i = 0;i < n;i++){
        if (i <= n){
            if (has[i]){
                return 0;
            }
            has[i] = true;
        }
    }
    int first_original = -1;
    for (int i = 0;i < n;i++){
        if (inputSeq[i] <= n){
            first_original = i;
            break;
        }
    }
    if (first_original == -1){
        return 1;
    }
    for (int i = 0;i < n;i++){
        if (inputSeq[i] <= n){
            int diff1 = i - first_original;
            int diff2 = inputSeq[i] - inputSeq[first_original];
            if (diff2 < 0){
                diff2 += n;
            }
            if (diff1 != diff2){
                return 0;
            }
        }
    }
    return 1;
}

int replacement(int n, int gondolaSeq[], int replacementSeq[]){
    int has_index = -1;
    int max_num = -1;
    for (int i = 0;i < n;i++){
        if (gondolaSeq[i] <= n){
            has_index = i;
        }
        max_num = max(max_num, gondolaSeq[i]);
        index[gondolaSeq[i]] = i;
    }
    if (has_index == -1){
        for (int i = 0;i < n;i++){
            original[i] = i + 1;
        }
    }
    else{
        int first = gondolaSeq[has_index] - has_index;
        if (first < 0){
            first += n;
        }
        for (int i = 0;i < n;i++){
            original[i] = (first + i) % n;
            if (original[i] == 0){
                original[i] = n;
            }
        }
    }
    int length = 0;
    for (int i = n + 1;i <= max_num;i++){
        if (index[i] != 0 or gondolaSeq[0] == i){
            replacementSeq[length] = original[index[i]];
            length++;
            original[index[i]] = i;
        }
        else{
            replacementSeq[length] = original[index[max_num]];
            length++;
            original[index[max_num]] = i;
        }
    }
    return length;
}

int countReplacement(int n, int inputSeq[]){
    if (valid(n, inputSeq) == 0){
        return 0;
    }
    sort(inputSeq, inputSeq + n);
    long long ans = 1;
    if (inputSeq[0] > n){
        ans = n;
    }
    for (int i = 0;i < n - 1;i++){
        if (inputSeq[i + 1] > n){
            ans *= binary_exponentiation(n - 1 - i, inputSeq[i + 1] - max(inputSeq[i], n) - 1);
            ans %= mod;
        }
    }
    return ans;
}

int main(){
    int t, n;
    cin >> t >> n;
    for (int i = 0;i < n;i++){
        cin >> input[i];
    }
    if (t <= 3){
        cout << valid(n, input) << '\n';
    }
    else if (t <= 6){
        int length = replacement(n, index, use_for_replacement);
        cout << length << '\n';
        for (int i = 0;i < length;i++){
            cout << use_for_replacement[i] << ' ';
        }
        cout << '\n';
    }
    else{
        cout << countReplacement(n, input);
    }
    return 0;
}

Compilation message

/usr/bin/ld: /tmp/cchmxUtZ.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccOTFToY.o:gondola.cpp:(.text.startup+0x0): first defined here
/usr/bin/ld: /tmp/cchmxUtZ.o: in function `main':
grader.cpp:(.text.startup+0xb6): undefined reference to `valid'
/usr/bin/ld: grader.cpp:(.text.startup+0x108): undefined reference to `countReplacement'
/usr/bin/ld: grader.cpp:(.text.startup+0x132): undefined reference to `replacement'
collect2: error: ld returned 1 exit status