ツリーのバランス部分に問題が発生しました。再帰的な挿入の後にcheckBalが呼び出されました。5、2、4を追加しようとすると、2のバランスがチェックされ、5まで戻り、右回転の左の部分に移動します。これは正しいことです。しかし、rotateLeft関数は2行目でエラーになります。
この実装の何が問題になっていますか?私はあちこちを検索し、私がしたことを人々がそれがどのように行われたかについて話す方法と比較しました。私はついにすべてが機能するようになりました。最後にNをKに設定するのを忘れていました。
//==============================================================================
//===== Set Balance ============================================================
sNode<T>* checkBal(sNode<T> *locRoot)
{
// Make sure to check the children are balanced as well.
if (locRoot->left != NULL)
locRoot->left = checkBal(locRoot->left);
if (locRoot->right != NULL)
locRoot->right = checkBal(locRoot->right);
if(getHeight(locRoot->left) - getHeight(locRoot->right) > 1)
{
if(getHeight(locRoot->left->right) > getHeight(locRoot->left->left))
locRoot->left = rotateRight(locRoot->left);
locRoot = rotateLeft(locRoot);
}
else if(getHeight(locRoot->right) - getHeight(locRoot->left) > 1)
{
if(getHeight(locRoot->right->left) > getHeight(locRoot->right->right))
locRoot->right = rotateLeft(locRoot->right);
locRoot = rotateRight(locRoot);
}
updateHeights(locRoot);
return locRoot;
}
/*
Extream cases of balancing a tree requires a double rotation
A
\
D
/
B
'A' is the current root
If right->left (grandchild) is larger then the right->right (grandchild)
First Right rotate the child then left rotate the parent
left > right by 2 or more
left.left < left.right (Double Right Rotation)
left.left >= left.right (Single Right Rotation)
right > left by 2 or more
right.right < right.left (Double Left Rotation)
right.right >= right.left (Single Left Rotation)
*/
sNode<T>* rotateRight(sNode<T> *N) const
{
/*
N K
/ \ / \
(a) K => N (c)
/ \ / \
(b) (c) (a) (b)
*/
// K is going to be our new Parent
// Move (c) from K->right to N->left
// Set K->right to be N
// Return the new parent node to update the one above.
sNode<T> *K = N->right;
N->right = K->left;
K->left = N;
return N = K;
}