Codeforces Round 261 Div.2 D Pashmak and Parmida's problem --树状数组

题意：给出数组A，定义f(l,r,x)为A[]的下标l到r之间，等于x的元素数。i和j符合f(1,i,a[i])>f(j,n,a[j])，求有多少对这样的(i,j).

解法：分别从左到右，由右到左预处理到某个下标为止有多少个数等于该下标，用map维护。

然后树状数组更新每个f(j,n,a[j])，预处理完毕，接下来，从左往右扫过去，每次从树状数组中删去a[i]，因为i != j，i不能用作后面的统计，然后统计getsum(inc[a[i]]-1),

(inc表示从左到右)，即查询比此时的a[i]小的f(j,n,a[j])个数。

代码：

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cstdlib>

#include <cmath>

#include <algorithm>

#include <map>

#define lll __int64

using namespace std;

using namespace __gnu_cxx;

#define N 1000007

#define M 22



int c[N],a[N];

map<int,int> inc,des;

int n;



int lowbit(int x)

{

    return x & (-x);

}



void modify(int x,int val)

{

    while(x <= n)

    {

        c[x] += val;

        x += lowbit(x);

    }

}



int getsum(int x)

{

    int res = 0;

    while(x > 0)

    {

        res += c[x];

        x -= lowbit(x);

    }

    return res;

}



int main()

{

    int i;

    inc.clear();

    des.clear();

    memset(c,0,sizeof(c));

    scanf("%d",&n);

    for(i=1;i<=n;i++)

        scanf("%d",&a[i]);

    for(i=n;i>=1;i--)

    {

        des[a[i]]++;

        modify(des[a[i]],1);

    }

    lll ans = 0;

    for(i=1;i<=n;i++)

    {

        inc[a[i]]++;

        modify(des[a[i]],-1);

        des[a[i]]--;

        ans += getsum(inc[a[i]]-1);

    }

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

    return 0;

}

View Code