POJ 1990 BIT

最新推荐文章于 2020-10-01 23:12:51 发布

SHOHOKUKU

最新推荐文章于 2020-10-01 23:12:51 发布

阅读量93

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/neweryyy/article/details/105300817

版权

数据结构专栏收录该内容

203 篇文章 1 订阅

订阅专栏

题意

传送门 POJ 1990

题解

音量乘积的求和

$\sum (max(v_{i},v_{j})\times |x_{i}-x_{j}|)$

基本思路是按照音量值升序排序，对于每一头牛，乘以音量值比它小的牛到它的距离和，复杂度 $O(n^2)$

$\sum (v_{i}\times |x_{i}-x_{j}|),v_{i}>v_{j}$

需要处理掉绝对值

$\sum_{v_{i}>v_{j},x_{i}>x_{j}} (v_{i}\times (x_{i}-x_{j}))+\sum_{v_{i}>v_{j},x_{i}<x_{j}} (v_{i}\times (x_{j}-x_{i}))$

每次处理的牛 $i$ 与已经处理的牛 $j$ 满足 $v_{i}>v_{j}$ ，只要知道已经处理的牛的坐标和头数即可求得音量乘积。这里用两个二维数组以坐标值为索引，维护牛的头数以及坐标和，总复杂度 $O (n l o g n)$ 。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define min(a,b)    (((a) < (b)) ? (a) : (b))
#define max(a,b)    (((a) > (b)) ? (a) : (b))
#define abs(x)    ((x) < 0 ? -(x) : (x))
#define INF 0x3f3f3f3f
#define delta 0.85
#define eps 1e-5
#define PI 3.14159265358979323846
#define MAX_N 20005
using namespace std;
typedef long long LL;
typedef pair<int, int> P;
int N;
P S[MAX_N];
int bit1[MAX_N + 1], bit2[MAX_N + 1]; // 数量, 坐标和

int sum(int *b, int i){
	int s = 0;
	while(i > 0){
		s += b[i];
		i -= i & -i;
	}
	return s;
}

void add(int *b, int i, int x){
	while(i <= MAX_N){
		b[i] += x;
		i += i & -i;
	}
}

int main(){
	while(~scanf("%d", &N) && N){
		memset(bit1, 0, sizeof(bit1));
		memset(bit2, 0, sizeof(bit2));
		for(int i = 0; i < N; i++) scanf("%d%d", &S[i].first, &S[i].second);
		sort(S, S + N);
		LL res = 0;
		for(int i = 0; i < N; i++){
			LL v = S[i].first, x = S[i].second;
			// 处理 vj < vi, xj < xi 的情况
			res += v * (sum(bit1, x) * x - sum(bit2, x));
			// 处理 vj < vi, xj > xi 的情况
			res += v * (sum(bit2, MAX_N) - sum(bit2, x) - (sum(bit1, MAX_N) - sum(bit1, x)) * x);
			add(bit1, x, 1);
			add(bit2, x, x);
		}
		printf("%lld\n", res);
	}
	return 0;
}