hdu 5147

本文介绍了一种使用树状数组解决特定四元组计数问题的方法。通过正序和逆序遍历数组,利用树状数组统计满足条件的元素对数量,最终计算出所有符合条件的四元组总数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5147

太久没做这种题啦,,而且本来树状数组理解的不好。。结果卡了好久。。- -

这道题目教我们求的是一个符合条件的index (a,b,c,d) 要符合a<b<c<d&&Aa<Ab,Ac<Ad。

正解是枚举c,求得c之前符合条件的a,b对数x,再求得比c大的个数d为y,那么答案就是x*y;

我们先正序求树状数组,用数组f[i]代表出现在i之前,且比A[i]小的数字的个数。

再逆序求,用数组g[i]代表出现在i之后,且比A[i]大的数字的个数。

这样,我们用sum保存f[0].+....f[i-1]的和,那么此时的答案就是sum*g[i],因为g[i]代表的意思上面也说过,也就是A[i]与其后面能够成符合条件的c,d(c<d&&Ac<Ad),

sum就是i之前(注意,不包含i  !!)符合条件的a,b的对数,这就是关键。

/*
 * ThinkingLion.cpp
 *
 *  Created on: 2014年12月22日
 *      Author: dell
 */
#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<time.h>
#include<queue>
#include<stack>
#include<iterator>
#include<math.h>
#include<stdlib.h>
#include<limits.h>
#include<map>
//#define ONLINE_JUDGE
#define eps 1e-8
#define INF 0x7fffffff
#define FOR(i,a) for((i)=0;i<(a);(i)++)
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,c)scanf("%d%d%d",&a,&b,&c)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
const double PI=acos(-1.0);
template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
template<class T> inline T Min(T a,T b){return a<b?a:b;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
using namespace std;
#define ll __int64
int n,m;
#define N  50005
int a[N];
ll c[N],f[N],g[N]; //f[i]代表出现在i之前且比a[i]小的数,g[i]代表出现在i之后且比a[i]大的数
int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int res = 0;
	while(x>0){
		res += c[x];
		x-=lowbit(x);
	}
return res;
}
void upDate(int x,int d){
	while(x<=n){
		c[x] += d;
		x += lowbit(x);
	}
}
int main(){
#ifndef ONLINE_JUDGE
	freopen("in.txt","r",stdin);
//	freopen("out.txt","w",stdout);
#endif
	int t;
	sf(t);
	while(t--){
		sf(n);
		memset(c,0,sizeof c);
		for(int i=1;i<=n;i++){
			sf(a[i]);
		}
		for(int i=1;i<=n;i++){		//
			upDate(a[i],1);
			f[i] = sum(a[i]) - 1;	//-1是减去本身,统计比a[i]小的数字的个数
		}
		memset(c,0,sizeof c);
		for(int i=n;i>=1;i--){		//reverse
			upDate(a[i],1);
			g[i] = n-i+1-sum(a[i]); //统计出现在a[i]后比a[i]大的数字的个数
		}
		ll sum = 0;
		ll ans = 0;
		for(int i=1;i<=n;i++){
			ans += sum*g[i];	//注意,这里sum的值其实是f[1]+...f[i-1]的值,
			//这样sum就是i之前符合条件的a,b的对数,g[i]就是比a[i]大的个数,即d的个数,也就是c,d的对数,sum*g[i]就是符合条件的a,b,c,d的数量
			sum += f[i];
		}
		printf("%I64d\n",ans);
	}
return 0;
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值