首页 > 代码库 > 自己写一个泛型集合类型,可实现添加和遍历

自己写一个泛型集合类型,可实现添加和遍历


在"C#中List<T>是怎么存放元素的"中,分析了List<T>的源码,了解了List<T>是如何存放元素的。这次,就自定义一个泛型集合类型,可实现添加元素,并支持遍历。

 

该泛型集合类型一定需要一个添加元素的方法,在添加元素的时候需要考虑:当添加的元素超过当前数组的容量,就让数组扩容;为了支持循环遍历,该泛型集合类型必须提供一个迭代器(实现IEnumerator接口)。

    public class MyList<T>
    {
        T[] items = new T[5];
        private int count;
        public void Add(T item)
        {
            if(count == items.Length)
               Array.Resize(ref  items, items.Length * 2);
            items[count++] = item;
        }
        public IEnumerator<T> GetEnumerator()
        {
            return new MyEnumeraor(this);
        }
        class MyEnumeraor : IEnumerator<T>
        {
            private int index = -1;
            private MyList<T> _myList;
            public MyEnumeraor(MyList<T> myList)
            {
                _myList = myList;
            }
            public T Current
            {
                get
                {
                    if (index < 0 || index >= _myList.count)
                    {
                        return default(T);
                    }
                    return _myList.items[index];
                }
            }
            public void Dispose()
            {
                
            }
            object System.Collections.IEnumerator.Current
            {
                get { return Current; }
            }
            public bool MoveNext()
            {
                return ++index < _myList.count;
            }
            public void Reset()
            {
                index = -1;
            }
        }
    }

○ 泛型集合类型维护着一个T类型的泛型数组
○ 私有字段count是用来计数的,每添加一个元素计数加1
○ 添加方法考虑了当count计数等于当前元素的长度,就让数组扩容为当前的2倍
○ 迭代器实现了IEnumerator<T>接口

 

客户端调用。

    class Program
    {
        static void Main(string[] args)
        {
            MyList<int> list = new MyList<int>();
            list.Add(1);
            list.Add(2);
            foreach (int item in list)
            {
                Console.WriteLine(item);
            }
            Console.ReadKey();
        }
    }

 

另外,IEnumerable和IEnumerator的区别是什么呢?
其实,真正执行迭代的是IEnumerator迭代器。IEnumerable接口就提供了一个方法,就是返回IEnumerator迭代器。

public interface IEnumerable
{
    IEnumerator GetEnumerator();
}

自己写一个泛型集合类型,可实现添加和遍历