基于二叉查找树的集合

     我们都知道Dictionary<TKey, TValue>查找元素非常快,其实现原理是:将你TKey的值散列到数组的指定位置,将TValue的值存入对应的位置,
 由于取和存用的是同一个算法,所以就很容易定位到TValue的位置,花费的时间基本上就是实现散列算法的时间,跟其中元素的个数没有关系,
 故取值的时间复杂度为O(1)。
     集合无非都是基于最基础语法的数组[],先欲分配,然后向其中添加元素,容量不够就创建一个2倍容量的数组,将之前的元素赋值过来,将之前的数组回收,
 但基于散列算法的集合这点上有点不同,他并不是每次创建一个2倍容量的数组,为了让元素均匀的分布到数组上,数组的容量是这么增长的:3,7,11,17,23,29,37,47,59,71,89,107,131,163,197,239,293,353,431,521,631,761,919,1103...
 以质数的方式增长。由于每次扩充数组的容量较小,如果要向其中添加很多元素的话,程序员又没有预先分配内存,那就会出现多次数组的创建、复制和回收。
     一直想做个有用的东西出来,让想用的人用,又能让自己练练手,于是这次做了一个基于二叉查找树的集合,我们知道在二叉查找树中查询元素的最优时间复杂度是O(logN)即在满二叉树的情况下,最坏时间复杂度是O(n)即除叶子节点外每个节点只有一个子节点,
 查找元素它也是很快的哦,而且查找的时候都只是做Int型的比较,而Dictionary<TKey, TValue>是基于一个散列算法,当然基于二插查找树的集合也有自身的缺点:
     1:元素必须实现接口IBinaryTree,其属性CompareValue主要用于比较生成二叉查找树
     2:元素必须是可以new的,即不支持基础类型int,char,string等
     3:每个节点都有左右两个子节点,他们只是起到指针的作用,指向该节点的子节点,只需占用额外的少量内存
     4:基本上都是基于递归实现,元素过多的话,会栈溢出,至于原因请看我的这篇博客
    优点是常用的一些功能都有,功能如下,练手吗,但会一直优化下去

    public class BinaryTree<T> : IDisposable, IEnumerable<T>, IEnumerable where T :IBinaryTree, new()
{
public BinaryTree();
public BinaryTree(IEnumerable<T> list);//将一个数组构造成二插查找树
public BinaryTree(T root); //指定跟节点
public int Count { get; }//元素个数
public T this[IBinaryTree iBinaryTree] { get; }//数组索引直接访问元素
public void Add(T t);//添加元素
public void Clear();//清除所有元素
public bool Contains(T iBinaryTree);//是否包含自定元素
public void Dispose();//释放资源,支持using
public T Find(IBinaryTree iBinaryTree);//查找元素
public T Find(IBinaryTree iBinaryTree, TreeNode<T> node);//在指定的节点下查找元素
public T FindMax();//最大元素
public T FindMin();//最小元素
public T FindMin(TreeNode<T> node);//在指定的节点下查找最小元素
public IEnumerator<T> GetEnumerator();//返回所有元素,支持foreach
public TreeNode<T> Remove(T t);//删除元素
public TreeNode<T> Remove(IBinaryTree iBinaryTree, TreeNode<T> node);//在指定的节点下删除元素
public IEnumerable<T> Sort();//排序(升序)
public IEnumerable<T> ToList();//返回所有元素
}

 

源码
namespace Utils
{
/// <summary>
/// 二叉树接口
/// </summary>
public interface IBinaryTree
{
/// <summary>
/// 用于比较的值
/// </summary>
int CompareValue
{
get;
set;
}
}

public class TreeNode<T> where T : IBinaryTree, new()
{
public TreeNode<T> Left
{
get;
set;
}

public TreeNode<T> Right
{
set;
get;
}

public T Data
{
get;
set;
}

public TreeNode(T t)
{
this.Data = t;
}

public TreeNode()
: this(default(T))
{

}
}

/// <summary>
/// 二插查找树
/// </summary>
public class BinaryTree<T> : IDisposable,IEnumerable<T> where T : IBinaryTree, new()
{
public BinaryTree()
{

}

public BinaryTree(T root)
{
if (root == null)
{
throw new NullReferenceException("Parameter is null");
}
Add(root);
}

public BinaryTree(IEnumerable<T> list)
{
if (list == null)
{
throw new NullReferenceException("Parameter is null");
}
foreach (var item in list)
{
Add(item);
}
}

//根节点
private TreeNode<T> root;

//添加节点(没有检查根节点是否为空,所以设为private)
private void Add(T t, TreeNode<T> root)
{
if (t == null)
{
return;
}
if (t.CompareValue < root.Data.CompareValue)
{
if (root.Left == null)
{
root.Left = new TreeNode<T>(t);
++Count;
}
else
{
Add(t, root.Left);
}
}
else if (t.CompareValue > root.Data.CompareValue)
{
if (root.Right == null)
{
root.Right = new TreeNode<T>(t);
++Count;
}
else
{
Add(t, root.Right);
}
}
else
{
root.Data = t;
}
}

//添加节点
public void Add(T t)
{
if (t == null)
{
return;
}
if (this.root == null)
{
this.root = new TreeNode<T>(t);
++Count;
}
else
{
Add(t, this.root);
}
}

//查找指定节点下的最小节点
public T FindMin(TreeNode<T> node)
{
if (node == null)
{
return default(T);
}
if (node.Left == null)
{
return node.Data;
}
else
{
return FindMin(node.Left);
}
}

//查找最小节点
public T FindMin()
{
return FindMin(this.root);
}

//查找最大节点
private T FindMax(TreeNode<T> node)
{
if (node.Right == null)
{
return node.Data;
}
else
{
return FindMax(node.Right);
}
}

//查找最大节点
public T FindMax()
{
return FindMax(this.root);
}

//删除指定节点下的节点
public TreeNode<T> Remove(IBinaryTree iBinaryTree, TreeNode<T> node)
{
if (node == null)
{
return null;
}
if (iBinaryTree == null)
{
return node;
}
if (iBinaryTree.CompareValue < node.Data.CompareValue)
{
node.Left = Remove(iBinaryTree, node.Left);
}
else if (iBinaryTree.CompareValue > node.Data.CompareValue)
{
node.Right = Remove(iBinaryTree, node.Right);
}
else
{
if (node.Left != null && node.Right != null)
{
T tmpNode = FindMin(node.Right);//查找当前节点有子树的最小节点
node.Data.CompareValue = tmpNode.CompareValue;//将右子树的最小节点取代当前要删除的节点
node.Right = Remove(tmpNode, node.Right);//这里是亮点,删除当前节点右子树的最小节点
}
else
{
if (node.Left == null)
{
node = node.Right;
}
else if (node.Right == null)
{
node = node.Left;
}
else
{
node = null;
}
if (this.root.Data.CompareValue == iBinaryTree.CompareValue)//处理根节点
{
this.root = node;
}
}
--Count;
}
return node;
}

//删除节点
public TreeNode<T> Remove(T t)
{
if (this.root == null || t==null)
{
return null;
}
return Remove(t, this.root);
}

//查找节点
public T Find(IBinaryTree iBinaryTree, TreeNode<T> node)
{
if (node == null || iBinaryTree == null)
{
return default(T);
}
if (iBinaryTree.CompareValue < node.Data.CompareValue)
{
return Find(iBinaryTree, node.Left);
}
else if (iBinaryTree.CompareValue > node.Data.CompareValue)
{
return Find(iBinaryTree, node.Right);
}
return node.Data;
}

//查找节点
public T Find(IBinaryTree iBinaryTree)
{
return Find(iBinaryTree, this.root);
}

//是否包含指定元素
private bool Contains(IBinaryTree iBinaryTree, TreeNode<T> node)
{
if (node == null || iBinaryTree == null)
{
return false; ;
}
if (iBinaryTree.CompareValue < node.Data.CompareValue)
{
return Contains(iBinaryTree, node.Left);
}
else if (iBinaryTree.CompareValue > node.Data.CompareValue)
{
return Contains(iBinaryTree, node.Right);
}
return iBinaryTree.Equals(node.Data);
}

//是否包含指定元素
public bool Contains(T iBinaryTree)
{
return Contains(iBinaryTree, this.root);
}

//清除所有节点
public void Clear()
{
while (this.Count > 0)
{
Remove(this.root.Data);
}
this.root = new TreeNode<T>();
}

//释放资源
public void Dispose()
{
while (this.Count > 0)
{
Remove(this.root.Data);
}
this.root = null;
}

//节点个数
public int Count
{
private set;
get;
}

//转换成集合
public IEnumerable<T> ToList()
{
IList<T> list = new List<T>(Count);
LCR(this.root,list);
return list;
}

//以前序遍历的方式将节点加入集合,用递归实现,如果元素很多可能会出现栈溢出
private void LCR(TreeNode<T> node, IList<T> list)
{
if (node == null)
{
return;
}
if (node.Left != null)
{
LCR(node.Left, list);
}
list.Add(node.Data);//添加元素
if (node.Right != null)
{
LCR(node.Right, list);
}
}

//排序
public IEnumerable<T> Sort()
{
return ToList();
}

//返回一个循环访问集合的枚举数
public IEnumerator<T> GetEnumerator()
{
return this.ToList().GetEnumerator();
}

//返回一个循环访问集合的枚举数
IEnumerator IEnumerable.GetEnumerator()
{
return this.GetEnumerator();
}

public T this[IBinaryTree iBinaryTree]
{
get {
return this.Find(iBinaryTree);
}
}

}

public class Node : IBinaryTree
{
/// <summary>
/// 用于比较的值
/// </summary>
public int CompareValue
{
get;
set;
}

public string Name
{
get;
set;
}

public override string ToString()
{
return string.Format("CompareValue:{0},Name:{1}", this.CompareValue, this.Name);
}

}
}

 

作者:陈太汉

博客:http://www.cnblogs.com/hlxs/

posted @ 2012-03-15 10:17  啊汉  阅读(2114)  评论(9编辑  收藏  举报