类似树状数组求逆序数
题意:f(1,i,ai)意思为区间1-i内与a[i]相等的个数
同理f(j,n,aj)为区间j-n内与a[j]相同的个数
在1<=i<j<=n的情况下有多少个f(1,i,ai)>f(j,n,aj);
以为a[i]很大所以统计a[i]的个数时用map存
分别统计左边开始a[i]在1-i内的个数
右边开始a[j]在j-n内的个数
然后更新dp[1]的值 每次更新区间[1-dp[1][i] ](增加);
求和的同时再次更新[1-dp[1][i] ](减小);
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <deque>
#include <set>
#include <map>
#define IN freopen ("in.txt" , "r" , stdin);
#define OUT freopen ("out.txt" , "w" , stdout);
typedef long long LL;
const int MAXN = 51111;//点数的最大值
const int MAXM = 20006;//边数的最大值
const int INF = 11521204;
const int mod=1000000007;
map<int,int >mb;
int dp[2][1000100],n;
int sum[1000100];
int lowbit(int x)
{
return x&(-x);
}
int getsum(int x)
{
int sum1=0;
while(x>0)
{
sum1+=sum[x];
x-=lowbit(x);
}
return sum1;
}
void update(int x,int add)
{
while(x<=n)
{
sum[x]+=add;
x+=lowbit(x);
}
}
int a[1000010];
int main()
{
memset(dp,0,sizeof(dp));
memset(sum,0,sizeof(sum));
scanf("%d",&n);
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
mb.clear();
for(int i=1; i<=n; i++)//左边开始
{
mb[a[i]]++;
dp[0][i]=mb[a[i]];
}
mb.clear();
for(int i=n; i>=1; i--)//右边开始
{
mb[a[i]]++;
dp[1][i]=mb[a[i]];
}
for (int i=1; i<=n; i++)//先统计
update(dp[1][i],1);
long long res=0;
for (int i=1; i<=n; i++)
{
update(dp[1][i],-1);//将这个位置的数减掉
res+=getsum(dp[0][i]-1);//统计res
}
printf("%I64d\n",res);
return 0;
}