Iterator своего класса

@piv199 · Регистрация: 21.12.2012

Author24 — интернет-сервис помощи студентам

Не могу понять как сделать генератор своего класса что бы потом использовать в foreach. Максимум получалось возвращать 1 элемент.

Вот мой нерабочий примерный итератор...

C#

    public class HorizontalGenerator 
    {
        private Queue queue;
        private RedBlackTree tree;
        private RedBlackNode deq;
        private RedBlackNode sentinelNode;
 
        public HorizontalGenerator(RedBlackTree itertree)
        {
            tree = itertree;
            sentinelNode = tree.SentinelNode;
            queue = new Queue();
            queue.Enqueue(tree.Root);
            deq = tree.Root;
        }
 
        public bool HasNext()
        {
            if (queue.Count > 0) return true;
            else return false;
        }
 
        public IEnumerator<RedBlackNode> NextElement()
        {
            if (HasNext()) yield return deq;
            if (deq.Left != sentinelNode) queue.Enqueue(deq.Left);
            if (deq.Right != sentinelNode) queue.Enqueue(deq.Right);
            if (HasNext()) deq = (RedBlackNode)queue.Dequeue();
        }
    }

Вот что я хочу получить

C#

            HorizontalGenerator G = new HorizontalGenerator(tree);
            foreach (RedBlackNode iter in G)
            {
                //dosemething();
            }

@kolorotur · 01.06.2014, 12:11

В данный момент ваш класс работает так:

C#

1	foreach (RedBlackNode iter in G.NextElement())

Измените его на такой:

C#

public class HorizontalGenerator : IEnumerable<RedBlackNode>
{
   ...
   public IEnumerator<RedBlackNode> GetEnumerator()
   {
            if (HasNext()) yield return deq;
            if (deq.Left != sentinelNode) queue.Enqueue(deq.Left);
            if (deq.Right != sentinelNode) queue.Enqueue(deq.Right);
            if (HasNext()) deq = (RedBlackNode)queue.Dequeue();
   }
   IEnumerator IEnumerable.GetEnumerator()
   {
      return GetEnumerator();
   }
}

@piv199 · 01.06.2014, 13:10 **[ТС]**

kolorotur, Спасибо помогли, но чуть-чуть не так

но то мелочи

Немного подправил и вот что получилось

C#

    public class HorizontalGenerator: IEnumerable<RedBlackNode> 
    {
        private Queue queue;
        private RedBlackTree tree;
        private RedBlackNode deq;
        private RedBlackNode sentinelNode;
 
        public HorizontalGenerator(RedBlackTree itertree)
        {
            tree = itertree;
            sentinelNode = tree.SentinelNode;
            queue = new Queue();
            queue.Enqueue(tree.Root);
            deq = tree.Root;
        }
        private bool HasNext
        {
            get
            {
                if (queue.Count > 0) return true;
                else return false;
            }
 
        }
        public IEnumerator<RedBlackNode> GetEnumerator()
        {
            while (HasNext)
            {
                deq = (RedBlackNode)queue.Dequeue();
                yield return deq;
                if (deq.Left != sentinelNode) queue.Enqueue(deq.Left);
                if (deq.Right != sentinelNode) queue.Enqueue(deq.Right);
                
            }
        }
        IEnumerator IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }
    }

	01.06.2014, 13:10

Опции темы