题意:有n个数a1, a2, ..., an,问i < j < k且ai > aj > ak的三角对有多少对(3 <= n <= 10^6, 1 <= ai <= 10^9)。
对于一个位置j,其左边有L[j]个比aj大的数,其右边有R[j]个比aj小的数,那么,ai可从L[j]个数中取,ak可从R[j]个数中取,此时可组成L[j] * R[j]个三角对,枚举j的位置,求和。
由于数据量不小,2层for以O(n^2)找L与R已超过题目所允许的时间,这里恰恰可用树状数组来求得R,而L可根据R推出。因为1 <= ai <= 10^9,数组开不下,但输入个数不超过10^6,所以可做一个重映射,使输入的数根据大小重映射为1到n,设重映射后的数组为f。用树状数组求出R后,因为R[i]表示第i个数的右边有多少个比f[i]小的数,而第i个数的右边共有n-i个数,所以第i个数的右边有n - i - R[i]个比f[i]大的数,而整个f有n-f[i]个比f[i]大的数,所以第i个数的左边有n - f[i] - (n - i - R[i]) = R[i] - f[i] + i个比f[i]大的数,即L[i] = R[i] -f[i] + i。
AC代码:
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAX = 1000010;
struct Node{
int v, id;
}node[MAX];
bool cmp(const Node &A, const Node &B){
return A.v < B.v;
}
int n;
int L[MAX], R[MAX];
int c[MAX], reflect[MAX];
int lowbit(int x){
return x & (-x);
}
void add(int x, int d){
while(x <= n){
c[x] += d;
x += lowbit(x);
}
}
int sum(int x){
int ret = 0;
while(x > 0){
ret += c[x];
x -= lowbit(x);
}
return ret;
}
int main(){
scanf("%d",&n);
for(int i=1; i<=n; i++){
scanf("%d",&node[i].v);
node[i].id = i;
}
sort(node+1, node+1+n, cmp);
for(int i=1; i<=n; i++)
reflect[node[i].id] = i;//离散化
memset(c, 0, sizeof(c));
for(int i=n; i>=1; i--){
R[i] = sum(reflect[i]);
add(reflect[i], 1);
}
for(int i=1; i<=n; i++)
L[i] = i - 1 - (reflect[i] - 1 - R[i]);
long long ans = 0;
for(int i=1; i<=n; i++)
ans += (long long)R[i] * L[i];//中间过程要强制类型转换
printf("%I64d\n",ans);
return 0;
}