#include "gondola.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 250100;
const ll mod = 1000000009;
map<int, bool>exist;
int fin_val[maxn];
ll cnt[maxn];
int valid(int n, int arr[]) {
bool check = true;
for(int i=1;i<n;i++) {
if(arr[i] > n) continue;
else if(arr[i] == 1) {
if(arr[i-1] < n)
check = false;
}
else {
if(arr[i-1] <= n && arr[i-1] != arr[i] - 1) check = false;
}
}
sort(arr, arr+n);
for(int i=1;i<n;i++) {
if(arr[i] == arr[i-1]) check = false;
}
if(check) return 1;
else return 0;
}
//----------------------
int replacement(int n, int arr[], int replacementSeq[]) {
int maxm = 0;
int found_i = 0, st_val = 1;
for(int i=0;i<n;i++) {
exist[arr[i]] = true;
if(arr[i] <= n) {
found_i = i;
st_val = arr[i];
}
maxm = max(maxm, arr[i]);
}
exist[n] = true;
int l = maxm - n;
for(int i=found_i;i<n;i++) {
fin_val[i] = st_val;
st_val++;
if(st_val == n + 1) st_val = 1;
}
for(int i=0;i<found_i;i++) {
fin_val[i] = st_val;
st_val++;
if(st_val == n + 1) st_val = 1;
}
for(int i=0;i<n;i++) {
while(!exist[arr[i]-1] && arr[i] > n) {
replacementSeq[arr[i]-n-1] = arr[i] - 1;
arr[i]--;
}
if(arr[i] > n) replacementSeq[arr[i]-n-1] = fin_val[i];
}
return l;
}
//----------------------
ll power(ll a, ll b) {
if(b == 0) return 1LL;
else if(b == 1) return 1LL * a % mod;
else {
if(b%2 == 0) {
ll half = power(a, b/2) % mod;
return (1LL*half * half)% mod;
}
else if(b%2==1) {
return (1LL*a*power(a, b-1)) % mod;
}
}
}
int countReplacement(int n, int arr[]) {
if(!valid(n, arr)) return 0;
vector<ll>nums;
for(int i=0;i<n;i++) {
if(arr[i] > n) nums.push_back(arr[i]);
}
nums.push_back(n);
sort(nums.begin(), nums.end());
ll result = 1LL;
for(int i=1;i<nums.size();i++) {
ll diff = 1LL * nums[i] - nums[i-1] - 1;
ll p = 1LL * int(nums.size()) - i;
result *= 1LL*power(p, diff);
result %= mod;
}
if(nums.size() == n + 1) {
result *= n;
result %= mod;
}
return result;
}
/*int main() {
int arr[] = {10, 20, 30, 40, 50, 60, 70};
int arr2[1];
cout<<countReplacement(7, arr);
}*/
Compilation message
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:99:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=1;i<nums.size();i++) {
~^~~~~~~~~~~~
gondola.cpp:106:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if(nums.size() == n + 1) {
~~~~~~~~~~~~^~~~~~~~
gondola.cpp: In function 'long long int power(long long int, long long int)':
gondola.cpp:87:1: warning: control reaches end of non-void function [-Wreturn-type]
}
^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
376 KB |
Output is correct |
2 |
Correct |
2 ms |
376 KB |
Output is correct |
3 |
Correct |
3 ms |
484 KB |
Output is correct |
4 |
Correct |
2 ms |
484 KB |
Output is correct |
5 |
Correct |
2 ms |
496 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
504 KB |
Output is correct |
2 |
Correct |
2 ms |
528 KB |
Output is correct |
3 |
Correct |
3 ms |
528 KB |
Output is correct |
4 |
Correct |
3 ms |
528 KB |
Output is correct |
5 |
Correct |
2 ms |
528 KB |
Output is correct |
6 |
Correct |
10 ms |
656 KB |
Output is correct |
7 |
Correct |
21 ms |
912 KB |
Output is correct |
8 |
Correct |
14 ms |
912 KB |
Output is correct |
9 |
Correct |
6 ms |
912 KB |
Output is correct |
10 |
Correct |
20 ms |
912 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
912 KB |
Output is correct |
2 |
Correct |
2 ms |
912 KB |
Output is correct |
3 |
Correct |
2 ms |
912 KB |
Output is correct |
4 |
Correct |
2 ms |
912 KB |
Output is correct |
5 |
Correct |
3 ms |
912 KB |
Output is correct |
6 |
Correct |
11 ms |
912 KB |
Output is correct |
7 |
Correct |
21 ms |
912 KB |
Output is correct |
8 |
Correct |
15 ms |
912 KB |
Output is correct |
9 |
Correct |
11 ms |
912 KB |
Output is correct |
10 |
Correct |
28 ms |
912 KB |
Output is correct |
11 |
Correct |
2 ms |
912 KB |
Output is correct |
12 |
Correct |
2 ms |
912 KB |
Output is correct |
13 |
Correct |
11 ms |
912 KB |
Output is correct |
14 |
Correct |
3 ms |
912 KB |
Output is correct |
15 |
Correct |
25 ms |
960 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
960 KB |
Output is correct |
2 |
Correct |
2 ms |
960 KB |
Output is correct |
3 |
Correct |
2 ms |
960 KB |
Output is correct |
4 |
Correct |
3 ms |
960 KB |
Output is correct |
5 |
Correct |
3 ms |
960 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
960 KB |
Output is correct |
2 |
Correct |
3 ms |
960 KB |
Output is correct |
3 |
Correct |
3 ms |
960 KB |
Output is correct |
4 |
Correct |
3 ms |
960 KB |
Output is correct |
5 |
Correct |
3 ms |
960 KB |
Output is correct |
6 |
Correct |
2 ms |
960 KB |
Output is correct |
7 |
Correct |
4 ms |
960 KB |
Output is correct |
8 |
Correct |
5 ms |
960 KB |
Output is correct |
9 |
Correct |
3 ms |
960 KB |
Output is correct |
10 |
Correct |
3 ms |
960 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
960 KB |
Output is correct |
2 |
Correct |
3 ms |
960 KB |
Output is correct |
3 |
Correct |
2 ms |
960 KB |
Output is correct |
4 |
Correct |
3 ms |
960 KB |
Output is correct |
5 |
Correct |
3 ms |
960 KB |
Output is correct |
6 |
Correct |
2 ms |
960 KB |
Output is correct |
7 |
Correct |
4 ms |
960 KB |
Output is correct |
8 |
Correct |
5 ms |
960 KB |
Output is correct |
9 |
Correct |
4 ms |
960 KB |
Output is correct |
10 |
Correct |
4 ms |
960 KB |
Output is correct |
11 |
Correct |
41 ms |
5036 KB |
Output is correct |
12 |
Correct |
53 ms |
5548 KB |
Output is correct |
13 |
Correct |
65 ms |
5548 KB |
Output is correct |
14 |
Correct |
47 ms |
5548 KB |
Output is correct |
15 |
Correct |
76 ms |
10536 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
10536 KB |
Output is correct |
2 |
Correct |
2 ms |
10536 KB |
Output is correct |
3 |
Correct |
2 ms |
10536 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
10536 KB |
Output is correct |
2 |
Correct |
2 ms |
10536 KB |
Output is correct |
3 |
Correct |
2 ms |
10536 KB |
Output is correct |
4 |
Correct |
2 ms |
10536 KB |
Output is correct |
5 |
Correct |
3 ms |
10536 KB |
Output is correct |
6 |
Correct |
3 ms |
10536 KB |
Output is correct |
7 |
Correct |
3 ms |
10536 KB |
Output is correct |
8 |
Correct |
3 ms |
10536 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
10536 KB |
Output is correct |
2 |
Correct |
2 ms |
10536 KB |
Output is correct |
3 |
Correct |
2 ms |
10536 KB |
Output is correct |
4 |
Correct |
3 ms |
10536 KB |
Output is correct |
5 |
Correct |
2 ms |
10536 KB |
Output is correct |
6 |
Correct |
2 ms |
10536 KB |
Output is correct |
7 |
Correct |
3 ms |
10536 KB |
Output is correct |
8 |
Correct |
3 ms |
10536 KB |
Output is correct |
9 |
Correct |
29 ms |
10536 KB |
Output is correct |
10 |
Correct |
23 ms |
10536 KB |
Output is correct |
11 |
Correct |
11 ms |
10536 KB |
Output is correct |
12 |
Correct |
11 ms |
10536 KB |
Output is correct |
13 |
Correct |
6 ms |
10536 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
10536 KB |
Output is correct |
2 |
Correct |
3 ms |
10536 KB |
Output is correct |
3 |
Correct |
3 ms |
10536 KB |
Output is correct |
4 |
Correct |
3 ms |
10536 KB |
Output is correct |
5 |
Correct |
2 ms |
10536 KB |
Output is correct |
6 |
Correct |
3 ms |
10536 KB |
Output is correct |
7 |
Correct |
2 ms |
10536 KB |
Output is correct |
8 |
Correct |
2 ms |
10536 KB |
Output is correct |
9 |
Correct |
30 ms |
10536 KB |
Output is correct |
10 |
Correct |
33 ms |
10536 KB |
Output is correct |
11 |
Correct |
11 ms |
10536 KB |
Output is correct |
12 |
Correct |
14 ms |
10536 KB |
Output is correct |
13 |
Correct |
5 ms |
10536 KB |
Output is correct |
14 |
Correct |
50 ms |
10536 KB |
Output is correct |
15 |
Correct |
43 ms |
10536 KB |
Output is correct |
16 |
Correct |
9 ms |
10536 KB |
Output is correct |
17 |
Correct |
29 ms |
10536 KB |
Output is correct |
18 |
Correct |
19 ms |
10536 KB |
Output is correct |