《算法竞赛进阶指南》平衡树

最新推荐文章于 2024-01-23 10:56:40 发布

原创

最新推荐文章于 2024-01-23 10:56:40 发布 · 547 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #图论

本文探讨了如何通过改进数据结构（如Treap和Set）以及优化算法（插入和查找操作）来提高营业额统计的效率，避免平衡二叉树的超时问题，重点展示了使用Treap进行快速前后元素查找的方法。

营业额统计

不是平衡二叉树，超时
指针

#include <bits/stdc++.h>
using namespace std;

const int INF = 1e7;
int n, x, res;
struct node {
   
   
	int val;
	node *l, *r;
	node (int x) {
   
   
		this->val = x;
		l = r = NULL;
	}
};
node *root;

void insert(node* &u, int x) {
   
   
	if (u == NULL) {
   
   
		u = new node(x);
	}
	else if (x > u->val) insert(u->r, x);
	else if (x < u->val) insert(u->l, x);
	//相等的数字不用添加
}

int get_prev(node* &u, int x) {
   
   
	if (u == NULL) return -INF;
	else if (x == u->val) return x;
	else if (x < u->val) return get_prev(u->l, x);
	else return max(u->val, get_prev(u->r, x));
}

int get_post(node* &u, int x) {
   
   
	if (u == NULL) return INF;
	else if (x == u->val) return x;
	else if (x > u->val) return get_post(u->r, x);
	else return min(u->val, get_post(u->l, x));
}

int main() {
   
   
	scanf("%d", &n);
	for (int i = 1; i <= n; i ++) {
   
   
		scanf("%d", &x);
		
		if (i == 1) res += x;
		else {
   
   
			int t1 = get_post(root, x);
			int t2 = get_prev(root, x);
			
			res += min(t1 - x, x - t2);
		}
		insert(root, x);
	}
	printf("%d", res);

	return 0;
}

不是平衡二叉树，超时
数组

#include <bits/stdc++.h>
using namespace std;

const int N =

最低0.47元/天解锁文章