[BAEKJOON] 10865 친구 친구

문제

도현이네 반 N명의 친구 관계가 입력으로 주어진다. 이때, 각 사람의 친구의 수를 출력하는 프로그램을 작성하시오.

각 사람은 1번부터 N번까지 번호가 매겨져 있다. A와 B가 친구면, B와 A도 친구이다. 자기 자신과 친구인 경우, 즉 A와 B가 같은 경우는 없다.

입력

첫째 줄에 도현이네 반 학생의 수 N(1 ≤ N ≤ 100,000), M(0 ≤ M ≤ 1,000,000)이 주어진다. 

둘째 줄부터 M개의 줄에는 친구 관계를 나타내는 A B가 한 줄에 하나씩 주어진다.

A B가 입력으로 주어진 경우에 B A가 다시 입력으로 주어지는 경우는 없다.

출력

첫째 줄부터 N번째 줄에 걸쳐서 각 학생의 친구의 수를 1번 학생부터 출력한다.

예제 입력 1 

5 5
1 2
3 4
2 5
5 1
4 2

예제 출력 1 

2
3
1
2
2

---

처음 시간 초과가 떠서 삽질했음 ..

 

n, m = map(int, input().split())

p = [[] for _ in range(n+1)]

for _ in range(m):
    a,b = map(int, input().split())
    p[a].append(b)
    p[b].append(a)

for fr in p[1:]:
    print(len(fr))

 

첫 작성 코드

시간 초과가 떳다

 

아 설마 len 함수때매 이런가?

 

n, m = map(int, input().split())

p = [0 for _ in range(n+1)]

for _ in range(m):
    a,b = map(int, input().split())
    p[a] += 1
    p[b] += 1

for i in p[1:]:
    print(i)

 

그래서 이렇게 수정했음 근데 도 시간초과가 ㄷㄷㄷ?

 

그래서 설마하고 sys 입력 문제인가?

 

import sys

n, m = map(int, sys.stdin.readline().split())

p = [0 for _ in range(n+1)]

for _ in range(m):
    a,b = map(int, sys.stdin.readline().split())
    p[a] += 1
    p[b] += 1

for i in p[1:]:
    print(i)

 

import sys

n, m = map(int, sys.stdin.readline().split())

p = [[] for _ in range(n+1)]

for _ in range(m):
    a,b = map(int, sys.stdin.readline().split())
    p[a].append(b)
    p[b].append(a)

for fr in p[1:]:
    print(len(fr))

 

그랬다

 

본의아니게 두가지 방법으로 풀음