#include <bits/stdc++.h> using namespace std; #define fi first #define se second #define pb push_back #define all(x) (x).begin(), (x).end() #define mid (l+r)/2 #define PI acos((ld)-1) #define BOOST ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0) typedef long long ll; typedef long double ld; typedef pair<int, int> ii; typedef vector<int> vi; typedef vector<ll> vll; typedef vector<ii> vii; typedef vector<vi> vvi; int main(){ BOOST; int n; ll ans = 0, mx; cin >> n; mx = n*2 + 1; vi t(n), pos(n + 1); for(int i=0; i<n; i++){ cin >> t[i]; pos[t[i]] = i; } for(int i=1; i<=n; i++){ int a = pos[n-i/2], b = pos[n]; if(a > b) swap(a, b); if(b-a+1 > i) continue; int be = max(0, b-i+1), en = min(n-1, a+i-1); ans += (en-be+1) - i + 1; } cout << mx << " " << ans; }
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 | #include <bits/stdc++.h> using namespace std; #define fi first #define se second #define pb push_back #define all(x) (x).begin(), (x).end() #define mid (l+r)/2 #define PI acos((ld)-1) #define BOOST ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0) typedef long long ll; typedef long double ld; typedef pair<int, int> ii; typedef vector<int> vi; typedef vector<ll> vll; typedef vector<ii> vii; typedef vector<vi> vvi; int main(){ BOOST; int n; ll ans = 0, mx; cin >> n; mx = n*2 + 1; vi t(n), pos(n + 1); for(int i=0; i<n; i++){ cin >> t[i]; pos[t[i]] = i; } for(int i=1; i<=n; i++){ int a = pos[n-i/2], b = pos[n]; if(a > b) swap(a, b); if(b-a+1 > i) continue; int be = max(0, b-i+1), en = min(n-1, a+i-1); ans += (en-be+1) - i + 1; } cout << mx << " " << ans; } |