mirror of
https://github.moeyy.xyz/https://github.com/trekhleb/javascript-algorithms.git
synced 2024-12-26 07:01:18 +08:00
Add uncle property to binary tree node.
This commit is contained in:
parent
e6de25ecb3
commit
02299b77e6
@ -1,4 +1,5 @@
|
||||
import Comparator from '../../utils/comparator/Comparator';
|
||||
import HashTable from '../hash-table/HashTable';
|
||||
|
||||
export default class BinaryTreeNode {
|
||||
/**
|
||||
@ -11,7 +12,7 @@ export default class BinaryTreeNode {
|
||||
this.value = value;
|
||||
|
||||
// Any node related meta information may be stored here.
|
||||
this.meta = new Map();
|
||||
this.meta = new HashTable();
|
||||
|
||||
// This comparator is used to compare binary tree nodes with each other.
|
||||
this.nodeComparator = new Comparator();
|
||||
@ -53,6 +54,37 @@ export default class BinaryTreeNode {
|
||||
return this.leftHeight - this.rightHeight;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get parent's sibling if it exists.
|
||||
* @return {BinaryTreeNode}
|
||||
*/
|
||||
get uncle() {
|
||||
// Check if current node has parent.
|
||||
if (!this.parent) {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
// Check if current node has grand-parent.
|
||||
if (!this.parent.parent) {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
// Check if grand-parent has more than two children.
|
||||
if (!this.parent.parent.left || !this.parent.parent.right) {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
// So for now we know that current node has grand-parent and this
|
||||
// grand-parent has two children. Let's find out who is the uncle.
|
||||
if (this.nodeComparator.equal(this.parent, this.parent.parent.left)) {
|
||||
// Right one is an uncle.
|
||||
return this.parent.parent.right;
|
||||
}
|
||||
|
||||
// Left one is an uncle.
|
||||
return this.parent.parent.left;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {BinaryTreeNode} node
|
||||
* @return {BinaryTreeNode}
|
||||
|
@ -207,4 +207,54 @@ describe('BinaryTreeNode', () => {
|
||||
expect(redNode.meta.get('color')).toBe('red');
|
||||
expect(blackNode.meta.get('color')).toBe('black');
|
||||
});
|
||||
|
||||
it('should detect right uncle', () => {
|
||||
const grandParent = new BinaryTreeNode('grand-parent');
|
||||
const parent = new BinaryTreeNode('parent');
|
||||
const uncle = new BinaryTreeNode('uncle');
|
||||
const child = new BinaryTreeNode('child');
|
||||
|
||||
expect(grandParent.uncle).not.toBeDefined();
|
||||
expect(parent.uncle).not.toBeDefined();
|
||||
|
||||
grandParent.setLeft(parent);
|
||||
|
||||
expect(parent.uncle).not.toBeDefined();
|
||||
expect(child.uncle).not.toBeDefined();
|
||||
|
||||
parent.setLeft(child);
|
||||
|
||||
expect(child.uncle).not.toBeDefined();
|
||||
|
||||
grandParent.setRight(uncle);
|
||||
|
||||
expect(parent.uncle).not.toBeDefined();
|
||||
expect(child.uncle).toBeDefined();
|
||||
expect(child.uncle).toEqual(uncle);
|
||||
});
|
||||
|
||||
it('should detect left uncle', () => {
|
||||
const grandParent = new BinaryTreeNode('grand-parent');
|
||||
const parent = new BinaryTreeNode('parent');
|
||||
const uncle = new BinaryTreeNode('uncle');
|
||||
const child = new BinaryTreeNode('child');
|
||||
|
||||
expect(grandParent.uncle).not.toBeDefined();
|
||||
expect(parent.uncle).not.toBeDefined();
|
||||
|
||||
grandParent.setRight(parent);
|
||||
|
||||
expect(parent.uncle).not.toBeDefined();
|
||||
expect(child.uncle).not.toBeDefined();
|
||||
|
||||
parent.setRight(child);
|
||||
|
||||
expect(child.uncle).not.toBeDefined();
|
||||
|
||||
grandParent.setLeft(uncle);
|
||||
|
||||
expect(parent.uncle).not.toBeDefined();
|
||||
expect(child.uncle).toBeDefined();
|
||||
expect(child.uncle).toEqual(uncle);
|
||||
});
|
||||
});
|
||||
|
Loading…
Reference in New Issue
Block a user