summaryrefslogtreecommitdiff
path: root/bin/ai/library/queue/priority_queue/main.nut
blob: bafc93ac5da03264b1cd4302fcb129c464cf8e97 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/* $Id$ */

/**
 * Priority Queue.
 *  Peek and Pop always return the current lowest value in the list.
 *  Sort is done on insertion only.
 */
class PriorityQueue
{
	_queue = null;
	_count = 0;

	constructor()
	{
		_count = 0;
		_queue = [];
	}

	/**
	 * Insert a new entry in the list.
	 *  The complexity of this operation is O(n).
	 * @param item The item to add to the list.
	 * @param priority The priority this item has.
	 */
	function Insert(item, priority);

	/**
	 * Pop the first entry of the list.
	 *  This is always the item with the lowest priority.
	 *  The complexity of this operation is O(1).
	 * @return The item of the entry with the lowest priority.
	 */
	function Pop();

	/**
	 * Peek the first entry of the list.
	 *  This is always the item with the lowest priority.
	 *  The complexity of this operation is O(1).
	 * @return The item of the entry with the lowest priority.
	 */
	function Peek();

	/**
	 * Get the amount of current items in the list.
	 *  The complexity of this operation is O(1).
	 * @return The amount of items currently in the list.
	 */
	function Count();

	/**
	 * Check if an item exists in the list.
	 *  The complexity of this operation is O(n).
	 * @param item The item to check for.
	 * @return True if the item is already in the list.
	 */
	function Exists(item);
};

function PriorityQueue::Insert(item, priority)
{
	/* Append dummy entry */
	_queue.append(0);
	_count++;

	local i;
	/* Find the point of insertion */
	for (i = _count - 2; i >= 0; i--) {
		if (priority > _queue[i][1]) {
			/* All items bigger move one place to the right */
			_queue[i + 1] = _queue[i];
		} else if (item == _queue[i][0]) {
			/* Same item, ignore insertion */
			return false;
		} else {
			/* Found place to insert at */
			break;
		}
	}
	/* Insert new pair */
	_queue[i + 1] = [item, priority];

	return true;
}

function PriorityQueue::Pop()
{
	if (_count == 0) return null;

	local node = _queue.pop();
	_count--;

	return node[0];
}

function PriorityQueue::Peek()
{
	if (_count == 0) return null;

	return _queue[_count - 1][0];
}

function PriorityQueue::Count()
{
	return _count;
}

function PriorityQueue::Exists(item)
{
	/* Brute-force find the item (there is no faster way, as we don't have the priority number) */
	foreach (node in _queue) {
		if (node[0] == item) return true;
	}

	return false;
}