TonBERRY/micropython
3
0
Fork 0
Code Pull Requests Activity

py/pairheap: Properly unlink node on pop and delete.

Browse Source 
This fixes a bug in the pairing-heap implementation when nodes are deleted
with mp_pairheap_delete and then reinserted later on.
This commit is contained in:
Damien George
2020-03-13 14:35:03 +11:00
parent 98ab7643a7
commit c47a3ddf4a
2 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
py/pairheap.c
View File

@@ -72,9 +72,11 @@ mp_pairheap_t *mp_pairheap_pairing(mp_pairheap_lt_t lt, mp_pairheap_t *child) {
while (!NEXT_IS_RIGHTMOST_PARENT(child)) {
mp_pairheap_t *n1 = child;
child = child->next;
n1->next = NULL;
if (!NEXT_IS_RIGHTMOST_PARENT(child)) {
mp_pairheap_t *n2 = child;
child = child->next;
n2->next = NULL;
n1 = mp_pairheap_meld(lt, n1, n2);
}
heap = mp_pairheap_meld(lt, heap, n1);
@@ -87,7 +89,9 @@ mp_pairheap_t *mp_pairheap_pairing(mp_pairheap_lt_t lt, mp_pairheap_t *child) {
mp_pairheap_t *mp_pairheap_delete(mp_pairheap_lt_t lt, mp_pairheap_t *heap, mp_pairheap_t *node) {
// Simple case of the top being the node to delete
if (node == heap) {
return mp_pairheap_pairing(lt, heap->child);
mp_pairheap_t *child = heap->child;
node->child = NULL;
return mp_pairheap_pairing(lt, child);
}
// Case where node is not in the heap
@@ -113,18 +117,22 @@ mp_pairheap_t *mp_pairheap_delete(mp_pairheap_lt_t lt, mp_pairheap_t *heap, mp_p
node->next = NULL;
return heap;
} else if (node == parent->child) {
mp_pairheap_t *child = node->child;
next = node->next;
node->child = NULL;
node->next = NULL;
node = mp_pairheap_pairing(lt, node->child);
node = mp_pairheap_pairing(lt, child);
parent->child = node;
} else {
mp_pairheap_t *n = parent->child;
while (node != n->next) {
n = n->next;
}
mp_pairheap_t *child = node->child;
next = node->next;
node->child = NULL;
node->next = NULL;
node = mp_pairheap_pairing(lt, node->child);
node = mp_pairheap_pairing(lt, child);
if (node == NULL) {
node = n;
} else {