stakach-algorithms 1.0.6 → 1.0.7
Sign up to get free protection for your applications and to get access to all the features.
- data/README.markdown +99 -97
- data/Rakefile +15 -27
- data/ext/algorithms/string/extconf.rb +4 -4
- data/ext/algorithms/string/string.c +68 -70
- data/ext/containers/bst/bst.c +249 -249
- data/ext/containers/bst/extconf.rb +4 -4
- data/ext/containers/deque/deque.c +248 -248
- data/ext/containers/deque/extconf.rb +4 -4
- data/ext/containers/rbtree_map/extconf.rb +4 -4
- data/ext/containers/rbtree_map/rbtree.c +500 -500
- data/ext/containers/splaytree_map/extconf.rb +4 -4
- data/ext/containers/splaytree_map/splaytree.c +421 -421
- data/lib/algorithms.rb +69 -69
- data/lib/algorithms/search.rb +85 -85
- data/lib/algorithms/sort.rb +242 -242
- data/lib/algorithms/string.rb +10 -10
- data/lib/algorithms/version.rb +3 -3
- data/lib/containers/deque.rb +176 -176
- data/lib/containers/heap.rb +501 -506
- data/lib/containers/kd_tree.rb +112 -112
- data/lib/containers/priority_queue.rb +116 -116
- data/lib/containers/queue.rb +71 -71
- data/lib/containers/rb_tree_map.rb +402 -402
- data/lib/containers/splay_tree_map.rb +273 -273
- data/lib/containers/stack.rb +70 -70
- data/lib/containers/suffix_array.rb +71 -71
- data/lib/containers/trie.rb +187 -187
- data/spec/bst_gc_mark_spec.rb +25 -0
- data/spec/bst_spec.rb +25 -0
- data/spec/deque_gc_mark_spec.rb +17 -0
- data/spec/deque_spec.rb +107 -0
- data/spec/heap_spec.rb +130 -0
- data/spec/helper.rb +4 -0
- data/spec/kd_expected_out.txt +10000 -0
- data/spec/kd_test_in.txt +10000 -0
- data/spec/kd_tree_spec.rb +33 -0
- data/spec/map_gc_mark_spec.rb +28 -0
- data/spec/priority_queue_spec.rb +74 -0
- data/spec/queue_spec.rb +60 -0
- data/spec/rb_tree_map_spec.rb +122 -0
- data/spec/search_spec.rb +27 -0
- data/spec/sort_spec.rb +26 -0
- data/spec/splay_tree_map_spec.rb +105 -0
- data/spec/stack_spec.rb +59 -0
- data/spec/string_spec.rb +12 -0
- data/spec/suffix_array_spec.rb +39 -0
- data/spec/trie_spec.rb +58 -0
- metadata +60 -4
data/README.markdown
CHANGED
@@ -1,97 +1,99 @@
|
|
1
|
-
# algorithms
|
2
|
-
|
3
|
-
|
4
|
-
|
5
|
-
|
6
|
-
|
7
|
-
|
8
|
-
|
9
|
-
|
10
|
-
|
11
|
-
|
12
|
-
|
13
|
-
|
14
|
-
|
15
|
-
|
16
|
-
algorithms
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
*
|
28
|
-
*
|
29
|
-
*
|
30
|
-
*
|
31
|
-
*
|
32
|
-
*
|
33
|
-
*
|
34
|
-
*
|
35
|
-
|
36
|
-
*
|
37
|
-
|
38
|
-
|
39
|
-
|
40
|
-
-
|
41
|
-
|
42
|
-
-
|
43
|
-
-
|
44
|
-
-
|
45
|
-
-
|
46
|
-
-
|
47
|
-
-
|
48
|
-
|
49
|
-
|
50
|
-
|
51
|
-
|
52
|
-
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
|
59
|
-
|
60
|
-
|
61
|
-
|
62
|
-
|
63
|
-
|
64
|
-
|
65
|
-
|
66
|
-
|
67
|
-
|
68
|
-
|
69
|
-
|
70
|
-
|
71
|
-
|
72
|
-
|
73
|
-
|
74
|
-
|
75
|
-
|
76
|
-
|
77
|
-
|
78
|
-
|
79
|
-
|
80
|
-
|
81
|
-
|
82
|
-
|
83
|
-
|
84
|
-
|
85
|
-
|
86
|
-
the
|
87
|
-
|
88
|
-
|
89
|
-
|
90
|
-
|
91
|
-
|
92
|
-
|
93
|
-
|
94
|
-
|
95
|
-
|
96
|
-
|
97
|
-
|
1
|
+
# algorithms
|
2
|
+
|
3
|
+
[![build status](https://secure.travis-ci.org/stakach/algorithms.png)](http://travis-ci.org/stakach/algorithms)
|
4
|
+
|
5
|
+
* Official homes are here on github, and at [RubyGems](https://rubygems.org/gems/stakach-algorithms)
|
6
|
+
* Documentation: [http://algorithms.rubyforge.org/](http://algorithms.rubyforge.org/)
|
7
|
+
|
8
|
+
## DESCRIPTION:
|
9
|
+
|
10
|
+
Started as a Google Summer of Code 2008 project
|
11
|
+
|
12
|
+
Written by [Kanwei Li](http://kanwei.com/), mentored by Austin Ziegler
|
13
|
+
|
14
|
+
Original Proposal: Using the right data structure or algorithm for the situation is an important
|
15
|
+
aspect of programming. In computer science literature, many data structures
|
16
|
+
and algorithms have been researched and extensively documented. However, there
|
17
|
+
is still no standard library in Ruby implementing useful structures and
|
18
|
+
algorithms like Red/Black Trees, tries, different sorting algorithms, etc.
|
19
|
+
This project will create such a library with documentation on when to use a
|
20
|
+
particular structure/algorithm. It will also come with a benchmark suite to
|
21
|
+
compare performance in different situations.
|
22
|
+
|
23
|
+
## FEATURES:
|
24
|
+
|
25
|
+
Done so far:
|
26
|
+
|
27
|
+
* Heaps Algorithms::Containers::Heap, Containers::MaxHeap, Containers::MinHeap
|
28
|
+
* Priority Queue Algorithms::Containers::PriorityQueue
|
29
|
+
* Deque Algorithms::Containers::Deque, Containers::CDeque (C extension), Containers::RubyDeque
|
30
|
+
* Stack Algorithms::Containers::Stack (uses Deque)
|
31
|
+
* Queue Algorithms::Containers::Queue (uses Deque)
|
32
|
+
* Red-Black Trees Algorithms::Containers::RBTreeMap, Containers::CRBTreeMap (C extension), Containers::RubyRBTreeMap
|
33
|
+
* Splay Trees Algorithms::Containers::SplayTreeMap, Containers::CSplayTreeMap (C extension), Containers::RubySplayTreeMap
|
34
|
+
* Tries Algorithms::Containers::Trie
|
35
|
+
* Suffix Array Algorithms::Containers::SuffixArray
|
36
|
+
* kd Tree Algorithms::Containers::KDTree
|
37
|
+
|
38
|
+
* Search algorithms
|
39
|
+
- Binary Search Algorithms::Algorithms::Search.binary_search
|
40
|
+
- Knuth-Morris-Pratt Algorithms::Algorithms::Search.kmp_search
|
41
|
+
* Sort algorithms
|
42
|
+
- Bubble sort Algorithms::Algorithms::Sort.bubble_sort
|
43
|
+
- Comb sort Algorithms::Algorithms::Sort.comb_sort
|
44
|
+
- Selection sort Algorithms::Algorithms::Sort.selection_sort
|
45
|
+
- Heapsort Algorithms::Algorithms::Sort.heapsort
|
46
|
+
- Insertion sort Algorithms::Algorithms::Sort.insertion_sort
|
47
|
+
- Shell sort Algorithms::Algorithms::Sort.shell_sort
|
48
|
+
- Quicksort Algorithms::Algorithms::Sort.quicksort
|
49
|
+
- Mergesort Algorithms::Algorithms::Sort.mergesort
|
50
|
+
|
51
|
+
## SYNOPSIS:
|
52
|
+
|
53
|
+
require 'rubygems'
|
54
|
+
require 'algorithms'
|
55
|
+
max_heap = Algorithms::Containers::MaxHeap.new
|
56
|
+
|
57
|
+
# To not have to type "Algorithms::" before each class, use:
|
58
|
+
include Algorithms
|
59
|
+
|
60
|
+
max_heap = Containers::MaxHeap.new
|
61
|
+
|
62
|
+
# To not have to type "Containers::" before each class, use:
|
63
|
+
include Containers
|
64
|
+
max_heap = MaxHeap.new
|
65
|
+
|
66
|
+
|
67
|
+
## REQUIREMENTS:
|
68
|
+
|
69
|
+
* Ruby 1.8 compatible Ruby, or Ruby 1.9, should also work with jRuby
|
70
|
+
* C compiler for C extensions (optional, but very much recommended for vast performance benefits)
|
71
|
+
|
72
|
+
## INSTALL:
|
73
|
+
|
74
|
+
* sudo gem install stakach-algorithms
|
75
|
+
|
76
|
+
## LICENSE:
|
77
|
+
|
78
|
+
(The MIT License)
|
79
|
+
|
80
|
+
Algorithms and Containers project is Copyright (c) 2009 Kanwei Li
|
81
|
+
|
82
|
+
Permission is hereby granted, free of charge, to any person obtaining
|
83
|
+
a copy of this software and associated documentation files (the
|
84
|
+
'Software'), to deal in the Software without restriction, including
|
85
|
+
without limitation the rights to use, copy, modify, merge, publish,
|
86
|
+
distribute, sublicense, and/or sell copies of the Software, and to
|
87
|
+
permit persons to whom the Software is furnished to do so, subject to
|
88
|
+
the following conditions:
|
89
|
+
|
90
|
+
The above copyright notice and this permission notice shall be
|
91
|
+
included in all copies or substantial portions of the Software.
|
92
|
+
|
93
|
+
THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
|
94
|
+
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
95
|
+
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
96
|
+
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
97
|
+
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
98
|
+
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
99
|
+
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
data/Rakefile
CHANGED
@@ -1,27 +1,15 @@
|
|
1
|
-
require
|
2
|
-
require
|
3
|
-
|
4
|
-
|
5
|
-
|
6
|
-
|
7
|
-
|
8
|
-
|
9
|
-
|
10
|
-
|
11
|
-
|
12
|
-
|
13
|
-
|
14
|
-
|
15
|
-
|
16
|
-
sh "git push" # Rubyforge
|
17
|
-
sh "git push --tags" # Rubyforge
|
18
|
-
sh "git push gh" # Github
|
19
|
-
sh "git push gh --tags" # Github
|
20
|
-
end
|
21
|
-
|
22
|
-
task :hanna do
|
23
|
-
sh "rm -fr doc"
|
24
|
-
sh "hanna -SN lib/ -m Algorithms"
|
25
|
-
sh "scp -rq doc/* kanwei@rubyforge.org:/var/www/gforge-projects/algorithms"
|
26
|
-
end
|
27
|
-
|
1
|
+
require "rake/clean"
|
2
|
+
require 'rake/extensiontask'
|
3
|
+
|
4
|
+
OBJ = FileList['**/*.rbc']
|
5
|
+
CLEAN.include(OBJ)
|
6
|
+
|
7
|
+
|
8
|
+
Rake::ExtensionTask.new('algorithms/string')
|
9
|
+
Rake::ExtensionTask.new('containers/bst')
|
10
|
+
Rake::ExtensionTask.new('containers/deque')
|
11
|
+
Rake::ExtensionTask.new('containers/rbtree_map')
|
12
|
+
Rake::ExtensionTask.new('containers/splaytree_map')
|
13
|
+
|
14
|
+
|
15
|
+
|
@@ -1,4 +1,4 @@
|
|
1
|
-
require 'mkmf'
|
2
|
-
extension_name = "CString"
|
3
|
-
dir_config(extension_name)
|
4
|
-
create_makefile(extension_name)
|
1
|
+
require 'mkmf'
|
2
|
+
extension_name = "CString"
|
3
|
+
dir_config(extension_name)
|
4
|
+
create_makefile(extension_name)
|
@@ -1,70 +1,68 @@
|
|
1
|
-
#include "ruby.h"
|
2
|
-
|
3
|
-
|
4
|
-
|
5
|
-
if (b < min)
|
6
|
-
min = b;
|
7
|
-
if( c < min)
|
8
|
-
min = c;
|
9
|
-
return min;
|
10
|
-
}
|
11
|
-
|
12
|
-
|
13
|
-
|
14
|
-
char * s = RSTRING_PTR(str1);
|
15
|
-
char * t = RSTRING_PTR(str2);
|
16
|
-
s1_len = RSTRING_LEN(str1);
|
17
|
-
s2_len = RSTRING_LEN(str2);
|
18
|
-
|
19
|
-
if (s1_len == 0) {
|
20
|
-
return s2_len;
|
21
|
-
} else if (s2_len == 0) {
|
22
|
-
return s1_len;
|
23
|
-
}
|
24
|
-
|
25
|
-
// We need one extra col and row for the matrix for starting values
|
26
|
-
s1_len++;
|
27
|
-
s2_len++;
|
28
|
-
|
29
|
-
d = malloc(sizeof(
|
30
|
-
|
31
|
-
for (i = 0; i < s1_len; i++) {
|
32
|
-
d[i] = i; // d[i, 0] = i
|
33
|
-
}
|
34
|
-
for (j = 0; j < s2_len; j++) {
|
35
|
-
d[j*s1_len] = j; // d[0, j] = j
|
36
|
-
}
|
37
|
-
|
38
|
-
for (i = 1; i < s1_len; i++) {
|
39
|
-
for (j = 1; j < s2_len; j++) {
|
40
|
-
if (s[i-1] == t[j-1]) {
|
41
|
-
d[j * s1_len + i] = d[(j-1) * s1_len + (i-1)];
|
42
|
-
} else {
|
43
|
-
d[j * s1_len + i] =
|
44
|
-
d[j * s1_len + (i-1)],
|
45
|
-
d[(j-1) * s1_len + i],
|
46
|
-
d[(j-1) * s1_len + (i-1)]
|
47
|
-
)
|
48
|
-
}
|
49
|
-
}
|
50
|
-
}
|
51
|
-
i = d[s1_len * s2_len -1];
|
52
|
-
free(d);
|
53
|
-
return i;
|
54
|
-
}
|
55
|
-
|
56
|
-
static VALUE lev_dist(VALUE self, VALUE str1, VALUE str2) {
|
57
|
-
return
|
58
|
-
}
|
59
|
-
|
60
|
-
static VALUE
|
61
|
-
static VALUE
|
62
|
-
|
63
|
-
|
64
|
-
|
65
|
-
|
66
|
-
|
67
|
-
|
68
|
-
|
69
|
-
}
|
70
|
-
|
1
|
+
#include "ruby.h"
|
2
|
+
|
3
|
+
long min_three(long a, long b, long c) {
|
4
|
+
long min = a;
|
5
|
+
if (b < min)
|
6
|
+
min = b;
|
7
|
+
if( c < min)
|
8
|
+
min = c;
|
9
|
+
return min;
|
10
|
+
}
|
11
|
+
|
12
|
+
long levenshtein_distance(VALUE str1, VALUE str2) {
|
13
|
+
long i, j, s1_len, s2_len, *d;
|
14
|
+
char * s = RSTRING_PTR(str1);
|
15
|
+
char * t = RSTRING_PTR(str2);
|
16
|
+
s1_len = RSTRING_LEN(str1);
|
17
|
+
s2_len = RSTRING_LEN(str2);
|
18
|
+
|
19
|
+
if (s1_len == 0) {
|
20
|
+
return s2_len;
|
21
|
+
} else if (s2_len == 0) {
|
22
|
+
return s1_len;
|
23
|
+
}
|
24
|
+
|
25
|
+
// We need one extra col and row for the matrix for starting values
|
26
|
+
s1_len++;
|
27
|
+
s2_len++;
|
28
|
+
|
29
|
+
d = malloc(sizeof(typeof(d)) * (s1_len) * (s2_len));
|
30
|
+
|
31
|
+
for (i = 0; i < s1_len; i++) {
|
32
|
+
d[i] = i; // d[i, 0] = i
|
33
|
+
}
|
34
|
+
for (j = 0; j < s2_len; j++) {
|
35
|
+
d[j*s1_len] = j; // d[0, j] = j
|
36
|
+
}
|
37
|
+
|
38
|
+
for (i = 1; i < s1_len; i++) {
|
39
|
+
for (j = 1; j < s2_len; j++) {
|
40
|
+
if (s[i-1] == t[j-1]) {
|
41
|
+
d[j * s1_len + i] = d[(j-1) * s1_len + (i-1)];
|
42
|
+
} else {
|
43
|
+
d[j * s1_len + i] = 1 + min_three(
|
44
|
+
d[j * s1_len + (i-1)],
|
45
|
+
d[(j-1) * s1_len + i],
|
46
|
+
d[(j-1) * s1_len + (i-1)]
|
47
|
+
);
|
48
|
+
}
|
49
|
+
}
|
50
|
+
}
|
51
|
+
i = d[s1_len * s2_len -1];
|
52
|
+
free(d);
|
53
|
+
return i;
|
54
|
+
}
|
55
|
+
|
56
|
+
static VALUE lev_dist(VALUE self, VALUE str1, VALUE str2) {
|
57
|
+
return LONG2FIX(levenshtein_distance( str1, str2 ));
|
58
|
+
}
|
59
|
+
|
60
|
+
static VALUE mAlgorithms;
|
61
|
+
static VALUE mString;
|
62
|
+
|
63
|
+
void Init_CString() {
|
64
|
+
mAlgorithms = rb_define_module("Algorithms");
|
65
|
+
mString = rb_define_module_under(mAlgorithms, "String");
|
66
|
+
rb_define_singleton_method(mString, "levenshtein_dist", lev_dist, 2);
|
67
|
+
}
|
68
|
+
|
data/ext/containers/bst/bst.c
CHANGED
@@ -1,249 +1,249 @@
|
|
1
|
-
#include "ruby.h"
|
2
|
-
|
3
|
-
typedef struct struct_bst_node {
|
4
|
-
VALUE key;
|
5
|
-
VALUE value;
|
6
|
-
struct struct_bst_node *left;
|
7
|
-
struct struct_bst_node *right;
|
8
|
-
struct struct_bst_node *parent;
|
9
|
-
} bst_node;
|
10
|
-
|
11
|
-
typedef struct struct_bst {
|
12
|
-
bst_node *root;
|
13
|
-
int (*compare_function)(VALUE key1, VALUE key2);
|
14
|
-
unsigned int size;
|
15
|
-
} bst;
|
16
|
-
|
17
|
-
static VALUE bst_initialize(VALUE self) {
|
18
|
-
return self;
|
19
|
-
}
|
20
|
-
|
21
|
-
static bst* get_bst_from_self(VALUE self) {
|
22
|
-
bst *tree;
|
23
|
-
Data_Get_Struct(self, bst, tree);
|
24
|
-
return tree;
|
25
|
-
}
|
26
|
-
|
27
|
-
static bst* bst_each_node(bst *tree, bst_node *node, void (*each)(bst *tree_, bst_node *node_, void* args), void* arguments) {
|
28
|
-
if (!node)
|
29
|
-
return NULL;
|
30
|
-
|
31
|
-
if (node->left)
|
32
|
-
bst_each_node(tree, node->left, each, arguments);
|
33
|
-
|
34
|
-
(*each)(tree, node, arguments);
|
35
|
-
|
36
|
-
if (node->right)
|
37
|
-
bst_each_node(tree, node->right, each, arguments);
|
38
|
-
return tree;
|
39
|
-
}
|
40
|
-
|
41
|
-
static bst* bst_each(bst *tree, void (*each)(bst *tree, bst_node *node, void *args), void* arguments) {
|
42
|
-
if (tree->root)
|
43
|
-
bst_each_node(tree, tree->root, each, arguments);
|
44
|
-
return tree;
|
45
|
-
}
|
46
|
-
|
47
|
-
static
|
48
|
-
|
49
|
-
static int bst_compare_function(VALUE a, VALUE b) {
|
50
|
-
if (a == b) return 0;
|
51
|
-
if (FIXNUM_P(a) && FIXNUM_P(b)) {
|
52
|
-
long x = FIX2LONG(a), y = FIX2LONG(b);
|
53
|
-
if (x == y) return 0;
|
54
|
-
if (x > y) return 1;
|
55
|
-
return -1;
|
56
|
-
}
|
57
|
-
if (TYPE(a) == T_STRING && rb_obj_is_kind_of(a, rb_cString) &&
|
58
|
-
TYPE(b) == T_STRING && rb_obj_is_kind_of(b, rb_cString)) {
|
59
|
-
return rb_str_cmp(a, b);
|
60
|
-
}
|
61
|
-
return FIX2INT(rb_funcall((VALUE) a, id_compare_operator, 1, (VALUE) b));
|
62
|
-
}
|
63
|
-
|
64
|
-
static void insert_element(bst *tree, bst_node **t,bst_node *newElement) {
|
65
|
-
int cmp;
|
66
|
-
bst_node *y = NULL;
|
67
|
-
bst_node *x = *t;
|
68
|
-
while (x != NULL) {
|
69
|
-
y = x;
|
70
|
-
cmp = tree->compare_function(newElement->key, x->key);
|
71
|
-
if (cmp < 0) x = x->left;
|
72
|
-
else x = x->right;
|
73
|
-
}
|
74
|
-
newElement->parent = y;
|
75
|
-
if (y == NULL) *t = newElement;
|
76
|
-
else {
|
77
|
-
cmp = tree->compare_function(newElement->key, y->key);
|
78
|
-
if (cmp < 0)
|
79
|
-
y->left = newElement;
|
80
|
-
else
|
81
|
-
y->right = newElement;
|
82
|
-
}
|
83
|
-
}
|
84
|
-
|
85
|
-
|
86
|
-
static bst_node* create_node(VALUE key_value,VALUE value) {
|
87
|
-
bst_node *new_node = ALLOC(bst_node);
|
88
|
-
new_node->value = value;
|
89
|
-
new_node->key = key_value;
|
90
|
-
new_node->left = NULL;
|
91
|
-
new_node->right = NULL;
|
92
|
-
new_node->parent = NULL;
|
93
|
-
return new_node;
|
94
|
-
}
|
95
|
-
|
96
|
-
static bst_node* tree_minimum (bst_node *tree) {
|
97
|
-
bst_node *x = tree;
|
98
|
-
while (x->left) x = x->left;
|
99
|
-
return x;
|
100
|
-
}
|
101
|
-
|
102
|
-
static bst_node* tree_maximum (bst_node *tree) {
|
103
|
-
bst_node *x = tree;
|
104
|
-
while (x->right) x = x->right;
|
105
|
-
return x;
|
106
|
-
}
|
107
|
-
|
108
|
-
static bst_node* node_successor (bst_node *tree,bst_node *x) {
|
109
|
-
if (x->right) return tree_minimum(x->right);
|
110
|
-
bst_node *y = x->parent;
|
111
|
-
while (y && x == y->right) {
|
112
|
-
x = y;
|
113
|
-
y = x->parent;
|
114
|
-
}
|
115
|
-
return y;
|
116
|
-
}
|
117
|
-
|
118
|
-
|
119
|
-
static bst_node* delete_node (bst_node **tree,bst_node *tobeDeleted) {
|
120
|
-
bst_node *y,*x;
|
121
|
-
|
122
|
-
if ((tobeDeleted->left == NULL) || (tobeDeleted->right == NULL)) y = tobeDeleted;
|
123
|
-
else y = node_successor(*tree,tobeDeleted);
|
124
|
-
|
125
|
-
if (y->left) x = y->left;
|
126
|
-
else x = y->right;
|
127
|
-
|
128
|
-
if (x) x->parent = y->parent;
|
129
|
-
|
130
|
-
if (y->parent == NULL) {
|
131
|
-
*tree = x;
|
132
|
-
return y;
|
133
|
-
} else if (y == y->parent->left) {
|
134
|
-
y->parent->left = x;
|
135
|
-
} else {
|
136
|
-
y->parent->right = x;
|
137
|
-
}
|
138
|
-
|
139
|
-
if (tobeDeleted != y) tobeDeleted->key = y->key;
|
140
|
-
return y;
|
141
|
-
}
|
142
|
-
|
143
|
-
static bst_node* search_node(bst *tree, bst_node *node, VALUE key) {
|
144
|
-
bst_node *x = node;
|
145
|
-
int cmp;
|
146
|
-
|
147
|
-
while(x) {
|
148
|
-
cmp = tree->compare_function(key, x->key);
|
149
|
-
if (cmp == 0) return x;
|
150
|
-
else if (cmp < 0) { x = x->left; }
|
151
|
-
else { x = x->right; }
|
152
|
-
}
|
153
|
-
return NULL;
|
154
|
-
}
|
155
|
-
|
156
|
-
static void recursively_mark_nodes(bst_node *node) {
|
157
|
-
if(node) {
|
158
|
-
rb_gc_mark(node->key);
|
159
|
-
rb_gc_mark(node->value);
|
160
|
-
recursively_mark_nodes(node->left);
|
161
|
-
recursively_mark_nodes(node->right);
|
162
|
-
}
|
163
|
-
}
|
164
|
-
|
165
|
-
static void bst_mark(bst *tree) {
|
166
|
-
if (tree) {
|
167
|
-
recursively_mark_nodes(tree->root);
|
168
|
-
}
|
169
|
-
}
|
170
|
-
|
171
|
-
static void recursively_free_nodes(bst_node *node) {
|
172
|
-
if(node) {
|
173
|
-
recursively_free_nodes(node->left);
|
174
|
-
recursively_free_nodes(node->right);
|
175
|
-
free(node);
|
176
|
-
}
|
177
|
-
}
|
178
|
-
|
179
|
-
static void bst_free(bst *tree) {
|
180
|
-
if (tree) {
|
181
|
-
recursively_free_nodes(tree->root);
|
182
|
-
}
|
183
|
-
}
|
184
|
-
|
185
|
-
static bst* create_bst(int (*compare_function)(VALUE, VALUE)) {
|
186
|
-
bst *tree = ALLOC(bst);
|
187
|
-
tree->compare_function = compare_function;
|
188
|
-
tree->root = NULL;
|
189
|
-
tree->size = 0;
|
190
|
-
return tree;
|
191
|
-
}
|
192
|
-
|
193
|
-
static VALUE bst_alloc(VALUE klass) {
|
194
|
-
bst *tree = create_bst(&bst_compare_function);
|
195
|
-
return Data_Wrap_Struct(klass, bst_mark, bst_free, tree);
|
196
|
-
}
|
197
|
-
|
198
|
-
static VALUE rb_bst_push_value(VALUE self, VALUE key, VALUE value) {
|
199
|
-
bst *tree = get_bst_from_self(self);
|
200
|
-
insert_element(tree, &(tree->root), create_node(key,value));
|
201
|
-
tree->size++;
|
202
|
-
return self;
|
203
|
-
}
|
204
|
-
|
205
|
-
static void bst_each_helper(bst *tree, bst_node *node, void *args) {
|
206
|
-
rb_yield(rb_ary_new3(2, node->key, node->value));
|
207
|
-
};
|
208
|
-
|
209
|
-
static VALUE rb_bst_each(VALUE self) {
|
210
|
-
bst *tree = get_bst_from_self(self);
|
211
|
-
bst_each(tree, &bst_each_helper, NULL);
|
212
|
-
return self;
|
213
|
-
}
|
214
|
-
|
215
|
-
static VALUE rb_bst_delete(VALUE self, VALUE key) {
|
216
|
-
bst *tree = get_bst_from_self(self);
|
217
|
-
bst_node *tobeDeleted = search_node(tree, tree->root, key);
|
218
|
-
if(tobeDeleted) {
|
219
|
-
tree->size -= 1;
|
220
|
-
bst_node *deletedNode = delete_node(&(tree->root),tobeDeleted);
|
221
|
-
return deletedNode->value;
|
222
|
-
}
|
223
|
-
return Qnil;
|
224
|
-
}
|
225
|
-
|
226
|
-
static VALUE rb_bst_size(VALUE self) {
|
227
|
-
bst *tree;
|
228
|
-
Data_Get_Struct(self,bst,tree);
|
229
|
-
return INT2FIX(tree->size);
|
230
|
-
}
|
231
|
-
|
232
|
-
static VALUE CBst;
|
233
|
-
static VALUE mContainers;
|
234
|
-
static VALUE namespace;
|
235
|
-
|
236
|
-
void Init_CBst() {
|
237
|
-
id_compare_operator = rb_intern("<=>");
|
238
|
-
|
239
|
-
namespace = rb_define_module("Algorithms");
|
240
|
-
mContainers = rb_define_module_under(namespace,"Containers");
|
241
|
-
CBst = rb_define_class_under(mContainers, "CBst", rb_cObject);
|
242
|
-
rb_define_alloc_func(CBst, bst_alloc);
|
243
|
-
rb_define_method(CBst, "initialize", bst_initialize, 0);
|
244
|
-
rb_define_method(CBst, "push", rb_bst_push_value, 2);
|
245
|
-
rb_define_alias(CBst, "[]=", "push");
|
246
|
-
rb_define_method(CBst, "each", rb_bst_each, 0);
|
247
|
-
rb_define_method(CBst, "delete", rb_bst_delete, 1);
|
248
|
-
rb_define_method(CBst, "size", rb_bst_size, 0);
|
249
|
-
}
|
1
|
+
#include "ruby.h"
|
2
|
+
|
3
|
+
typedef struct struct_bst_node {
|
4
|
+
VALUE key;
|
5
|
+
VALUE value;
|
6
|
+
struct struct_bst_node *left;
|
7
|
+
struct struct_bst_node *right;
|
8
|
+
struct struct_bst_node *parent;
|
9
|
+
} bst_node;
|
10
|
+
|
11
|
+
typedef struct struct_bst {
|
12
|
+
bst_node *root;
|
13
|
+
int (*compare_function)(VALUE key1, VALUE key2);
|
14
|
+
unsigned int size;
|
15
|
+
} bst;
|
16
|
+
|
17
|
+
static VALUE bst_initialize(VALUE self) {
|
18
|
+
return self;
|
19
|
+
}
|
20
|
+
|
21
|
+
static bst* get_bst_from_self(VALUE self) {
|
22
|
+
bst *tree;
|
23
|
+
Data_Get_Struct(self, bst, tree);
|
24
|
+
return tree;
|
25
|
+
}
|
26
|
+
|
27
|
+
static bst* bst_each_node(bst *tree, bst_node *node, void (*each)(bst *tree_, bst_node *node_, void* args), void* arguments) {
|
28
|
+
if (!node)
|
29
|
+
return NULL;
|
30
|
+
|
31
|
+
if (node->left)
|
32
|
+
bst_each_node(tree, node->left, each, arguments);
|
33
|
+
|
34
|
+
(*each)(tree, node, arguments);
|
35
|
+
|
36
|
+
if (node->right)
|
37
|
+
bst_each_node(tree, node->right, each, arguments);
|
38
|
+
return tree;
|
39
|
+
}
|
40
|
+
|
41
|
+
static bst* bst_each(bst *tree, void (*each)(bst *tree, bst_node *node, void *args), void* arguments) {
|
42
|
+
if (tree->root)
|
43
|
+
bst_each_node(tree, tree->root, each, arguments);
|
44
|
+
return tree;
|
45
|
+
}
|
46
|
+
|
47
|
+
static VALUE id_compare_operator;
|
48
|
+
|
49
|
+
static int bst_compare_function(VALUE a, VALUE b) {
|
50
|
+
if (a == b) return 0;
|
51
|
+
if (FIXNUM_P(a) && FIXNUM_P(b)) {
|
52
|
+
long x = FIX2LONG(a), y = FIX2LONG(b);
|
53
|
+
if (x == y) return 0;
|
54
|
+
if (x > y) return 1;
|
55
|
+
return -1;
|
56
|
+
}
|
57
|
+
if (TYPE(a) == T_STRING && rb_obj_is_kind_of(a, rb_cString) &&
|
58
|
+
TYPE(b) == T_STRING && rb_obj_is_kind_of(b, rb_cString)) {
|
59
|
+
return rb_str_cmp(a, b);
|
60
|
+
}
|
61
|
+
return FIX2INT(rb_funcall((VALUE) a, id_compare_operator, 1, (VALUE) b));
|
62
|
+
}
|
63
|
+
|
64
|
+
static void insert_element(bst *tree, bst_node **t,bst_node *newElement) {
|
65
|
+
int cmp;
|
66
|
+
bst_node *y = NULL;
|
67
|
+
bst_node *x = *t;
|
68
|
+
while (x != NULL) {
|
69
|
+
y = x;
|
70
|
+
cmp = tree->compare_function(newElement->key, x->key);
|
71
|
+
if (cmp < 0) x = x->left;
|
72
|
+
else x = x->right;
|
73
|
+
}
|
74
|
+
newElement->parent = y;
|
75
|
+
if (y == NULL) *t = newElement;
|
76
|
+
else {
|
77
|
+
cmp = tree->compare_function(newElement->key, y->key);
|
78
|
+
if (cmp < 0)
|
79
|
+
y->left = newElement;
|
80
|
+
else
|
81
|
+
y->right = newElement;
|
82
|
+
}
|
83
|
+
}
|
84
|
+
|
85
|
+
|
86
|
+
static bst_node* create_node(VALUE key_value,VALUE value) {
|
87
|
+
bst_node *new_node = ALLOC(bst_node);
|
88
|
+
new_node->value = value;
|
89
|
+
new_node->key = key_value;
|
90
|
+
new_node->left = NULL;
|
91
|
+
new_node->right = NULL;
|
92
|
+
new_node->parent = NULL;
|
93
|
+
return new_node;
|
94
|
+
}
|
95
|
+
|
96
|
+
static bst_node* tree_minimum (bst_node *tree) {
|
97
|
+
bst_node *x = tree;
|
98
|
+
while (x->left) x = x->left;
|
99
|
+
return x;
|
100
|
+
}
|
101
|
+
|
102
|
+
static bst_node* tree_maximum (bst_node *tree) {
|
103
|
+
bst_node *x = tree;
|
104
|
+
while (x->right) x = x->right;
|
105
|
+
return x;
|
106
|
+
}
|
107
|
+
|
108
|
+
static bst_node* node_successor (bst_node *tree,bst_node *x) {
|
109
|
+
if (x->right) return tree_minimum(x->right);
|
110
|
+
bst_node *y = x->parent;
|
111
|
+
while (y && x == y->right) {
|
112
|
+
x = y;
|
113
|
+
y = x->parent;
|
114
|
+
}
|
115
|
+
return y;
|
116
|
+
}
|
117
|
+
|
118
|
+
|
119
|
+
static bst_node* delete_node (bst_node **tree,bst_node *tobeDeleted) {
|
120
|
+
bst_node *y,*x;
|
121
|
+
|
122
|
+
if ((tobeDeleted->left == NULL) || (tobeDeleted->right == NULL)) y = tobeDeleted;
|
123
|
+
else y = node_successor(*tree,tobeDeleted);
|
124
|
+
|
125
|
+
if (y->left) x = y->left;
|
126
|
+
else x = y->right;
|
127
|
+
|
128
|
+
if (x) x->parent = y->parent;
|
129
|
+
|
130
|
+
if (y->parent == NULL) {
|
131
|
+
*tree = x;
|
132
|
+
return y;
|
133
|
+
} else if (y == y->parent->left) {
|
134
|
+
y->parent->left = x;
|
135
|
+
} else {
|
136
|
+
y->parent->right = x;
|
137
|
+
}
|
138
|
+
|
139
|
+
if (tobeDeleted != y) tobeDeleted->key = y->key;
|
140
|
+
return y;
|
141
|
+
}
|
142
|
+
|
143
|
+
static bst_node* search_node(bst *tree, bst_node *node, VALUE key) {
|
144
|
+
bst_node *x = node;
|
145
|
+
int cmp;
|
146
|
+
|
147
|
+
while(x) {
|
148
|
+
cmp = tree->compare_function(key, x->key);
|
149
|
+
if (cmp == 0) return x;
|
150
|
+
else if (cmp < 0) { x = x->left; }
|
151
|
+
else { x = x->right; }
|
152
|
+
}
|
153
|
+
return NULL;
|
154
|
+
}
|
155
|
+
|
156
|
+
static void recursively_mark_nodes(bst_node *node) {
|
157
|
+
if(node) {
|
158
|
+
rb_gc_mark(node->key);
|
159
|
+
rb_gc_mark(node->value);
|
160
|
+
recursively_mark_nodes(node->left);
|
161
|
+
recursively_mark_nodes(node->right);
|
162
|
+
}
|
163
|
+
}
|
164
|
+
|
165
|
+
static void bst_mark(bst *tree) {
|
166
|
+
if (tree) {
|
167
|
+
recursively_mark_nodes(tree->root);
|
168
|
+
}
|
169
|
+
}
|
170
|
+
|
171
|
+
static void recursively_free_nodes(bst_node *node) {
|
172
|
+
if(node) {
|
173
|
+
recursively_free_nodes(node->left);
|
174
|
+
recursively_free_nodes(node->right);
|
175
|
+
free(node);
|
176
|
+
}
|
177
|
+
}
|
178
|
+
|
179
|
+
static void bst_free(bst *tree) {
|
180
|
+
if (tree) {
|
181
|
+
recursively_free_nodes(tree->root);
|
182
|
+
}
|
183
|
+
}
|
184
|
+
|
185
|
+
static bst* create_bst(int (*compare_function)(VALUE, VALUE)) {
|
186
|
+
bst *tree = ALLOC(bst);
|
187
|
+
tree->compare_function = compare_function;
|
188
|
+
tree->root = NULL;
|
189
|
+
tree->size = 0;
|
190
|
+
return tree;
|
191
|
+
}
|
192
|
+
|
193
|
+
static VALUE bst_alloc(VALUE klass) {
|
194
|
+
bst *tree = create_bst(&bst_compare_function);
|
195
|
+
return Data_Wrap_Struct(klass, bst_mark, bst_free, tree);
|
196
|
+
}
|
197
|
+
|
198
|
+
static VALUE rb_bst_push_value(VALUE self, VALUE key, VALUE value) {
|
199
|
+
bst *tree = get_bst_from_self(self);
|
200
|
+
insert_element(tree, &(tree->root), create_node(key,value));
|
201
|
+
tree->size++;
|
202
|
+
return self;
|
203
|
+
}
|
204
|
+
|
205
|
+
static void bst_each_helper(bst *tree, bst_node *node, void *args) {
|
206
|
+
rb_yield(rb_ary_new3(2, node->key, node->value));
|
207
|
+
};
|
208
|
+
|
209
|
+
static VALUE rb_bst_each(VALUE self) {
|
210
|
+
bst *tree = get_bst_from_self(self);
|
211
|
+
bst_each(tree, &bst_each_helper, NULL);
|
212
|
+
return self;
|
213
|
+
}
|
214
|
+
|
215
|
+
static VALUE rb_bst_delete(VALUE self, VALUE key) {
|
216
|
+
bst *tree = get_bst_from_self(self);
|
217
|
+
bst_node *tobeDeleted = search_node(tree, tree->root, key);
|
218
|
+
if(tobeDeleted) {
|
219
|
+
tree->size -= 1;
|
220
|
+
bst_node *deletedNode = delete_node(&(tree->root),tobeDeleted);
|
221
|
+
return deletedNode->value;
|
222
|
+
}
|
223
|
+
return Qnil;
|
224
|
+
}
|
225
|
+
|
226
|
+
static VALUE rb_bst_size(VALUE self) {
|
227
|
+
bst *tree;
|
228
|
+
Data_Get_Struct(self,bst,tree);
|
229
|
+
return INT2FIX(tree->size);
|
230
|
+
}
|
231
|
+
|
232
|
+
static VALUE CBst;
|
233
|
+
static VALUE mContainers;
|
234
|
+
static VALUE namespace;
|
235
|
+
|
236
|
+
void Init_CBst() {
|
237
|
+
id_compare_operator = rb_intern("<=>");
|
238
|
+
|
239
|
+
namespace = rb_define_module("Algorithms");
|
240
|
+
mContainers = rb_define_module_under(namespace,"Containers");
|
241
|
+
CBst = rb_define_class_under(mContainers, "CBst", rb_cObject);
|
242
|
+
rb_define_alloc_func(CBst, bst_alloc);
|
243
|
+
rb_define_method(CBst, "initialize", bst_initialize, 0);
|
244
|
+
rb_define_method(CBst, "push", rb_bst_push_value, 2);
|
245
|
+
rb_define_alias(CBst, "[]=", "push");
|
246
|
+
rb_define_method(CBst, "each", rb_bst_each, 0);
|
247
|
+
rb_define_method(CBst, "delete", rb_bst_delete, 1);
|
248
|
+
rb_define_method(CBst, "size", rb_bst_size, 0);
|
249
|
+
}
|