#include <iostream>
using namespace std;
void fastscan(int &number)
{
bool negative = false;
register int c;
number = 0;
c = getchar();
if (c=='-')
{
negative = true;
c = getchar();
}
for (; (c>47 && c<58); c=getchar())
number = number *10 + c - 48;
if (negative)
number *= -1;
}
int main()
{
int n;
fastscan(n);
int tab[n] = {}, input;
for(int i = 0; i < n; i++)
{
fastscan(input);
tab[input - 1] = i; // mapa na pozycje w ciagu
}
int pk[n][2]; // 0 - p, 1 - k
pk[n-1][0] = pk[n-1][1] = tab[n-1];
// budowa tablicy (p - k)
if(n%2 == 0)
{
for(int i = 0; i < n/2; i++)
{
pk[n - 2 - 2 * i][0] = min(pk[n-1-2*i][0],tab[n-i-2]);
pk[n - 2 - 2 * i][1] = max(pk[n-1-2*i][1],tab[n-i-2]);
pk[n - 2 - 2 * i-1][0] = pk[n - 2 - 2 * i][0];
pk[n - 2 - 2 * i-1][1] = pk[n - 2 - 2 * i][1];
}
pk[0][0] = min(pk[1][0], tab[n/2 - 1]);
pk[0][1] = max(pk[1][1], tab[n/2 - 1]);
}
else
{
for(int i = 0; i < n/2; i++)
{
pk[n - 2 - 2 * i][0] = min(pk[n-1-2*i][0],tab[n-i-2]);
pk[n - 2 - 2 * i][1] = max(pk[n-1-2*i][1],tab[n-i-2]);
pk[n - 2 - 2 * i-1][0] = pk[n - 2 - 2 * i][0];
pk[n - 2 - 2 * i-1][1] = pk[n - 2 - 2 * i][1];
}
}
//test
/*cout << "Mapa:\t";
for(int i = 0; i < n; i++)
cout << tab[i]+1 << " ";
cout<<endl;
cout<<"PK:\t";
for(int i = 0; i < n; i++)
cout<<pk[i][0]+1 << " " <<pk[i][1]+1<<" ";
cout<<endl;*/
int zapas, l, p;
long long cnt = 1;
for(int i = n - 2; i >= 0; i--)
{
zapas = n - i - (pk[i][1] - pk[i][0] + 1);
//cout << zapas << " " << n - i << " ";
if(zapas == 0)
cnt++;
if(zapas > 0)
{
l = min(pk[i][0], zapas);
p = min(n - pk[i][1] - 1, zapas);
cnt += l + p - zapas + 1;
}
}
cout << 2*n + 1 << " " << cnt;
}
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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | #include <iostream> using namespace std; void fastscan(int &number) { bool negative = false; register int c; number = 0; c = getchar(); if (c=='-') { negative = true; c = getchar(); } for (; (c>47 && c<58); c=getchar()) number = number *10 + c - 48; if (negative) number *= -1; } int main() { int n; fastscan(n); int tab[n] = {}, input; for(int i = 0; i < n; i++) { fastscan(input); tab[input - 1] = i; // mapa na pozycje w ciagu } int pk[n][2]; // 0 - p, 1 - k pk[n-1][0] = pk[n-1][1] = tab[n-1]; // budowa tablicy (p - k) if(n%2 == 0) { for(int i = 0; i < n/2; i++) { pk[n - 2 - 2 * i][0] = min(pk[n-1-2*i][0],tab[n-i-2]); pk[n - 2 - 2 * i][1] = max(pk[n-1-2*i][1],tab[n-i-2]); pk[n - 2 - 2 * i-1][0] = pk[n - 2 - 2 * i][0]; pk[n - 2 - 2 * i-1][1] = pk[n - 2 - 2 * i][1]; } pk[0][0] = min(pk[1][0], tab[n/2 - 1]); pk[0][1] = max(pk[1][1], tab[n/2 - 1]); } else { for(int i = 0; i < n/2; i++) { pk[n - 2 - 2 * i][0] = min(pk[n-1-2*i][0],tab[n-i-2]); pk[n - 2 - 2 * i][1] = max(pk[n-1-2*i][1],tab[n-i-2]); pk[n - 2 - 2 * i-1][0] = pk[n - 2 - 2 * i][0]; pk[n - 2 - 2 * i-1][1] = pk[n - 2 - 2 * i][1]; } } //test /*cout << "Mapa:\t"; for(int i = 0; i < n; i++) cout << tab[i]+1 << " "; cout<<endl; cout<<"PK:\t"; for(int i = 0; i < n; i++) cout<<pk[i][0]+1 << " " <<pk[i][1]+1<<" "; cout<<endl;*/ int zapas, l, p; long long cnt = 1; for(int i = n - 2; i >= 0; i--) { zapas = n - i - (pk[i][1] - pk[i][0] + 1); //cout << zapas << " " << n - i << " "; if(zapas == 0) cnt++; if(zapas > 0) { l = min(pk[i][0], zapas); p = min(n - pk[i][1] - 1, zapas); cnt += l + p - zapas + 1; } } cout << 2*n + 1 << " " << cnt; } |
English