http://acm.hdu.edu.cn/showproblem.php?pid=2492
題目大意:給出n 個數的序列,枚舉一個數,然後查找該數 左邊大於它的數的個數 a 和小於他的數的個數 b ,查找右邊大於他的數的個數 c 和小於他的數的個數 d 。
最後,res = a * d + b * c ; 其實就是一個查找逆序數的過程。
可以用樹狀數組,和線段樹解決。
這裏要注意一點,如果給出的數據比較大的情況,數組無法開到這麼大,可以用離散化的思想,把數組轉化成一個較小的數組。
樹狀數組:
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
# define N 400000
int inf = 100000000;
int maxx[N],b[N],c[N];
int lmin[N],lmax[N],rmin[N],rmax[N];
int n;
struct A
{
int v,index;
}a[400000];
int cmp(A p1, A p2)
{
return p1.v < p2.v;
}
int lowbit(int x)
{
return x & (-x);
}
void update(int x, int y)
{
while(x <= n)
{
c[x] += y;
x += lowbit(x);
}
}
int getsum(int x)
{
int ans = 0;
while(x > 0)
{
ans += c[x];
x -= lowbit(x);
}
return ans;
}
int main()
{
int T,i;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
memset(c,0,sizeof(c));
for(i =1 ; i <= n; i++)//離散化
{
scanf("%d",&a[i].v);
a[i].index = i;
}
sort(a+1,a+n+1,cmp);
for(i = 1; i <= n; i++)
b[a[i].index] = i;//離散化
for(i = 1; i <= n; i++)
{
lmin[i] = getsum(b[i]);
lmax[i] = i - lmin[i] - 1;
update(b[i],1);
}
memset(c,0,sizeof(c));
for(i = n; i > 0; i--)
{
rmin[i] = getsum(b[i]);
rmax[i] = n-i - rmin[i];
update(b[i],1);
}
long long res = 0;//第一次 WA 就是因爲這個地方,定義的時候用了int
for(i = 1; i <= n; i++)
res += lmin[i] * rmax[i] + lmax[i] * rmin[i];
printf("%I64d\n",res);
}
return 0;
}