백준 15372번 / A Simple Problem - 파이썬

📚문제

출처 : 백준 / 15372번 (https://www.acmicpc.net/problem/15372)

 

📝풀이

# 15372 A Simple Problem
import sys
for _ in range(int(input())):
	print(int(sys.stdin.readline())**2)

문제를 이해하는게 너무 어려웠다...

그래서 어떤걸 출력하라는건지 잘 이해가 안됐는데

게시판을 살펴보고 하니까 주어진 정수(n)만큼 테스트케이스를 반복하는데

그 다음 주어지는 정수(t)의 제곱이 되는 수를 출력하는 문제다

 

코드 자체는 어렵지 않은데 

시간제한이 걸려있어서 input()보다 입력받는 속도가 빠른 sys.stdin.readline()을 사용해야한다는 것이 포인트다

전에 몇번 보기도 하고 구글링하다가 다른 풀이를 보면 readline을 많이 사용하긴 하는데 

나는 jupyter notebook에서 돌려보고 코드를 제출해서 잘 사용하지 않았었다

 

근데 for문을 돌면서 반복문을 여러줄 입력받아야할 때는 이런 문제들처럼 시간초과에 걸릴 수 있다

 

참고링크 : 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

[Python 문법] 파이썬 입력 받기(sys.stdin.readline)