树状数组 + 离散化

/**********
Author Smile
学到了:
离散化:
1、用一个辅助的数组把你要离散的所有数据存下来。

2、排序,排序是为了后面的二分。

3、去重,因为我们要保证相同的元素离散化后数字相同。

4、索引,再用二分把离散化后的数字放回原数组。
本题的离散化是用于求在该速度之前的符和条件的点数
求出点数是为了用树状数组的思想减去前缀和
差值即为贡献
在树状数组中的数组,一个是记录点数,一个是记录前缀和
lowerbit 的本质最多才不超过30次忽略不计
时间复杂度 n * log n ;
 
https://vjudge.csgrandeur.cn/contest/482090#problem/F
**********/
#include <iostream>
#include <algorithm>
#define int long long
using namespace std;
const int N = 2e5 + 10;
int n;
int sum[N], cnt[N], vtemp[N];
struct node
{
	int x, v;
} a[N];
int ans = 0;

bool comp(node a, node b)
{
	return a.x < b.x;
}

int lowerbit(int x)
{
	return x & (-x);
}

int query(int x, int w[])
{
	int ans = 0;
	while (x > 0)
	{
		ans += w[x];
		x -= lowerbit(x);
	}
	return ans;
}

void add(int x, int w[], int value)
{
	while (x <= N)
	{
		w[x] += value;
		x += lowerbit(x);
	}
}

signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	cin >> n;
	for (int i = 1; i <= n; ++i)
		cin >> a[i].x;
	for (int i = 1; i <= n; ++i)
	{
		cin >> a[i].v;
		vtemp[i] = a[i].v;
	}		
	sort(vtemp + 1, vtemp + 1 + n);
	int len = unique(vtemp + 1, vtemp + 1 + n) - vtemp - 1;
	sort(a + 1, a + 1 + n, comp);
	for (int i = 1; i <= n; ++i)
	{
		int pos = lower_bound(vtemp + 1, vtemp + 1 + len, a[i].v) - vtemp;
		ans += query(pos, cnt) * a[i].x - query(pos, sum);
		add(pos, cnt, 1);
		add(pos, sum, a[i].x);
	}
	cout << ans << endl;
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值