#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> segmTree;
void construct(int node, int left, int right)
{
if(left == right){
segmTree[node] = 1;
return;
}
int mid = (left + right)/2;
construct(2*node, left, mid);
construct(2*node+1, mid+1, right);
segmTree[node] = segmTree[2*node] + segmTree[2*node+1];
}
void update(int node, int left, int right, int poz)
{
if(left == right){
segmTree[node] = 0;
return;
}
int mid = (left + right)/2;
if(poz <= mid){
update(2*node, left, mid, poz);
}
else{
update(2*node+1, mid+1, right, poz);
}
segmTree[node] = segmTree[2*node] + segmTree[2*node+1];
}
int query(int node, int left, int right, int a, int b)
{
if(left >= a && right <= b){
return segmTree[node];
}
int mid = (left + right)/2;
int sum = 0;
if(a <= mid){
sum += query(2*node, left, mid, a, b);
}
if(b > mid){
sum += query(2*node+1, mid+1, right, a, b);
}
return sum;
}
int main()
{
int N;
cin >> N;
vector<pair<int, int>> szamok(N+1);
segmTree.assign(4*N+4, 0);
for(int i = 1; i <= N; i++){
int a;
cin >> a;
szamok[i] = {a, i};
}
sort(szamok.begin()+1, szamok.end());
construct(1, 1, N);
long long ans = 0;
for(int i = 1; i <= N; i++){
int qleft = max(1, szamok[i].second - szamok[i].first);
int qright = min(N, szamok[i].second + szamok[i].first);
ans += query(1, 1, N, qleft, qright) - 1;
update(1, 1, N, szamok[i].second);
}
cout << ans << endl;
return 0;
}
| Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Elfogadva | 1ms | 508 KiB | ||||
| 2 | Elfogadva | 111ms | 3892 KiB | ||||
| subtask2 | 11/11 | ||||||
| 3 | Elfogadva | 1ms | 508 KiB | ||||
| 4 | Elfogadva | 4ms | 316 KiB | ||||
| 5 | Elfogadva | 4ms | 316 KiB | ||||
| 6 | Elfogadva | 4ms | 316 KiB | ||||
| 7 | Elfogadva | 4ms | 344 KiB | ||||
| 8 | Elfogadva | 4ms | 316 KiB | ||||
| 9 | Elfogadva | 3ms | 316 KiB | ||||
| 10 | Elfogadva | 4ms | 544 KiB | ||||
| subtask3 | 12/12 | ||||||
| 11 | Elfogadva | 185ms | 5848 KiB | ||||
| 12 | Elfogadva | 167ms | 5684 KiB | ||||
| 13 | Elfogadva | 181ms | 5792 KiB | ||||
| 14 | Elfogadva | 120ms | 5360 KiB | ||||
| 15 | Elfogadva | 174ms | 5680 KiB | ||||
| 16 | Elfogadva | 118ms | 5428 KiB | ||||
| 17 | Elfogadva | 159ms | 5736 KiB | ||||
| subtask4 | 31/31 | ||||||
| 18 | Elfogadva | 152ms | 5428 KiB | ||||
| 19 | Elfogadva | 172ms | 5876 KiB | ||||
| 20 | Elfogadva | 116ms | 5428 KiB | ||||
| 21 | Elfogadva | 159ms | 5868 KiB | ||||
| 22 | Elfogadva | 186ms | 6220 KiB | ||||
| 23 | Elfogadva | 195ms | 6196 KiB | ||||
| 24 | Elfogadva | 172ms | 6452 KiB | ||||
| 25 | Elfogadva | 203ms | 6196 KiB | ||||
| 26 | Elfogadva | 156ms | 6452 KiB | ||||
| subtask5 | 46/46 | ||||||
| 27 | Elfogadva | 188ms | 5684 KiB | ||||
| 28 | Elfogadva | 218ms | 6080 KiB | ||||
| 29 | Elfogadva | 225ms | 5940 KiB | ||||
| 30 | Elfogadva | 128ms | 5476 KiB | ||||
| 31 | Elfogadva | 215ms | 6196 KiB | ||||
| 32 | Elfogadva | 115ms | 5428 KiB | ||||
| 33 | Elfogadva | 156ms | 6248 KiB | ||||
| 34 | Elfogadva | 153ms | 6252 KiB | ||||
| 35 | Elfogadva | 192ms | 6196 KiB | ||||
| 36 | Elfogadva | 115ms | 5352 KiB | ||||
| 37 | Elfogadva | 184ms | 6368 KiB | ||||
| 38 | Elfogadva | 203ms | 5940 KiB | ||||
| 39 | Elfogadva | 202ms | 6196 KiB | ||||
| 40 | Elfogadva | 168ms | 6452 KiB | ||||
| 41 | Elfogadva | 209ms | 6188 KiB | ||||