CSC3100
其实是存一下代码
1. AVL的java实现
维护一下每个点左右子树深度差，差绝对值大于2就转，转的方式和treap, splay转的方式差不多。旋转操作可以使两端差归零变得更平衡。
虽然平衡但转的次数太多反而慢了(?)，有空回来整理下，先咕着[旺柴]
1 import java.util.*;
2 class TestMain {
3 public static void main(String[] args) {
4 AvlTree mytree = new AvlTree();
5 Scanner jdr = new Scanner(System.in);
6 int t = jdr.nextInt();
7 while(t > 0) {
8 int opt = jdr.nextInt(), x = jdr.nextInt();
9 switch (opt) {
10 case 1:
11 mytree.insert(x);
12 break;
13 case 2:
14 mytree.delete(x);
15 break;
16 case 3:
17 System.out.println(mytree.queryRank(x));
18 break;
19 case 4:
20 System.out.println(mytree.numAt(x));
21 break;
22 case 5:
23 System.out.println(mytree.queryPrev(x));
24 break;
25 case 6:
26 System.out.println(mytree.queryPost(x));
27 break;
28 }
29 t--;
30 }
31 jdr.close();
32 return ;
33 }
34 }
35
36 public class AvlTree {
37 TreeNode root;
38 static TreeNode buffer;
39 AvlTree() {
40 root = null;
41 }
42 public void insert(int x) {
43 root = insert(root, null, x);
44 return ;
45 }
46 public void delete(int x) {
47 root = delete(root, null, x);
48 return ;
49 }
50 public int queryRank(int x) {
51 return queryRank(root, x);
52 }
53 public int numAt(int index) {
54 return queryVal(root, index);
55 }
56 public int queryPrev(int x) {
57 findPrev(root, x);
58 return buffer.value;
59 }
60 public int queryPost(int x) {
61 findPost(root, x);
62 return buffer.value;
63 }
64 private static TreeNode insert(TreeNode node, TreeNode fa, int x) {
65 if(node == null) return new TreeNode(x, fa);
66 if(x < node.value) node.child[0] = insert(node.child[0], node, x);
67 else if(node.value < x) node.child[1] = insert(node.child[1], node, x);
68 else node.num ++;
69 node.push_up();
70 if(node.diff==-2 || node.diff==2) {
71 int p = (node.diff + 2) / 4, q = -node.diff/2;
72 if(node.child[p].diff == q)
73 node.child[p] = rotate(node.child[p], 1 - p);
74 node = rotate(node, p);
75 }
76 return node;
77 }
78 private static TreeNode delete(TreeNode node, TreeNode fa, int x) {
79 if(node == null) return null;
80 if(x < node.value) node.child[0] = delete(node.child[0], node, x);
81 else if(node.value < x) node.child[1] = delete(node.child[1], node, x);
82 else {
83 if(node.child[0]==null && node.child[1]==null) {
84 node.num --;
85 if(node.num == 0) return null;
86 }else {
87 int p = (node.diff > 0) ? 1 : 0;
88 node = rotate(node, p);
89 node.child[1-p] = delete(node.child[1-p], node, x);
90 }
91 }
92 node.push_up();
93 if(node.diff==-2 || node.diff==2) {
94 int p = (node.diff + 2) / 4, q = -node.diff/2;
95 if(node.child[p].diff == q)
96 node.child[p] = rotate(node.child[p], 1 - p);
97 node = rotate(node, p);
98 }
99 return node;
100 }
101 private static TreeNode rotate(TreeNode node, int direction) {
102 int p = direction, q = 1 - direction;
103 TreeNode tmp = node.child[p];
104 node.child[p] = tmp.child[q];
105 tmp.child[q] = node;
106 tmp.father = node.father;
107 node.father = tmp;
108 node.push_up();
109 tmp.push_up();
110 return tmp;
111 }
112 private static int queryRank(TreeNode node, int x) {
113 if(node == null) return 0;
114 if(x < node.value)
115 return queryRank(node.child[0], x);
116 else if(node.value < x)
117 return queryRank(node.child[1], x) + node.num + node.childWeight(0);
118 return node.childWeight(0) + 1;
119 }
120 private static int queryVal(TreeNode node, int index) {
121 if(node == null) return 0;
122 if(node.childWeight(0) >= index)
123 return queryVal(node.child[0], index);
124 else if(node.childWeight(0) + node.num < index)
125 return queryVal(node.child[1], index - node.childWeight(0) - node.num );
126 return node.value;
127 }
128 private static void findPrev(TreeNode node, int x) {
129 if(node == null) return ;
130 if(node.value < x) {
131 buffer = node;
132 findPrev(node.child[1], x);
133 }else findPrev(node.child[0], x);
134 return ;
135 }
136 private static void findPost(TreeNode node, int x) {
137 if(node == null) return ;
138 if(node.value > x) {
139 buffer = node;
140 findPost(node.child[0], x);
141 }else findPost(node.child[1], x);
142 return ;
143 }
144 }
145 class TreeNode {
146 int value, diff, height, num, weight;
147 TreeNode father;
148 TreeNode[] child;
149 TreeNode() {}
150 TreeNode(int x) {
151 value = x;
152 diff = 0;
153 num = 1;
154 weight = 1;
155 height = 0;
156 father = null;
157 child = new TreeNode[2];
158 }
159 TreeNode(int x, TreeNode fa) {
160 value = x;
161 diff = 0;
162 num = 1;
163 weight = 1;
164 height = 0;
165 father = fa;
166 child = new TreeNode[2];
167 }
168 public void push_up() {
169 int h1 = heightOf(child[1]), h0 = heightOf(child[0]);
170 height = max(h0, h1) + 1;
171 diff = h1 - h0;
172 weight = childWeight(0) + childWeight(1) + num;
173 return ;
174 }
175 public int childWeight(int i){
176 if(child[i] != null) return child[i].weight;
177 return 0;
178 }
179 private static int heightOf(TreeNode node) {
180 if(node != null) return node.height;
181 return -1;
182 }
183 private static int max(int x, int y){
184 if(x > y) return x;
185 return y;
186 }
187 }
CSC3100的更多相关文章
随机推荐
- 后台运行程序-服务器、python
0前言 最近遇到一个需求,我有一个很小的python程序,需要一直在服务器器上跑,但是我不想一直开着浏览器或者Xshell 7,所以记录一下怎么解决的. 用到的指令是nohup 具体代码就两行 sou ...
- Spring 类名后缀理解
Aware 理解 实现Spring的Aware接口. 定义为感知.意识,核心意义在于通过Aware可以把spring底层组件注入到自定义的bean中. 对于bean与容器的关系,bean不应该知道自身 ...
- 你所不知道的 C# 10新特性
我们很高兴地宣布 C# 10 作为 .NET 6 和 Visual Studio 2022 的一部分已经发布了.在这篇文章中,我们将介绍 C# 10 的许多新功能,这些功能使您的代码更漂亮.更具表现力 ...
- 使用Java的GUI技术实现 “ 贪吃蛇 ” 游戏
详细教程: 使用Java的GUI技术实现 " 贪吃蛇 " 游戏_IT打工酱的博客-CSDN博客
- AfterLogicWebMail CSRF导致密码可修改
实验目的 了解CSRF漏洞导致Webmail管理员帐号密码任意被修改 实验原理 当我们打开或者登陆某个网站的时候,浏览器与网站所存放的服务器将会产生一个会话(cookies),在这个会话没有结束时,你 ...
- 新建一个scrapy项目
此次是做一个豆瓣的top250信息的抓取 首先打开pycharm 在pycharm的下端的Terminal中输入scrapy startproject douban 此时系统就生成了以下文件(spid ...
- [c语言]运算符的优先级与结合性
c语言中运算符的优先级和结合性常常被人混淆一谈,本文目的在于简单谈谈两者的区别.本文举几个简单的例子说明,这些运算符也特别常用. 首先要明白的是:优先级决定表达式中各种不同的运算符起作用的优先次序:而 ...
- 从这3个方面考虑BI工具，选型一选一个准
BI工具在很多场合都能听到,那么BI工具有什么功能呢?能给企业带来什么?好用的BI工具长什么样?今天跟着小编走近BI工具,一探究竟! 首先要了解BI工具的定义,什么是BI工具.BI工具是指利用现代数据 ...
- AHUACM寒假集训II（线段树）
B.Mayor's posters POJ2528 题目大意: D.Count Color POJ2777 题目大意:长为 L ( L ≤ 1 0 5 ) L( L\leq10^5) L(L≤105) ...
- 控制台console不打印信息的解决办法
一直用控制台调试,突然不知道怎么回事看不到控制台输出的信息了: 需要检查下面几方面的问题: 1:我的就属于第一个问题,不知道怎么搜索的时候就改变了Filter; 2:确保以上选项是勾选的 3:点击设置 ...