Codeforces 459D 差异对数

题目描述
BSNY在解决一个难题，这个难题是这样的。给你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)%$。
样例输入
7
1 2 1 1 2 2 1
样例输出
8
提示
【样例说明】
其他样例：
输入：
3
1 1 1
输出：
1
输入：
52
1 2 3 4 5
输出：
0
【数据规模和约定】
1<=n<=10^6 1<=ai<=10^9

Solution

题目有点绕嘴，但说白了就是要你统计有几对(i,j)，1-i的a[i]个数大于j-n的a[j]个数。经过思考，我们发现其实可以用两个数组来记录这个f的值，f1[i]表示1-i中a[i]的个数，f2[i]表示i-n中a[i]的个数，这样，就可以直接拿来用了。
可是，怎么构造出这两个数组呢？在这种无序的情况下，有序总能让我们的思路更加清晰。事实确实如此，我们可以对数值大小进行排序，然后按坐标构造f1和f2。这样第一个样例就能得到f1和f2如下：
f1：1 1 2 3 2 3 4
f2：4 3 3 2 2 1 1
然后接下来的问题，是不是就是锁定一个f1的i，然后在f2的i+1-n中统计比它小的有几个？那怎么统计能快一点呢？最好能把比它小的成片累加。因此，我们可以根据数组里的值构造一个区间和的线段树，因为n最大10^6，所以线段树是开得下的。我们只要倒着枚举f1的i，然后累加线段树中1-（f1[i]-1）的和，然后再把当前的f2[i]添加进去。（想想为什么要倒着枚举i），这样，就可以AC了。
离散化+线段树

#include<cstdio>
#include<iostream>
#include<algorithm>
#define ll long long
#define p1 id<<1
#define p2 id<<1^1
using namespace std;
int n;
int tree[4500000];
struct ty
{
    int a,b;
}p[1000005];
int f1[1000005],f2[1000005];
long long ans;
bool cmp(ty x,ty y)
{
    if(x.a!=y.a) return x.a<y.a;
    return x.b<y.b;
}
int query(int id,int l,int r,int x,int y)
{
    if(x<=l&&r<=y) return tree[id];
    if(l==r) return 0;
    int mid=(l+r)>>1;
    int s=0;
    if(y<=mid) s=query(p1,l,mid,x,y);
    else
    if(x>mid) s=query(p2,mid+1,r,x,y);
    else s=query(p1,l,mid,x,mid)+query(p2,mid+1,r,mid+1,y);
    return s;
}
void update(int id,int l,int r,int x)
{
    if(l==r&&l==x)
    {
        tree[id]++;
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid) update(p1,l,mid,x); else update(p2,mid+1,r,x);
    tree[id]=tree[p1]+tree[p2];
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&p[i].a);
        p[i].b=i;
    }
    sort(p+1,p+n+1,cmp);
    f1[p[1].b]=1;
    int k=1;
    for(int i=2;i<=n;i++)
    {
        if(p[i].a!=p[i-1].a) k=1; else k++;
        f1[p[i].b]=k;
    }
    f2[p[n].b]=1;
    k=1;
    for(int i=n-1;i>=1;i--)
    {
        if(p[i].a!=p[i+1].a) k=1;else k++;
        f2[p[i].b]=k;
    }
    for(int i=n-1;i>=1;i--)
    {
        update(1,1,n,f2[i+1]);
        ans=ans+query(1,1,n,1,f1[i]-1);
    }
    cout<<ans<<endl;
    return 0;
}