CodeForces - 961E(树状数组)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/chenquanwei_/article/details/79870628

链接:CodeForces - 961E

题意:给出一个序列A,下标1-N,求满足(1) x < y (2) a[x] >= y (3) a[y] >= x的数对有多少个。

题解:求数对个数,考虑到用树状数组。但只用树状数组却无法同时满足条件(2)(3),故先用vector满足条件(3),再用树状数组。

vector[min(i - 1, a[i])].push_back(i);即在满足条件(3)的最大下标处加入i。

因为大于n的数在此处产生效果等同于n,故将大于n数改成n,可避免离散化。

#include <bits/stdc++.h>
using namespace std;

const double EPS = 1e-8;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int maxn = 2e5 + 10;
int n;
int a[maxn];
int bit[maxn];
vector<int> v[maxn];

long long sum(int i)
{
    long long res = 1;
    while(i){
        res += bit[i];
        i -= i & -i;
    }
    return res;
}

void add(int i, int x)
{
    while(i <= n){
        bit[i] += x;
        i += i & -i;
    }
}

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
        a[i] = min(a[i], n);
        v[min(i-1, a[i])].push_back(i);
    }

    long long ans = 0;
    for(int i = 1; i <= n; i++){
        add(a[i], 1);
        for(vector<int>::iterator it = v[i].begin(); it != v[i].end(); it++){
            ans += sum(n) - sum(*it - 1);
        }
    }

    printf("%I64d\n", ans);

    return 0;
}

没有更多推荐了,返回首页