CSES - Putka Open 2020 – 1/5 - Results
Submission details
Task:Vaihdot
Sender:tykkipeli
Submission time:2020-09-06 03:04:32 +0300
Language:C++ (C++11)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED13
#2ACCEPTED23
#3ACCEPTED64
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 3details
#2ACCEPTED0.01 s1, 2, 3details
#3ACCEPTED0.01 s1, 2, 3details
#4ACCEPTED0.01 s1, 2, 3details
#5ACCEPTED0.01 s1, 2, 3details
#6ACCEPTED0.01 s1, 2, 3details
#7ACCEPTED0.01 s1, 2, 3details
#8ACCEPTED0.01 s1, 2, 3details
#9ACCEPTED0.01 s2, 3details
#10ACCEPTED0.01 s2, 3details
#11ACCEPTED0.01 s2, 3details
#12ACCEPTED0.01 s2, 3details
#13ACCEPTED0.01 s2, 3details
#14ACCEPTED0.01 s2, 3details
#15ACCEPTED0.01 s2, 3details
#16ACCEPTED0.01 s2, 3details
#17ACCEPTED0.07 s3details
#18ACCEPTED0.13 s3details
#19ACCEPTED0.10 s3details
#20ACCEPTED0.10 s3details
#21ACCEPTED0.10 s3details
#22ACCEPTED0.11 s3details
#23ACCEPTED0.15 s3details
#24ACCEPTED0.15 s3details

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[1000000];
bool seen[1000000];
int inv[1000000];
int cntIso[1000000];
int cntPieni[1000000];
int solve(int n) {
int ans = 0;
for (int i = 1; i <= n; i++) seen[i] = false;
for (int i = 1; i <= n; i++) {
if (a[i] == 1 || !seen[a[i]-1]) ans++;
seen[a[i]] = true;
}
return ans;
}
bool identity(int n) {
for (int i = 2; i <= n; i++) {
if (a[i] != a[i-1]+1) return false;
}
return true;
}
#define N (1<<18)
ll p[2*N];
void muuta(int k, int x) {
k += N;
p[k] += x;
for (k /= 2; k >= 1; k /= 2) {
p[k] = p[2*k]+p[2*k+1];
}
}
void muutavali(int a, int b, int x) {
if (a > b) return;
muuta(a,x);
muuta(b+1, -x);
}
ll summa(int a, int b) {
a += N;
b += N;
ll s = 0;
while (a <= b) {
if (a%2 == 1) s += p[a++];
if (b%2 == 0) s += p[b--];
a /= 2; b /= 2;
}
return s;
}
int apu[1000000];
void muunna(int n) {
for (int i = 0; i <= n+1; i++) apu[i] = a[n+1-i];
for (int i = 0; i <= n+1; i++) a[i] = apu[i];
for (int i = 0; i <= n+1; i++) a[i] = n + 1 - a[i];
for (int i = 0; i <= n+1; i++) inv[a[i]] = i;
/*
cout << "a : ";
for (int i = 0; i <= n+1; i++) cout << a[i] << " ";
cout << endl;
cout << "b : ";
for (int i = 0; i <= n+1; i++) cout << inv[i] << " ";
cout << endl;
*/
}
ll ratkaise(int n) {
ll ans = 0;
//cout << "summa: " << 0 << " " << summa(1, 1) << endl;
for (int i = 1; i <= n; i++) {
if (inv[a[i]-1] < inv[a[i]] && inv[a[i]] < inv[a[i]+1]) { // 7 8 9
muutavali(inv[a[i]-1]+1, inv[a[i]]-1, 1);
}
else if (inv[a[i]-1] < inv[a[i]+1] && inv[a[i]+1] < inv[a[i]]) { // 7 9 8
//muutavali(1,inv[a[i]]-1, 1);
muutavali(1, inv[a[i]+1]-1, 1);
muutavali(inv[a[i]+1]+1, inv[a[i]]-1, 1);
}
else if (inv[a[i]] < inv[a[i]+1] && inv[a[i]+1] < inv[a[i]-1]) { // 8 9 7
muutavali(1,inv[a[i]]-1, 1);
}
else if (inv[a[i]] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]+1]) { // 8 7 9
muutavali(1,inv[a[i]]-1, 1);
}
else if (inv[a[i]+1] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]]) { // 9 7 8
muutavali(1,inv[a[i]+1]-1, 1);
muutavali(inv[a[i]-1]+1, inv[a[i]]-1, 1);
}
else if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) { // 9 8 7
muutavali(1,inv[a[i]]-1, 1);
}
/*
cout << "summa: " << a[i] << endl;
for (int i = 1; i <= n; i++) {
cout << summa(1,i) << " ";
}
cout << endl;
*/
}
//cout << endl;
for (int i = 1; i <= n; i++) {
//cout << "summa: " << i << " " << a[i] << endl;
if (inv[a[i]-1] < inv[a[i]] && inv[a[i]] < inv[a[i]+1]) { // 7 8 9
muutavali(inv[a[i]-1]+1, inv[a[i]]-1, -1);
}
else if (inv[a[i]-1] < inv[a[i]+1] && inv[a[i]+1] < inv[a[i]]) { // 7 9 8
//muutavali(1,inv[a[i]]-1, 1);
muutavali(1, inv[a[i]+1]-1, -1);
muutavali(inv[a[i]+1]+1, inv[a[i]]-1, -1);
}
else if (inv[a[i]] < inv[a[i]+1] && inv[a[i]+1] < inv[a[i]-1]) { // 8 9 7
muutavali(1,inv[a[i]]-1, -1);
}
else if (inv[a[i]] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]+1]) { // 8 7 9
muutavali(1,inv[a[i]]-1, -1);
}
else if (inv[a[i]+1] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]]) { // 9 7 8
muutavali(1,inv[a[i]+1]-1, -1);
muutavali(inv[a[i]-1]+1, inv[a[i]]-1, -1);
}
else if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) { // 9 8 7
muutavali(1,inv[a[i]]-1, -1);
}
cntIso[a[i]] = summa(1, inv[a[i]+1]);
cntPieni[a[i]] = summa(1,inv[a[i]-1]);
/*
for (int i = 1; i <= n; i++) {
cout << summa(1,i) << " ";
}
cout << endl;
*/
}
for (int i = 1; i <= n; i++) {
//cout << i << " " << ans << endl;
if (inv[a[i]] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]+1]) {
ans += cntIso[a[i]-1] - cntPieni[a[i]+1];
/*
cout << "here" << endl;
cout << cntIso[a[i]-1] << " " << a[i]-1 << endl;
cout << cntPieni[a[i]+1] << " " << a[i] + 1 << endl;
cout << endl;
*/
}
if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) {
ans += cntIso[a[i]-1];
}
}
return ans;
}
int main() {
int n;
cin >> n;
for (int i = 1; i <= n; i++) cin >> a[i];
if (identity(n)) {
cout << 2 << " " << n-1 << "\n";
return 0;
}
int startval = solve(n);
a[n+1] = n+1;
for (int i = 0; i <= n+1; i++) inv[a[i]] = i;
for (int i = 1; i <= n; i++) {
if (inv[i-1] < inv[i+1] && inv[i+1] < inv[i]) {
muutavali(inv[i-1]+1, inv[i+1]-1, 1);
}
if (inv[i+1] < inv[i] && inv[i] < inv[i-1]) {
muutavali(1,inv[i+1]-1, 1);
}
}
for (int i = 1; i <= n; i++) {
if (inv[a[i]-1] < inv[a[i]+1] && inv[a[i]+1] < inv[a[i]]) { // 7 9 8
muutavali(inv[a[i]-1]+1, inv[a[i]+1]-1, -1);
}
if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) { // 9 8 7
muutavali(1,inv[a[i]+1]-1, -1);
}
cntIso[a[i]] = summa(1, inv[a[i]+1]);
cntPieni[a[i]] = summa(1,inv[a[i]-1]);
}
ll ans = 0;
for (int i = 1; i <= n; i++) {
if (inv[a[i]] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]+1]) {
ans += cntIso[a[i]-1] - cntPieni[a[i]+1];
}
if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) {
ans += cntIso[a[i]-1];
}
}
/*
for (int i = 1; i <= n; i++) {
cout << i << " " << cntIso[i] << " " << cntPieni[i] << endl;
}
*/
if (ans > 0) {
cout << startval - 2 << " " << ans << "\n";
return 0;
}
/*
for (int i = 1; i <= n; i++) {
cntIso[i] = 0;
cntPieni[i] = 0;
}
*/
//for (int i = 0; i < 2*N; i++) p[i] = 0;
for (int i = 1; i < n; i++) {
if (inv[i+1] < inv[i] && (inv[i+2] < inv[i+1] || inv[i+2] > inv[i]) && (inv[i-1] < inv[i+1] || inv[i-1] > inv[i])) ans++;
}
//cout << ans << endl;
ans += ratkaise(n);
muunna(n);
//for (int i = 0; i <= n+1; i++) cout << a[i] << " ";
//cout << endl;
//cout << ans << endl;
ans += ratkaise(n);
muunna(n);
for (int i = 1; i < n; i++) {
if (inv[i] < inv[i+1] && (inv[i-1] > inv[i] && inv[i-1] < inv[i+1]) && (inv[i+2] > inv[i] && inv[i+2] < inv[i+1])) ans--;
}
if (ans > 0) {
cout << startval - 1 << " " << ans << "\n";
return 0;
}
ans = (ll)n*(n-1)/2;
//cout << ans << endl;
for (int i = 1; i < n; i++) {
if (inv[i] < inv[i+1]) {
ans -= n - (inv[i+1] - inv[i]);
} else {
ans += n - (inv[i] - inv[i+1]);
}
}
for (int i = 0; i <= n+1; i++) apu[i] = a[n+1-i];
for (int i = 0; i <= n+1; i++) a[i] = apu[i];
for (int i = 0; i <= n+1; i++) inv[a[i]] = i;
ll ans2 = 0;
for (int i = 1; i <= n; i++) {
if (inv[i-1] < inv[i+1] && inv[i+1] < inv[i]) {
muutavali(inv[i-1]+1, inv[i+1]-1, 1);
}
if (inv[i+1] < inv[i] && inv[i] < inv[i-1]) {
muutavali(1,inv[i+1]-1, 1);
}
}
for (int i = 1; i <= n; i++) {
if (inv[a[i]-1] < inv[a[i]+1] && inv[a[i]+1] < inv[a[i]]) { // 7 9 8
muutavali(inv[a[i]-1]+1, inv[a[i]+1]-1, -1);
}
if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) { // 9 8 7
muutavali(1,inv[a[i]+1]-1, -1);
}
cntIso[a[i]] = summa(1, inv[a[i]+1]);
cntPieni[a[i]] = summa(1,inv[a[i]-1]);
}
for (int i = 1; i <= n; i++) {
if (inv[a[i]] < inv[a[i]-1] && inv[a[i]-1] < inv[a[i]+1]) {
ans2 += cntIso[a[i]-1] - cntPieni[a[i]+1];
}
if (inv[a[i]+1] < inv[a[i]] && inv[a[i]] < inv[a[i]-1]) {
ans2 += cntIso[a[i]-1];
}
}
cout << startval << " " << ans+ans2 << "\n";
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
2 99

user output
2 99

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
100 99 98 97 96 95 94 93 92 91...

correct output
98 4851

user output
98 4851

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
1 2 88 4 5 6 7 8 9 10 11 12 13...

correct output
16 5

user output
16 5

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
51 48 74 70 45 71 24 88 55 99 ...

correct output
49 131

user output
49 131

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
45 67 29 62 70 77 41 74 52 95 ...

correct output
52 268

user output
52 268

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
47 98 2 75 6 21 84 8 4 89 27 9...

correct output
48 149

user output
48 149

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
73 68 17 94 71 63 61 13 58 10 ...

correct output
47 116

user output
47 116

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
17 16 45 94 6 1 36 81 31 13 51...

correct output
45 95

user output
45 95

Test 9

Group: 2, 3

Verdict: ACCEPTED

input
5000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
2 4999

user output
2 4999

Test 10

Group: 2, 3

Verdict: ACCEPTED

input
5000
5000 4999 4998 4997 4996 4995 ...

correct output
4998 12492501

user output
4998 12492501

Test 11

Group: 2, 3

Verdict: ACCEPTED

input
5000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
19 10

user output
19 10

Test 12

Group: 2, 3

Verdict: ACCEPTED

input
5000
1 2 3 4 5 6 264 8 9 10 11 12 1...

correct output
190 96

user output
190 96

Test 13

Group: 2, 3

Verdict: ACCEPTED

input
5000
1 2 3 4 5 6 7 8 9 2400 11 12 1...

correct output
1378 27938

user output
1378 27938

Test 14

Group: 2, 3

Verdict: ACCEPTED

input
5000
4012 2 4820 4208 1868 1728 362...

correct output
2511 436307

user output
2511 436307

Test 15

Group: 2, 3

Verdict: ACCEPTED

input
5000
3877 3972 1112 3669 1959 4640 ...

correct output
2497 417065

user output
2497 417065

Test 16

Group: 2, 3

Verdict: ACCEPTED

input
5000
2774 998 4525 2884 487 1995 41...

correct output
2518 426448

user output
2518 426448

Test 17

Group: 3

Verdict: ACCEPTED

input
200000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
2 199999

user output
2 199999

Test 18

Group: 3

Verdict: ACCEPTED

input
200000
200000 199999 199998 199997 19...

correct output
199998 19999700001

user output
199998 19999700001

Test 19

Group: 3

Verdict: ACCEPTED

input
200000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
19 10

user output
19 10

Test 20

Group: 3

Verdict: ACCEPTED

input
200000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
199 100

user output
199 100

Test 21

Group: 3

Verdict: ACCEPTED

input
200000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
1979 1030

user output
1979 1030

Test 22

Group: 3

Verdict: ACCEPTED

input
200000
1 2 184153 4 5 6 7 8 164545 10...

correct output
18081 477187

user output
18081 477187

Test 23

Group: 3

Verdict: ACCEPTED

input
200000
151013 68675 119105 58292 3335...

correct output
86328 318722426

user output
86328 318722426

Test 24

Group: 3

Verdict: ACCEPTED

input
200000
11562 33356 106752 170825 2723...

correct output
99873 663048119

user output
99873 663048119