J - Pashmak and Parmida's problem(树状数组)(逆序对)

点击打开链接

给你n个整数a1, a2, …, an。 可以定义一个函数f(l,r,x), 能范围在区间[l, r]范围内，有多少个数字等于x。公式化的说： 统计满足l<=k<=r 同时 ak=x 的数量。现在问题，在这个序列中，有多少对i, j， 满足1<=i<j<=n 同时 f(1,i,ai)>f(j,n,aj)。 

就是一个求逆序对的问题，只不过逆序对的值换成了l,r内与f[ai]相同的值的个数

分别用两个数组来存正着遍历和反着遍历到该点时前面元素与该元素相同的个数

然后就是算有多少个f(1,i,ai)>f(j,n,aj)。 

由于输入量比较大，要用树状数组来存f(j,n,aj)的值为的大小各有多少个

求和时就求a[i]-1的和 Query(a[i]-1);

更新时由于下一次i要加一,所以去掉树状数组中b[i+1]的值. add(b[i+1],-1)

#include <iostream>
#include<stdio.h>
#include<map>
#define N 1000005
#define LL long long
using namespace std;
int n;
int A[N],a[N],b[N],s[N];
void add(int x,int d)//第x值的增量为d
{
    while(x<=n)
    {
        s[x]+=d;
        x+=(x&-x);
    }
}
LL Query(int x)//求1到x的值
{
    LL sum=0;
    while(x>0)
    {
        sum+=s[x];
        x-=(x&-x);
    }
    return sum;
}
int main()
{
    cin>>n;
    map<int,int> q;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",A+i);
        q[A[i]]++;
        a[i]=q[A[i]];
    }
    q.clear();
    for(int i=n;i>=2;i--)
    {
        q[A[i]]++;
        b[i]=q[A[i]];
        add(b[i],1);
    }
    long long sum=0;
    for(int i=1;i<n;i++)
    {
        sum+=Query(a[i]-1);
        add(b[i+1],-1);
    }
    cout<<sum<<endl;
}