add priority queue
This commit is contained in:
parent
fcbf673dbe
commit
f904019d40
51
src/misc/priority_queue.js
Normal file
51
src/misc/priority_queue.js
Normal file
@ -0,0 +1,51 @@
|
||||
export class PriorityQueue {
|
||||
constructor() {
|
||||
this.val = [null];
|
||||
}
|
||||
|
||||
push(value, priority) {
|
||||
if (priority == null) {
|
||||
priority = value;
|
||||
}
|
||||
|
||||
this.val.push([value, priority]);
|
||||
let id = this.val.length - 1;
|
||||
while (id > 1 && this.val[id][1] > this.val[id >>> 1][1]) {
|
||||
const kid = id >>> 1;
|
||||
[this.val[id], this.val[kid]] = [this.val[kid], this.val[id]];
|
||||
id = kid;
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
pop() {
|
||||
let lv = this.val.pop();
|
||||
if (this.val.length == 1) { return lv; }
|
||||
|
||||
let res = this.val[1];
|
||||
this.val[1] = lv;
|
||||
let id = 1;
|
||||
while (id * 2 < this.val.length) {
|
||||
if (this.val[id][1] > this.val[id * 2][1] && (id * 2 + 1 >= this.val.length || this.val[id][1] > this.val[id * 2 + 1][1])) {
|
||||
break;
|
||||
}
|
||||
|
||||
let kid = id * 2 + 1 >= this.val.length || this.val[id * 2][1] > this.val[id * 2 + 1][1] ? id * 2 : id * 2 + 1;
|
||||
[this.val[id], this.val[kid]] = [this.val[kid], this.val[id]];
|
||||
id = kid;
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
top() {
|
||||
return this.val[1];
|
||||
}
|
||||
|
||||
size() {
|
||||
return this.val.length - 1;
|
||||
}
|
||||
|
||||
empty() {
|
||||
return this.size() == 0;
|
||||
}
|
||||
}
|
18
tests/priority_queue.test.js
Normal file
18
tests/priority_queue.test.js
Normal file
@ -0,0 +1,18 @@
|
||||
import assert from 'node:assert';
|
||||
import { PriorityQueue } from '../src/misc/priority_queue.js';
|
||||
|
||||
describe('PriorityQueue', function () {
|
||||
it('everything', function() {
|
||||
let q = new PriorityQueue();
|
||||
q.push(1).push(2).push(-1).push('hi', 4);
|
||||
assert.equal(q.size(), 4);
|
||||
assert.equal(q.top()[0], 'hi');
|
||||
|
||||
assert.equal(q.pop()[0], 'hi');
|
||||
assert.equal(q.pop()[0], 2);
|
||||
assert.equal(q.pop()[0], 1);
|
||||
assert.equal(q.pop()[0], -1);
|
||||
|
||||
assert.ok(q.empty());
|
||||
});
|
||||
});
|
Loading…
x
Reference in New Issue
Block a user