백준 1764번 파이썬 - 듣보잡(출력 오류 관련)

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

1764번: 듣보잡

문제

김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.

듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.

출력

듣보잡의 수와 그 명단을 사전순으로 출력한다.

예제 입력 1 복사

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

예제 출력 1 복사

2
baesangwook
ohhenrie

 

처음 실행 코드 

import sys
input = sys.stdin.readline
a,b = map(int,input().split())
lst1= []
lst2= []
for s in range(a):
  x = input()
  lst1.append(x)
for n in range(b):
  y= input()
  lst2.append(y)

aa = set(lst1) & set(lst2)
print(len(aa))
for s in sorted(aa,reverse=False):
  print(s)

처음 이 코드를 실행시켰을 때 '출력 형식이 잘못되었습니다.' 라고 뜨는것이다.

나는 코테 풀때 구글코랩으로 하는데, 알 수 없는 이유 때문에 sys.stdin.readline이 계속 오류가 나서 주석처리를 하고, input() 함수 그자체로 푼다.시간 줄이기 위해서 제출할때만 주석을 풀고 진행한다. 

구글코랩으로 실행하는데 아무리 봐도 출력형식이 무엇이 잘못되었는지 모르겠다...

 

검색해보니 input() 대신 sys.stdin.readline을 쓸 때, 끝에 \n의 문자를 포함하므로 출력할때도 \n가 실행된다고 한다.

그래서 strip()으로 제거를 해준다.  

import sys
input = sys.stdin.readline
a,b = map(int,input().split())
lst1= []
lst2= []
for s in range(a):
  x = input().strip()
  lst1.append(x)
for n in range(b):
  y= input().strip()
  lst2.append(y)

aa = set(lst1) & set(lst2)
print(len(aa))
for s in sorted(aa,reverse=False):
  print(s)

위와 같이 수정하면 출력형식 오류가 해결된다.