[코딩테스트/백준/Python]수 정렬하기 2

🔍수 정렬하기 2

<문제>

N개의 수가 주어졌을 때, 이를 오름차순으로 정렬하는 프로그램을 작성하시오.

---

<입력>
첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 수가 주어진다. 이 수는 절댓값이 1,000,000보다 작거나 같은 정수이다. 수는 중복되지 않는다.

---

<출력>
첫째 줄부터 N개의 줄에 오름차순으로 정렬한 결과를 한 줄에 하나씩 출력한다.

💡생각

수 정렬하기 1 코드에서 시간을 단축시키기 위해 출력할 때는 for문을 사용하지 않고 sep 사용.

❓잘못된 코드1

문제점1. 시간초과

N = int(input())

def main(N):
    num = []
    for i in range(0, N):
        num.append(int(input()))

    num.sort()
    print(*num, sep='\n')
        
main(N)

❗해결: import sys를 해줌.

💡테스트 통과한 코드

import sys
N = int(sys.stdin.readline())

def main(N):
    num = []
    for i in range(0, N):
        num.append(int(sys.stdin.readline()))

    num.sort()
    print(*num, sep='\n')
        
main(N)

---

✅sys.stdin.readline()

• 한 두줄 입력받는 문제들과 다르게, 반복문으로 여러줄을 입력 받아야 할 때는 input()으로 입력 받는다면 시간초과가 발생할 수 있음.

• 반복문으로 여러줄 입력받는 상황에서는 반드시 sys.stdin.readline()을 사용해야 시간초과가 발생하지 않음.

  import sys
  T = int(input()) #Test case
  for i in range(T):
         a,b = map(int, sys.stdin.readline().split())
         print(a+b)

1. 한 개의 정수를 입력받을 때

import sys
a = int(sys.stdin.readline())

2. 정해진 개수의 정수를 한줄에 입력받을 때

import sys
a,b,c = map(int,sys.stdin.readline().split())

3. 임의의 개수의 정수를 한줄에 입력받아 리스트에 저장할 때

import sys
data = list(map(int,sys.stdin.readline().split()))

4. 임의의 개수의 정수를 n줄 입력받아 2차원 리스트에 저장할 때

import sys
data = []
n = int(sys.stdin.readline())
for i in range(n):
    data.append(list(map(int,sys.stdin.readline().split())))

5. 문자열 n줄을 입력받아 리스트에 저장할 때

import sys
n = int(sys.stdin.readline())
data = [sys.stdin.readline().strip() for i in range(n)]

🔗참고
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

🔗백준 - 수 정렬하기 2
https://www.acmicpc.net/problem/2751