using System;

using System.Collections.Generic;

​

​

namespace Grasshopper

{

    public class Program

    {

        public static void Main(string[] args)

        {

​

            // bug inthe code tex (10,4) should be fixed

            //Grasshopper g = new Grasshopper(5,4);

            //

            Grasshopper g = new Grasshopper(5,4);

            Console.WriteLine(g.WhereAmI());

​

            g.EatAndHopRight();

            Console.WriteLine(g.WhereAmI());

​

            g.EatAndHopLeft();

            Console.WriteLine(g.WhereAmI());

​

            Console.ReadLine();

        }   

    }

​

    //2->3->4->5->1

    public class Grasshopper

    {

        public int Root { get; set; }

        public int Nodes { get; set; }

        public List<int> LeavesList = new List<int>();

        public Grasshopper(int node, int root)

        {

            this.Root = root;

            this.Nodes = node;

        }

        public void EatAndHopLeft()

        {

            EatAndHopRight();

        }

​

        public void EatAndHopRight()

        {

            // Works for all numbers under 5

            // TODO fix all number > 5

            if(Nodes <=5 && LeavesList.Count >=3 ) // Numbers under 5

            {

                for(int i = LeavesList.IndexOf(Root); i <= LeavesList.Count;)

                {

                    LeavesList.RemoveAt(i);

​

                    if (++i < 3 && LeavesList.Count > 2) // used 2 for root and one node should be at least 3 leaves

                    {

                        Root = LeavesList[i];

                    }

                    else if(i > LeavesList.Count && LeavesList.Count > 2)

                    {

                        Root = LeavesList[1]; // move to index 1 from the list when number of node is same of number of root ex(5,5)

                    }

                    else

                    {

                        Root = LeavesList[0];  //-> point the end of the list to the start

                    }

                    break;

                }

            }

        }

​

        // Method to genereate the leaves

        // and add it to list

        public void GenerateLeaves()

        {

            for (int i = 1; i <= Nodes; i++)

            {

                LeavesList.Add(i);

            }

        }

        public int WhereAmI()

        {

            if(LeavesList.Count != 0)

            {

                return Root;

            }

            else

            {

                GenerateLeaves();

            }

            return Root;

        }

    }

}

​

​