Search code examples
c#data-structuresred-black-tree

Red-Black Tree Insertion Fixup Error


I have a homework problem about inserting into red-black trees in c#. I wrote the code below and the program works without any problems for adding first 3 numbers. When I try to add the 4th number, I get a NullReferenceException. I'm trying to solve the error for 2 days but I can't figure it out.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace Algoritmalar3b
{
class rbtNode
{
    public int? key;
    public char? color;
    public rbtNode left, right, p;
    public rbtNode(int? key, char? color, rbtNode left, rbtNode right, rbtNode p)
    {
        this.key = key;
        this.color = color;
        this.left = left;
        this.right = right;
        this.p = p;
    }
}

class Program
{
    static rbtNode Tnil = new rbtNode(null, 'B', null, null, null);

    static void Main(string[] args)
    {
        rbtNode root = new rbtNode(null, 'B', Tnil, Tnil, null);

        RB_Insert(root, 7);
        RB_Insert(root, 3);
        RB_Insert(root, 89);
        RB_Insert(root, 4);
        RB_Insert(root, 9);
        RB_Insert(root, 15);
        RB_Insert(root, 35);
        RB_Insert(root, 8);
        RB_Insert(root, 24); 
        preOrderWalk(root);
    }

    static void RB_Insert(rbtNode T, int? deger)
    {
        rbtNode z = new rbtNode(deger, null, Tnil, Tnil, null);

        if (T.key == null)
            T.key = deger;
        else
        {
            rbtNode y = new rbtNode(null, null, Tnil, Tnil, null);
            y = Tnil;
            rbtNode x = new rbtNode(null, null, Tnil, Tnil, null);
            x = T;
            while (x != Tnil)
            {
                y = x;
                if (z.key < x.key)
                    x = x.left;
                else
                    x = x.right;
            }
            z.p = y;
            if (y == Tnil)
                T = z;
            else if (z.key < y.key)
                y.left = z;
            else
                y.right = z;
            z.left = Tnil;
            z.right = Tnil;
            z.color = 'R';
            RB_Insert_Fixup(T, z);
        }
    }

    static void RB_Insert_Fixup(rbtNode T, rbtNode z)
    {
        rbtNode y = new rbtNode(null, null, Tnil, Tnil, null);

        while (z.p.color == 'R')
        {
            if (z.p == z.p.p.left)
            {
                y = z.p.p.right;
                if (y.color == 'R')
                {
                    z.p.color = 'B';
                    y.color = 'B';
                    z.p.p.color = 'R';
                    z = z.p.p;
                }
                else if (z == z.p.right)
                {
                    z = z.p;
                    Left_Rotate(T, z);
                    z.p.color = 'B';
                    z.p.p.color = 'R';
                    Right_Rotate(T, z.p.p);
                }
            }
            else
            {
                y = z.p.p.left;
                if (y.color == 'R')
                {
                    z.p.color = 'B';
                    y.color = 'B';
                    z.p.p.color = 'R';
                    z = z.p.p;
                }
                else if (z == z.p.left)
                {
                    z = z.p;
                    Left_Rotate(T, z);
                    z.p.color = 'B';
                    z.p.p.color = 'R';
                    Right_Rotate(T, z.p.p);
                }
            }
        }
        T.color = 'B';
    }

    static void Left_Rotate(rbtNode T, rbtNode x)
    {
        rbtNode y = new rbtNode(null, null, Tnil, Tnil, null);
        y = x.right;
        x.right = y.left;
        if (y.left != Tnil)
            y.left.p = x;
        y.p = x.p;
        if (x.p == Tnil)
            T = y;
        else if (x == x.p.left)
            x.p.left = y;
        else
            x.p.right = y;
        y.left = x;
        x.p = y; 
    }

    static void Right_Rotate(rbtNode T, rbtNode x)
    {
        rbtNode y = new rbtNode(null, null, Tnil, Tnil, null);
        y = x.left;
        x.left = y.right;
        if (y.right != null)
            y.right.p = x;
        y.p = x.p;
        if (x.p == null)
            T = y;
        else if (x == x.p.right)
            x.p.right = y;
        else
            x.p.left = y;
        y.right = x;
        x.p = y;
    }

    static void preOrderWalk(rbtNode T)
    {
        if (T.color == 'B')
            Console.WriteLine("-{0}", T.key);
        else
            Console.WriteLine("{0}", T.key);
        if (T.left != null)
            preOrderWalk(T.left);
        if (T.right != null)
            preOrderWalk(T.right);
    }
}

}


Solution

  • It seems you have problems in 3 areas:

    • The RB tree algorithm. But it looks like you're almost there
    • Debugging. It should only take 2 minutes to backtrack an error like this, maybe 2 hours if you're new. But not 2 days.
    • Asking a good question. You have a nullref, but on what line? and what are the values of the vars/fields on that line?

    When I take a quick look at the Fixup method and the rest of the code I suspect you may have a p (Parent) ref being null at the wrong point.

    As a diagnosis tool, change the .p member in a regular property:

    class rbtNode
    {    
        private rbtNode _parent;
    
        public rbtNode Parent
        {
             get { return _parent; }
             set
             {
                 System.Diagnostics.Debug.Assert(value != null);
                 _parent = value;
             }
        }
        ....
    

    I think you will have to allow _parent=null but only in the constructor.

    The get/set members also give you a good spot to place (conditional) breakpoints.