[Codility] Lesson7. Fish (C#)

문제: app.codility.com/programmers/lessons/7-stacks_and_queues/fish/

Fish coding task - Learn to Code - Codility

 

 

            Stack<int> up = new Stack<int>();
            Stack<int> down = new Stack<int>();

            for (int i = 0; i < A.Length; i++)
            {
                if (B[i] == 0)
                {
                    if (down.Count == 0)
                        up.Push(A[i]);
                    else
                    {
                        while (down.Count > 0)
                        {
                            if (down.Peek() < A[i])
                            {
                                down.Pop();

                                if (down.Count == 0)
                                {
                                    up.Push(A[i]);
                                    break;
                                }
                            }
                            else
                                break;
                        }
                    }
                }
                else
                {
                    down.Push(A[i]);
                }
            }

            return up.Count + down.Count;