hdu 2492 Ping pong

囧啊。。最近很喜欢用“囧”。。。呵呵

随便找个北京赛区的水题，居然就是线段树（树状数组），囧啊。。。

这题是求以一个点为中点，前面比它小后面比它大的对数 以及 前面比它大后面比它小的对数 的和。

第一反应用线段树。一个点一个点地插入，插入的时候，计算当前节点前比它小的，根据它的rank（第几小），计算出前面比它大，比它小，后面比它大，比它小的数。

记得用long long。。。

发现很纠结啊，如果用线段树的话，得计算出来它是第几小的数，然后才能计算。

用了俩sort，800+MS。。。囧。。

后来觉得还可以用树状数组，因为树状数组我好久没用了，总是不敢用。。。

套了下，500+MS。。

纠结他们的100+MS是怎么做的。。

网上搜了个代码，他们没有用sort计算第几大，而是用了两次树状数组。纠结啊，按理来说都是N*LOGN的算法，怎么差别这么大捏。。

线段树做法：

/*囧啊。。。还是线段树*/
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#define MID(x,y) ( ( x + y ) >> 1 )
#define L(x) ( x << 1 )
#define R(x) ( x << 1 | 1 )
#define FOR(i,s,t) for(int i=(s); i<(t); i++)
#define BUG puts("here!!!")
#define STOP system("pause")
#define file_r(x) freopen(x, "r", stdin)
#define file_w(x) freopen(x, "w", stdout)

using namespace std;

const int MAX_N = 20010;
const int MAX = 100010;

struct Tnode{				// 一维线段树 
    int l,r;
  	int sum;
    int len() { return r - l;}
    int mid() { return MID(l,r);}
    bool in(int ll,int rr) { return l >= ll && r <= rr; }
    void lr(int ll,int rr){ l = ll; r = rr;}
};
Tnode node[MAX<<2];
void Updata_sum(int t)
{
	node[t].sum = node[L(t)].sum + node[R(t)].sum;
}
void Build(int t,int l,int r)
{
	node[t].lr(l,r);
	node[t].sum = 0;
	if( node[t].len() == 1 )
		return ;
	int mid = MID(l,r);
	Build(L(t),l,mid);
	Build(R(t),mid,r);
	Updata_sum(t);
}

void Updata(int t,int l,int r)
{
	if( node[t].in(l,r) )
	{
		node[t].sum = 1;
		return ;
	}
	if( node[t].len() == 1 ) return ;
	int mid = node[t].mid();
	if( l < mid ) Updata(L(t),l,r);
	if( r > mid ) Updata(R(t),l,r);
	Updata_sum(t);
}

int Query(int t,int l,int r)
{
	if( node[t].in(l,r) )	return node[t].sum;
	if( node[t].len() == 1 ) return 0;
	int mid = node[t].mid();
	int ans = 0;
	if( l < mid ) ans += Query(L(t),l,r);
	if( r > mid ) ans += Query(R(t),l,r);
	return ans;
}
struct NODE
{
	int key, id, rank;
};
NODE a[MAX_N];
bool cmp_key(NODE a, NODE b)
{
	return a.key < b.key;
}
bool cmp_id(NODE a, NODE b)
{
	return a.id < b.id;
}
long long solve(int n)
{
	sort(a, a+n, cmp_key);
	FOR(i, 0, n)
		a[i].rank = i;
	sort(a, a+n, cmp_id);
	Build(1, 0, MAX);
	long long ans = 0;
	Updata(1, a[0].key, a[0].key + 1);
	FOR(i, 1, n-1)
	{
		long long aa = Query(1, 0, a[i].key);	// i之前比a[i]小的数的个数
		int bb = i - aa;	// i之前比a[i]大的数的个数 
		int cc = a[i].rank - aa; // i之后比a[i] 小的数的个数 
		int dd = n - i - 1 - cc; // i之后比a[i] 大的数的个数 
		ans += ( aa * 1ll * dd + bb * 1ll * cc );
		Updata(1, a[i].key, a[i].key + 1);
	}
	return ans;
}

int main()
{
	int ncases, n;
	
	scanf("%d", &ncases);
	
	while( ncases-- )
	{
		scanf("%d", &n);
		FOR(i, 0, n)
		{
			scanf("%d", &a[i].key);
			a[i].id = i;
		}
		
		if( n <= 2 )
		{
			printf("0\n");
			continue;
		}
		
		long long ans = solve(n);
		
		printf("%I64d\n", ans);
	}

return 0;
}

树状数组：

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#define MID(x,y) ( ( x + y ) >> 1 )
#define L(x) ( x << 1 )
#define R(x) ( x << 1 | 1 )
#define FOR(i,s,t) for(int i=(s); i<(t); i++)
#define BUG puts("here!!!")
#define STOP system("pause")
#define file_r(x) freopen(x, "r", stdin)
#define file_w(x) freopen(x, "w", stdout)

using namespace std;

const int MAX_N = 20010;
const int MAX = 100010;

int c[MAX];
int Lowbit(int x)
{
	return x & (-x);
}

void Updata(int x,int num)// num 可能都为1，具体问题具体分析 
{
	int i;
	for(i=x; i<MAX; i+=Lowbit(i))
		c[i] += num;	// 若num 为1，c[i]++ 
}

int Getsum(int x)//sum的返回值的类型可能是long long ，根据题意判断 
{
	int sum = 0,i;
	for(i=x; i>0; i-=Lowbit(i))
		sum += c[i];
	return sum;
}

struct NODE
{
	int key, id, rank;
};
NODE a[MAX_N];
bool cmp_key(NODE a, NODE b)
{
	return a.key < b.key;
}
bool cmp_id(NODE a, NODE b)
{
	return a.id < b.id;
}
long long solve(int n)
{
	memset(c, 0, sizeof(c));
	sort(a, a+n, cmp_key);
	FOR(i, 0, n)
		a[i].rank = i;
	sort(a, a+n, cmp_id);
	long long ans = 0;
	Updata(a[0].key, 1);
	FOR(i, 1, n-1)
	{
		long long aa = Getsum(a[i].key);	// i之前比a[i]小的数的个数
		int bb = i - aa;	// i之前比a[i]大的数的个数 
		int cc = a[i].rank - aa; // i之后比a[i] 小的数的个数 
		int dd = n - i - 1 - cc; // i之后比a[i] 大的数的个数 
		ans += ( aa * 1ll * dd + bb * 1ll * cc );
		Updata(a[i].key, 1);
	}
	return ans;
}

int main()
{
	int ncases, n;
	
	scanf("%d", &ncases);
	
	while( ncases-- )
	{
		scanf("%d", &n);
		FOR(i, 0, n)
		{
			scanf("%d", &a[i].key);
			a[i].id = i;
		}
		
		if( n <= 2 )
		{
			printf("0\n");
			continue;
		}
		
		long long ans = solve(n);
		
		printf("%I64d\n", ans);
	}

return 0;
}