# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
1064973 |
2024-08-18T20:35:26 Z |
kkzyr |
Gondola (IOI14_gondola) |
C++17 |
|
0 ms |
0 KB |
#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/ccVfywvz.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