#include "gondola.h"
#include <bits/stdc++.h>
using namespace std;
const int nax = 1e5 + 3;
const int mod = 1e9 + 9;
map<int,int>cnt;
int expected[nax];
void get_expected(int n, int inputSeq[]) {
for(int i=0; i<n; ++i) {
expected[i] = 0;
}
int min_at = min_element(inputSeq, inputSeq+n) - inputSeq;
int cur_val = inputSeq[min_at];
int cur_id = min_at;
for(int i=cur_id; ; ++i, ++cur_val) {
if(i==n) i = 0;
if(cur_val>n) cur_val = 1;
if(expected[i]) break;
expected[i] = cur_val;
}
}
int valid(int n, int inputSeq[]) {
cnt.clear();
for(int i=0; i<n; ++i) {
cnt[inputSeq[i]]++;
if(cnt[inputSeq[i]]>1) {
return 0;
}
}
get_expected(n, inputSeq);
for(int i=0; i<n; ++i) if(inputSeq[i]<=n) {
if(inputSeq[i]!=expected[i]) return 0;
}
return 1;
}
//----------------------
int replacement(int n, int gondolaSeq[], int replacementSeq[]) {
get_expected(n, gondolaSeq);
vector<pair<int,int>>v;
for(int i=0; i<n; ++i) if(gondolaSeq[i]>n) {
v.emplace_back(gondolaSeq[i], expected[i]);
}
sort(v.begin(), v.end());
int l = 0;
int at = 0;
while(at<v.size()) {
int cur_gondola = v[at].second;
while(l+n+1<=v[at].first) {
replacementSeq[l] = cur_gondola;
cur_gondola = l+n+1;
++l;
}
++at;
}
return l;
}
//----------------------
int mul(int x, int y) {
return (x * 1LL * y) % mod;
}
int quickpow(int a, int b) {
int ret = 1;
while(b) {
if(b&1) ret = mul(ret, a);
a = mul(a, a);
b >>= 1;
}
return ret;
}
int countReplacement(int n, int inputSeq[]) {
if(!valid(n, inputSeq)) return 0;
int mx = 0, gtn = 0;
vector<int>v;
for(int i=0; i<n; ++i) {
mx = max(mx, inputSeq[i]);
if(inputSeq[i]>n) {
++gtn;
v.push_back(inputSeq[i]);
}
}
v.push_back(n);
sort(v.begin(), v.end());
long long ans = 1;
for(int i=1; i<v.size(); ++i) {
int cnt = v[i] - v[i-1] - 1;
int option = gtn - i + 1;
ans *= quickpow(option, cnt);
ans %= mod;
}
if(gtn==n) {
ans *= n;
ans %= mod;
}
return ans;
}
Compilation message
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:63:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
63 | while(at<v.size()) {
| ~~^~~~~~~~~
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:108:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
108 | for(int i=1; i<v.size(); ++i) {
| ~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
12 ms |
2260 KB |
Output is correct |
7 |
Correct |
13 ms |
596 KB |
Output is correct |
8 |
Correct |
29 ms |
4204 KB |
Output is correct |
9 |
Correct |
9 ms |
1492 KB |
Output is correct |
10 |
Correct |
33 ms |
4808 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
12 ms |
2344 KB |
Output is correct |
7 |
Correct |
7 ms |
620 KB |
Output is correct |
8 |
Correct |
20 ms |
4180 KB |
Output is correct |
9 |
Correct |
7 ms |
1492 KB |
Output is correct |
10 |
Correct |
28 ms |
4836 KB |
Output is correct |
11 |
Correct |
0 ms |
212 KB |
Output is correct |
12 |
Correct |
0 ms |
212 KB |
Output is correct |
13 |
Correct |
17 ms |
2140 KB |
Output is correct |
14 |
Correct |
0 ms |
212 KB |
Output is correct |
15 |
Correct |
48 ms |
4940 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
1 ms |
340 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
1 ms |
340 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
8 ms |
912 KB |
Output is correct |
12 |
Correct |
8 ms |
980 KB |
Output is correct |
13 |
Correct |
12 ms |
1364 KB |
Output is correct |
14 |
Correct |
7 ms |
852 KB |
Output is correct |
15 |
Correct |
28 ms |
2204 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
39 ms |
4692 KB |
Output is correct |
10 |
Correct |
30 ms |
3924 KB |
Output is correct |
11 |
Correct |
11 ms |
1728 KB |
Output is correct |
12 |
Correct |
17 ms |
1936 KB |
Output is correct |
13 |
Correct |
3 ms |
724 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
42 ms |
4636 KB |
Output is correct |
10 |
Correct |
28 ms |
3936 KB |
Output is correct |
11 |
Correct |
10 ms |
1620 KB |
Output is correct |
12 |
Correct |
15 ms |
1988 KB |
Output is correct |
13 |
Correct |
3 ms |
724 KB |
Output is correct |
14 |
Correct |
48 ms |
6336 KB |
Output is correct |
15 |
Correct |
69 ms |
6984 KB |
Output is correct |
16 |
Correct |
14 ms |
1592 KB |
Output is correct |
17 |
Correct |
37 ms |
4712 KB |
Output is correct |
18 |
Correct |
18 ms |
2984 KB |
Output is correct |