[C#] 백준 18258번 큐 2

[ 문제 ]

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

 

[ 코드 ]

1. 맞은 코드

using System.Text;

StreamReader sr = new StreamReader(new BufferedStream(Console.OpenStandardInput()));
StreamWriter sw = new StreamWriter(new BufferedStream(Console.OpenStandardOutput()));
StringBuilder sb = new StringBuilder();

int n = int.Parse(sr.ReadLine());
int[] q = new int[n];
int back = 0 , front = 0;

for (int i = 0; i < n; i++)
{
    string[] s = sr.ReadLine().Split();
    var a = s[0];

    switch (a)
    {
        case "push":
            int b = int.Parse(s[1]);
            q[back++] = b;
            break;
        case "pop":
            sb.AppendLine($"{(front == back ? -1 : q[front])}");
            if(front != back) front++;
            break;
        case "size":
            sb.AppendLine($"{back - front}");
            break;
        case "empty":
            sb.AppendLine($"{(front == back ? 1 : 0)}");
            break;
        case "front":
            sb.AppendLine($"{(front == back ? -1 : q[front])}");
            break;
        case "back":
            sb.AppendLine($"{(front == back ? -1 : q[back - 1])}");
            break;
    }
}

sw.Write(sb.ToString());
sw.Flush();
sw.Close();
sr.Close();

 

2. 시간초과 코드 2

- 큐 문제인데 큐의 Last는 사용하면 시간 초과 납니다용

해결 방법은 push할때 d의 값이 Last값이니깐 그걸로 출력해야합니다. 

using System.Text;

StreamReader sr = new StreamReader(new BufferedStream(Console.OpenStandardInput()));
StreamWriter sw = new StreamWriter(new BufferedStream(Console.OpenStandardOutput()));
StringBuilder sb = new StringBuilder();

int n = int.Parse(Console.ReadLine());
Queue<int> q = new Queue<int>();

for (int i = 0; i < n; i++)
{
    string[] s = Console.ReadLine().Split();
    var a = s[0];
    int b = 0;

    if (s.Length >= 2)
    {
        b = int.Parse(s[1]);
    }

    if (a == "push")
    {
        q.Enqueue(b);
    }
    else if(a == "pop")
    {
        sb.AppendLine($"{(q.Count == 0 ? -1 : q.Dequeue())}");
    }
    else if (a == "size")
    {
        sb.AppendLine($"{q.Count}");
    }
    else if (a == "empty")
    {
        sb.AppendLine($"{(q.Count == 0 ? 1 : 0)}");
    }
    else if (a == "front")
    {
        sb.AppendLine($"{(q.Count == 0 ? -1 : q.Peek())}");
    }
    else if (a == "back")
    {
        sb.AppendLine($"{(q.Count == 0 ? -1 : q.Last())}");
    }
}

sw.Write( sb.ToString() );
sw.Flush();
sw.Close();
sr.Close();

 

- LinkedList도 마찬가지... 일까요...?

using System.Text;

StreamReader sr = new StreamReader(new BufferedStream(Console.OpenStandardInput()));
StreamWriter sw = new StreamWriter(new BufferedStream(Console.OpenStandardOutput()));
StringBuilder sb = new StringBuilder();

int n = int.Parse(sr.ReadLine());
LinkedList<int> q = new LinkedList<int>();

for (int i = 0; i < n; i++)
{
    string[] s = sr.ReadLine().Split();
    var a = s[0];
    int b = 0;

    if (s.Length >= 2)
    {
        b = int.Parse(s[1]);
    }

    if (a == "push")
    {
        q.AddLast(b);
    }
    else if (a == "pop")
    {
        sb.AppendLine($"{(q.Count == 0 ? -1 : q.First.Value)}");
        if (q.Count > 0) q.RemoveFirst();
    }
    else if (a == "size")
    {
        sb.AppendLine($"{q.Count}");
    }
    else if (a == "empty")
    {
        sb.AppendLine($"{(q.Count == 0 ? 1 : 0)}");
    }
    else if (a == "front")
    {
        sb.AppendLine($"{(q.Count == 0 ? -1 : q.First.Value)}");
    }
    else if (a == "back")
    {
        sb.AppendLine($"{(q.Count == 0 ? -1 : q.Last())}");
    }
}

sw.Write(sb.ToString());
sw.Flush();
sw.Close();
sr.Close();

 

 

[ 실행화면 ]

case: 1

 

 

 

 

 

---

문제링크: https://www.acmicpc.net/problem/18258

18258번: 큐 2