1-xxubin04

[xxubin04]

🔗 문제 링크

https://www.acmicpc.net/problem/28278

✔️ 소요된 시간

5분

✨ 수도 코드

• 문제를 읽고 이해하면 쉽게 구현할 수 있는 문제이다. 2번에서 헷갈릴 수 있으니 예제를 잘 보자.
• 먼저, 반복할 횟수를 입력받고 for문을 만든다.

    import sys

    n = int(sys.stdin.readline())
    stack = []

    for i in range(n):
        num = sys.stdin.readline().split()

• 명령 5가지를 충족하는 if문을 작성한다.

    if num[0] == "1":
        stack.append(num[-1])

    elif num[0] == "2":
        if len(stack) != 0:
            print(stack.pop(-1))
        else:
            print(-1)

    elif num[0] == "3":
        print(len(stack))

    elif num[0] == "4":
        if len(stack) == 0:
            print(1)
        else:
            print(0)

    elif num[0] == "5":
        if len(stack) != 0:
            print(stack[-1])
        else:
            print(-1)

• 1부터 5를 제외한 숫자가 num에 입력될 수 있으니 다음과 같이 지정한다.

    else:
        break

• 전체 코드는 아래와 같다.

  import sys
  
  n = int(sys.stdin.readline())
  stack = []
  
  for i in range(n):
      num = sys.stdin.readline().split()
  
      if num[0] == "1":
          stack.append(num[-1])
  
      elif num[0] == "2":
          if len(stack) != 0:
              print(stack.pop(-1))
          else:
              print(-1)
  
      elif num[0] == "3":
          print(len(stack))
  
      elif num[0] == "4":
          if len(stack) == 0:
              print(1)
          else:
              print(0)
  
      elif num[0] == "5":
          if len(stack) != 0:
              print(stack[-1])
          else:
              print(-1)
      else:
          break

📚 새롭게 알게된 내용

• 계속 시간초과가 되어서 왜인지 찾아보았다.
  input()을 이용해서 반복문으로 여러 줄을 입력받으면 그러하다고 한다.
  따라서, 반복문 + 여러 줄 입력 -> sys.stdin.readline()
  (https://velog.io/@yeseolee/Python-%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EC%9E%85%EB%A0%A5-%EC%A0%95%EB%A6%ACsys.stdin.readline)

[9kyo-hwang]

고생 많았습니다!
자료구조 수업을 아직 수강하지 않아서 스택같은 특수한 자료구조를 사용하는 문제를 풀기 쉽지 않았을 텐데, 이미 1학년에서 상위 10%에 속하네요 😎

[9kyo-hwang]

• sys.stdin 이 친구는 python에서 "빠른 입력"을 받기 위해 사용됩니다. 그렇기에 readline()을 사용해서 한 번에 입력을 받을 필요까지는 없어보입니다.
• 제가 확인해본 결과, 빠른 입력을 사용해주기만 하면 숫자를 하나씩 따로 입력받더라도 무난히 통과하네요.

ios_base::sync_with_stdio(false);  //  cpp에서 빠른 입력을 받기 위한 코드입니다.
cin.tie(nullptr);
cout.tie(nullptr);

int N; cin >> N;
vector<int> stack;
for(int i = 0; i < N; i++) {
  int n; cin >> n;
  ... // 로직 수행
}

[xxubin04]

오호 그렇군요! 좋은 말씀 감사합니다~😊👍

[Munbin-Lee]

아니요. 파이썬에서는 하나씩 입력받는 기능이 없습니다. 😥

[Dolchae]

수고하셨습니다!! 시간 초과 때문에도 고생 많이 하셨어요😅☺️

[Munbin-Lee]

if len(stack) != 0 을 if stack으로 줄일 수 있어요.

[xxubin04]

앗 그렇군요!! 감사합니다😊
다음부터는 그렇게 쓰도록 하겠습니다!