#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
int n;
int a[1000000];
int odw[1000001];
void wczytaj(){
scanf("%d", &n);
for(int i=0; i<n; i++){
scanf("%d", a+i);
odw[a[i]]=i;
}
}
long long rozw(){
long long zwrot=0;
int najwzlewej=-1;
int najmzprawej=n;
int l, p;
for(int k=0; k<n; k++){
while(2*a[najwzlewej+1]<n+k) najwzlewej++;
while(2*a[najmzprawej-1]<n+k) najmzprawej--;
l=najwzlewej+1;
p=n-najmzprawej;
l=min(l, k);
p=min(p, k);
if(k<=l+p){
zwrot+=l+p-k+1;
}
//cout << k << ": " << l << " " << p << endl;
}
return zwrot;
}
int main()
{
wczytaj();
printf("%d %lld", 2*n+1, rozw());
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | #include <iostream> #include <fstream> #include <algorithm> using namespace std; int n; int a[1000000]; int odw[1000001]; void wczytaj(){ scanf("%d", &n); for(int i=0; i<n; i++){ scanf("%d", a+i); odw[a[i]]=i; } } long long rozw(){ long long zwrot=0; int najwzlewej=-1; int najmzprawej=n; int l, p; for(int k=0; k<n; k++){ while(2*a[najwzlewej+1]<n+k) najwzlewej++; while(2*a[najmzprawej-1]<n+k) najmzprawej--; l=najwzlewej+1; p=n-najmzprawej; l=min(l, k); p=min(p, k); if(k<=l+p){ zwrot+=l+p-k+1; } //cout << k << ": " << l << " " << p << endl; } return zwrot; } int main() { wczytaj(); printf("%d %lld", 2*n+1, rozw()); return 0; } |
English