61E Enemy is weak

题意：有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;
}