问题 D: 数列-训练套题T10T3

数列(sequence.pas/c/cpp)

• 问题描述

一个简单的数列问题：给定一个长度为n的数列，求这样的三个元素ai, aj, ak的个数，满足ai < aj > ak，且i < j < k。

• 输入数据

第一行是一个整数n(n <= 50000)。

第二行n个整数ai(0 <= ai <= 32767)。

• 输出数据

一个数，满足ai < aj > ak (i < j < k)的个数。

• 样例输入

5
1 2 3 4 1

• 样例输出

6

解题思路

求的是满足ai < aj > ak (i < j < k)的个数，即对数列的每一个数找左边小于a[j]的数的个数m，右边小于a[j]的个数n，则对于a[j]满足题意的的总个数为m*n，意思就是从左边m个挑一个，和右边n个挑一个和a[j]组成ai < aj > ak。因此总个数为m*n。
使用两个数组low,up分别存储每个位置上的左边小于a[i]的个数，右边小于a[i]的个数。
使用树状数组求解，在输入数据时，得到左边小于a[i]的个数。
输入完后，将数组数组初始化为0从最后一个数倒序求解右边小于a[i]的个数。----也可以输入完后遍历数组得到当前小于a[i]的个数 - low[i] 就为右边小于a[i]的个数
然后将每个位置上的low,up相乘在加起来。大坑和用long long
因为树状数组不能处理为0的数据，所以我对代码中做了一些细节处理。
另外还有一种简便的方法，将输入数据每个+1这样数据就变成了1-32768
就不需要更改updata方法，以及那些细节处理

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
using namespace std;
#define lowbit(i) ((i) & (-i))
int getSum(int x);
void updata(int x, int v);
const int maxn = 50100;
int c[maxn] = {0}, a[maxn];
int low[maxn] = {0}, up[maxn] = {0};
int main(int argc, char** argv) {

	int n, x;
	scanf("%d", &n);
	for (int i = 1 ; i <= n; i++){
		scanf("%d", &a[i]);
		updata(a[i], 1);
		if (a[i] != 0)
			low[i] = getSum(a[i] - 1) + c[0]; //  左边小于a[i]的个数 
		else
			low[i] = 0;
	}
	long long sum = 0;
	fill(c, c + maxn, 0);
	for (int i = n; i >= 1; i--) {
		updata(a[i], 1);
		if (a[i] != 0)
			up[i] = getSum(a[i] - 1) + c[0]; // 右边 小于A[I]的个数 
		else
			up[i] = 0;
	}
	for (int i = 1; i <= n; i++) {
		sum += low[i] * up[i];
	}
	printf("%lld\n", sum);
	return 0;
}
int getSum(int x) {
	int sum = 0;
	for (int i = x; i > 0; i -= lowbit(i)) {
		sum += c[i];
	}
	return sum;
}
void updata(int x, int v) {
	if (x == 0)
		c[0] += v;
	else
	for (int i = x; i <= maxn; i += lowbit(i)) {
		c[i] += v;
	}
}