BZOJ1588: 营业额统计 题解

本文介绍了一种使用平衡树解决特定问题的方法,并提供了详细的代码实现。通过对比set等其他数据结构,展示了平衡树的优势,特别是在处理动态数据集时的高效表现。

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

这道题其实可以用set水过去

但最近在练习平衡树,就写了一个平衡树的版本

PS: 据说这道题的数据有问题,有一个点少了一个数字,所以读入的时候要加一句 if (scanf("%d",&x)==EOF) x=0;

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <utility>
#include <cctype>
#include <algorithm>
#include <bitset>
#include <set>
#include <map>
#include <vector>
#include <queue>
#include <deque>
#include <stack>
#include <cmath>
#define LL long long
#define LB long double
#define x first
#define y second
#define Pair pair<int,int>
#define pb push_back
#define pf push_front
#define mp make_pair
#define LOWBIT(x) x & (-x)
using namespace std;

const int MOD=1e9+7;
const LL LINF=2e16;
const int INF=2e9;
const int magic=348;
const double eps=1e-10;

inline int getint()
{
    char ch;int res;bool f;
    while (!isdigit(ch=getchar()) && ch!='-') {}
    if (ch=='-') f=false,res=0; else f=true,res=ch-'0';
    while (isdigit(ch=getchar())) res=res*10+ch-'0';
    return f?res:-res;
}

struct Treap
{
	Treap *left,*right;
	int value,priority;
}*Root,*res;
typedef pair<Treap*,Treap*> pii;

int n;

inline int myabs(int x)
{
	return x>=0?x:-x;
}

Treap* Create(int key,int priority,Treap *left,Treap *right)
{
	res=new Treap;
	res->left=left;res->right=right;
	res->value=key;res->priority=priority;
	return res;
}

pii split(Treap *root,int key)
{
	if (root==NULL) return make_pair<Treap*,Treap*>(NULL,NULL);
	if (root->value<=key)
	{
		pii splitted=split(root->right,key);
		root->right=splitted.x;
		return mp(root,splitted.y);
	}
	else
	{
		pii splitted=split(root->left,key);
		root->left=splitted.y;
		return mp(splitted.x,root);
	}
}

Treap* add(Treap *root,int key,int priority)
{
	if (root==NULL) return Create(key,priority,NULL,NULL);
	if (priority<root->priority)
	{
		pii splitted=split(root,key);
		return Create(key,priority,splitted.x,splitted.y);
	}
	else
	{
		if (root->value<=key)
		{
			root->right=add(root->right,key,priority);
			return root;
		}
		else
		{
			root->left=add(root->left,key,priority);
			return root;
		}
	}
}

int query(Treap *root,int key)
{
	if (root==NULL) return INF;
	if (root->value==key) return 0;
	if (root->value<key)
	{
		int res=query(root->right,key),ret=root->value;
		if (myabs(ret-key)<res) return myabs(ret-key); else return res;
	}
	else
	{
		int res=query(root->left,key),ret=root->value;
		if (myabs(ret-key)<res) return myabs(ret-key); else return res;
	}
}


int main ()
{
	int i,x,prior;
	Root=NULL;
	scanf("%d",&n);
	int ans=0;
	for (i=1;i<=n;i++)
	{
		if (scanf("%d",&x)==EOF) x=0;
		prior=rand();
		if (i==1) ans+=x; else ans+=query(Root,x);
		Root=add(Root,x,prior);
	}
	printf("%d\n",ans);
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值