acwing算法提高之数据结构--平衡树Treap

2024-05-02 06:44

本文主要是介绍acwing算法提高之数据结构--平衡树Treap,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

目录

  • 1 介绍
  • 2 训练

1 介绍

本博客用来记录使用平衡树求解的题目。

插入、删除、查询操作的时间复杂度都是O(logN)

动态维护一个有序序列。

2 训练

题目1:253普通平衡树

C++代码如下,

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>using namespace std;const int N = 100010, INF = 1e8;int n;
struct Node {int l, r;int key, val;int cnt, size;
}tr[N];int root, idx;void pushup(int p) {tr[p].size = tr[tr[p].l].size + tr[tr[p].r].size + tr[p].cnt;
}int get_node(int key) {tr[++idx].key = key;tr[idx].val = rand();tr[idx].cnt = tr[idx].size = 1;return idx;
}void zig(int &p) {int q = tr[p].l;tr[p].l = tr[q].r, tr[q].r = p, p = q;pushup(tr[p].r), pushup(p);
}void zag(int &p) {int q = tr[p].r;tr[p].r = tr[q].l, tr[q].l = p, p = q;pushup(tr[p].l), pushup(p);
}void build() {get_node(-INF), get_node(INF);root = 1, tr[1].r = 2;pushup(root);if (tr[1].val < tr[2].val) zag(root);
}void insert(int &p, int key) {if (!p) p = get_node(key);else if (tr[p].key == key) tr[p].cnt++;else if (tr[p].key > key) {insert(tr[p].l, key);if (tr[tr[p].l].val > tr[p].val) zig(p);} else {insert(tr[p].r, key);if (tr[tr[p].r].val > tr[p].val) zag(p);}pushup(p);
}void remove(int &p, int key) {if (!p) return;if (tr[p].key == key) {if (tr[p].cnt > 1) tr[p].cnt --;else if (tr[p].l || tr[p].r) {if (!tr[p].r || tr[tr[p].l].val > tr[tr[p].r].val) {zig(p);remove(tr[p].r, key);} else {zag(p);remove(tr[p].l, key);}} else {p = 0;}} else if (tr[p].key > key) remove(tr[p].l, key);else remove(tr[p].r, key);pushup(p);
}int get_rank_by_key(int p, int key) {if (!p) return 0;if (tr[p].key == key) return tr[tr[p].l].size + 1;if (tr[p].key > key) return get_rank_by_key(tr[p].l, key);return tr[tr[p].l].size + tr[p].cnt + get_rank_by_key(tr[p].r, key);
}int get_key_by_rank(int p, int rank) {if (!p) return INF;if (tr[tr[p].l].size >= rank) return get_key_by_rank(tr[p].l, rank);if (tr[tr[p].l].size + tr[p].cnt >= rank) return tr[p].key;return get_key_by_rank(tr[p].r, rank - tr[tr[p].l].size - tr[p].cnt);
}int get_prev(int p, int key) {if (!p) return -INF;if (tr[p].key >= key) return get_prev(tr[p].l, key);return max(tr[p].key, get_prev(tr[p].r, key));
}int get_next(int p, int key) {if (!p) return INF;if (tr[p].key <= key) return get_next(tr[p].r, key);return min(tr[p].key, get_next(tr[p].l, key));
}int main() {build();scanf("%d", &n);while (n--) {int opt, x;scanf("%d%d", &opt, &x);if (opt == 1) insert(root, x);else if (opt == 2) remove(root, x);else if (opt == 3) printf("%d\n", get_rank_by_key(root, x) - 1);else if (opt == 4) printf("%d\n", get_key_by_rank(root, x + 1));else if (opt == 5) printf("%d\n", get_prev(root, x));else printf("%d\n", get_next(root, x));        }return 0;
}

使用C++的multiset,超时了,通过了 6/11个数据

#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>using namespace std;int main() {multiset<int> s;int m;cin >> m;while (m--) {int op, x;cin >> op >> x;if (op == 1) {s.insert(x);} else if (op == 2) {s.extract(x);} else if (op == 3) {int idx = distance(s.begin(), s.lower_bound(x));cout << 1 + idx << endl;} else if (op == 4) {x -= 1; //下标从0开始auto it = s.begin();advance(it, x);cout << *it << endl;} else if (op == 5) {auto it = s.lower_bound(x);it--; //保证一定有解cout << *it << endl;} else if (op == 6) {auto it = s.upper_bound(x);cout << *it << endl;}}return 0;
}

使用python3的库sortedcontainers,发现acwing报错没有该模块,

在这里插入图片描述

from sortedcontainers import SortedList
import bisectsl = SortedList()m = int(input())
for i in range(m):line = intput()ls_line = line.split()ls_line = [int(x) for x in ls_line]op = ls_line[0]x = ls_line[1]if op == 1:sl.add(x)elif op == 2:sl.remove(x)elif op == 3:i = bisect.bisect_left(sl, x)print(i + 1)elif op == 4:print(sl[x-1])elif op == 5:it = bisect.bisect_left(ls, x)it -= 1print(sl[it])elif op == 6:it = biset.bisect_right(ls, x)print(sl[it])

题目2:265营业额统计

使用C++的multiset

#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>
#include <climits>using namespace std;int main() {int n;cin >> n;long long res = 0;multiset<int> s;for (int i = 0; i < n; ++i) {int t;cin >> t;if (i == 0) {res += t;s.insert(t); //插入数tcontinue;}int ans = 0x3f3f3f3f;//求大于等于t的迭代器auto iter1 = s.lower_bound(t);if (iter1 != s.end()) {int x = *iter1;ans = min(ans, abs(x - t));}if (iter1 != s.begin()) {auto iter2 = iter1;iter2--;int x = *iter2;ans = min(ans, abs(x - t));}res += ans;s.insert(t); //插入数t}cout << res << endl;return 0;
}

C++代码如下,

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>using namespace std;typedef long long LL;const int N = 33010, INF = 1e7;int n;
struct Node {int l, r;int key, val;
}tr[N];int root, idx;int get_node(int key) {tr[++idx].key = key;tr[idx].val = rand();return idx;
}void build() {get_node(-INF), get_node(INF);root = 1, tr[1].r = 2;
}void zig(int &p) {int q = tr[p].l;tr[p].l = tr[q].r, tr[q].r = p, p = q;
}void zag(int &p) {int q = tr[p].r;tr[p].r = tr[q].l, tr[q].l = p, p = q;
}void insert(int &p, int key) {if (!p) p = get_node(key);else if (tr[p].key == key) return;else if (tr[p].key > key) {insert(tr[p].l, key);if (tr[tr[p].l].val > tr[p].val) zig(p);} else {insert(tr[p].r, key);if (tr[tr[p].r].val > tr[p].val) zag(p);}
}int get_prev(int p, int key) {if (!p) return -INF;if (tr[p].key > key) return get_prev(tr[p].l, key);return max(tr[p].key, get_prev(tr[p].r, key));
}int get_next(int p, int key) {if (!p) return INF;if (tr[p].key < key) return get_next(tr[p].r, key);return min(tr[p].key, get_next(tr[p].l, key));
}int main() {build();scanf("%d", &n);LL res = 0;for (int i = 1; i <= n; ++i) {int x;scanf("%d", &x);if (i == 1) res += x;else res += min(x - get_prev(root, x), get_next(root, x) - x);insert(root, x);}printf("%lld\n", res);return 0;
}

这篇关于acwing算法提高之数据结构--平衡树Treap的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/953611

相关文章

不懂推荐算法也能设计推荐系统

本文以商业化应用推荐为例,告诉我们不懂推荐算法的产品,也能从产品侧出发, 设计出一款不错的推荐系统。 相信很多新手产品,看到算法二字,多是懵圈的。 什么排序算法、最短路径等都是相对传统的算法(注:传统是指科班出身的产品都会接触过)。但对于推荐算法,多数产品对着网上搜到的资源,都会无从下手。特别当某些推荐算法 和 “AI”扯上关系后,更是加大了理解的难度。 但,不了解推荐算法,就无法做推荐系

康拓展开(hash算法中会用到)

康拓展开是一个全排列到一个自然数的双射(也就是某个全排列与某个自然数一一对应) 公式: X=a[n]*(n-1)!+a[n-1]*(n-2)!+...+a[i]*(i-1)!+...+a[1]*0! 其中,a[i]为整数,并且0<=a[i]<i,1<=i<=n。(a[i]在不同应用中的含义不同); 典型应用: 计算当前排列在所有由小到大全排列中的顺序,也就是说求当前排列是第

csu 1446 Problem J Modified LCS (扩展欧几里得算法的简单应用)

这是一道扩展欧几里得算法的简单应用题,这题是在湖南多校训练赛中队友ac的一道题,在比赛之后请教了队友,然后自己把它a掉 这也是自己独自做扩展欧几里得算法的题目 题意:把题意转变下就变成了:求d1*x - d2*y = f2 - f1的解,很明显用exgcd来解 下面介绍一下exgcd的一些知识点:求ax + by = c的解 一、首先求ax + by = gcd(a,b)的解 这个

综合安防管理平台LntonAIServer视频监控汇聚抖动检测算法优势

LntonAIServer视频质量诊断功能中的抖动检测是一个专门针对视频稳定性进行分析的功能。抖动通常是指视频帧之间的不必要运动,这种运动可能是由于摄像机的移动、传输中的错误或编解码问题导致的。抖动检测对于确保视频内容的平滑性和观看体验至关重要。 优势 1. 提高图像质量 - 清晰度提升:减少抖动,提高图像的清晰度和细节表现力,使得监控画面更加真实可信。 - 细节增强:在低光条件下,抖

【数据结构】——原来排序算法搞懂这些就行,轻松拿捏

前言:快速排序的实现最重要的是找基准值,下面让我们来了解如何实现找基准值 基准值的注释:在快排的过程中,每一次我们要取一个元素作为枢纽值,以这个数字来将序列划分为两部分。 在此我们采用三数取中法,也就是取左端、中间、右端三个数,然后进行排序,将中间数作为枢纽值。 快速排序实现主框架: //快速排序 void QuickSort(int* arr, int left, int rig

poj 3974 and hdu 3068 最长回文串的O(n)解法(Manacher算法)

求一段字符串中的最长回文串。 因为数据量比较大,用原来的O(n^2)会爆。 小白上的O(n^2)解法代码:TLE啦~ #include<stdio.h>#include<string.h>const int Maxn = 1000000;char s[Maxn];int main(){char e[] = {"END"};while(scanf("%s", s) != EO

秋招最新大模型算法面试,熬夜都要肝完它

💥大家在面试大模型LLM这个板块的时候,不知道面试完会不会复盘、总结,做笔记的习惯,这份大模型算法岗面试八股笔记也帮助不少人拿到过offer ✨对于面试大模型算法工程师会有一定的帮助,都附有完整答案,熬夜也要看完,祝大家一臂之力 这份《大模型算法工程师面试题》已经上传CSDN,还有完整版的大模型 AI 学习资料,朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费

6.1.数据结构-c/c++堆详解下篇(堆排序,TopK问题)

上篇:6.1.数据结构-c/c++模拟实现堆上篇(向下,上调整算法,建堆,增删数据)-CSDN博客 本章重点 1.使用堆来完成堆排序 2.使用堆解决TopK问题 目录 一.堆排序 1.1 思路 1.2 代码 1.3 简单测试 二.TopK问题 2.1 思路(求最小): 2.2 C语言代码(手写堆) 2.3 C++代码(使用优先级队列 priority_queue)

dp算法练习题【8】

不同二叉搜索树 96. 不同的二叉搜索树 给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。 示例 1: 输入:n = 3输出:5 示例 2: 输入:n = 1输出:1 class Solution {public int numTrees(int n) {int[] dp = new int

Codeforces Round #240 (Div. 2) E分治算法探究1

Codeforces Round #240 (Div. 2) E  http://codeforces.com/contest/415/problem/E 2^n个数,每次操作将其分成2^q份,对于每一份内部的数进行翻转(逆序),每次操作完后输出操作后新序列的逆序对数。 图一:  划分子问题。 图二: 分而治之,=>  合并 。 图三: 回溯: