[백준] 10807번 - 개수 세기

문제

총 N개의 정수가 주어졌을 때, 정수 v가 몇 개인지 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 정수의 개수 N(1 ≤ N ≤ 100)이 주어진다. 둘째 줄에는 정수가 공백으로 구분되어져있다. 셋째 줄에는 찾으려고 하는 정수 v가 주어진다. 입력으로 주어지는 정수와 v는 -100보다 크거나 같으며, 100보다 작거나 같다.

 

출력

첫째 줄에 입력으로 주어진 N개의 정수 중에 v가 몇 개인지 출력한다.

 

---

[ 제출 코드 ] _ 30840KB, 68ms

import sys

n = int(input())
data = list(map(int, sys.stdin.readline().split()))
v = int(input())

print(data.count(v))

 

[ TIL ] 

python에선 보통 입력 받는 방법으로 input 함수를 사용한다.

그런데 반복문으로 여러줄을 입력 받을 때 input()을 사용하면 시간초과가 발생한다고 한다.

그때 사용할 수 있는 것이 sys 라이브러리다.

sys.stdin.readline()을 사용하면 처리속도도 빠르고 메모리도 적게 소모한다.

 

input 함수는 parameter로 prompt message를 받기 때문에 입력받기 전 이를 출력해야한다.

또한 input 함수는 입력받은 값의 개행 문자를 삭제시켜서 리턴한다. 다시말하면, rstrip() 함수를 적용시켜서 리턴한다.

반면, sys.stdin.readline()은 prompt message를 인수로 받지 않고, 개행문자를 포함한 값을 리턴한다.

따라서 sys.stdin.readline()이 input()에 비해 빠르다.