[BZOJ 1367][Baltic2004]sequence

本文介绍了一种使用左偏树实现可并堆的数据结构方法,并提供了完整的C++代码实现。通过该方法可以高效地进行元素的合并操作。

左偏树, 可并堆



#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define maxn 1000010
using namespace std;

int n;

struct Node{
	int val, l, r, dis;
	Node(){dis = l = r = val = 0;}
	bool operator<(const Node& k)const{
		return val < k.val;
	}
};

Node T[maxn];
int merge(int a, int b){
	if(a == 0 || b == 0)
		return a + b;
	if(T[a] < T[b])
		swap(a, b);
	T[a].r = merge(T[a].r, b);
	if(T[T[a].l].dis < T[T[a].r].dis)
		swap(T[a].l, T[a].r);
	T[a].dis = T[a].r ? T[T[a].r].dis + 1 : 0;
	return a;
}

int root[maxn], m, size[maxn];


int main(){
	scanf("%d", &n);
	T[0].dis = -1;
	for(int i = 1; i <= n; i ++){
		scanf("%d", &T[i].val);
		T[i].val -= i;
		root[++ m] = i;
		size[m] = 1;
		while(m > 1 && T[root[m]].val < T[root[m - 1]].val){
			int k = size[m] & size[m - 1] & 1;
			root[m - 1] = merge(root[m - 1], root[m]);
			if(k)root[m - 1] = merge(T[root[m - 1]].l, T[root[m - 1]].r);
			size[m - 1] += size[m];
			m --;
		}
	}
	
	int cur = 1;
	long long ans = 0;
	for(int i = 1; i <= m; i ++)
		for(int j = 1; j <= size[i]; j ++, cur ++)
			ans += abs(T[root[i]].val - T[cur].val);
	printf("%lld\n", ans);
	return 0;
}


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值