글 읽기 - 왜 틀렸는지 봐주세요..ㅠㅠ

gozj3319 3년 전 0

아무리해도 뭐가 틀렸는지 모르겠네요..

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**7)

def dfs(x, array):
    for c in array[x]:
        if not dp[c]:
            dp[c] = dfs(c, array)
        dp[x] += dp[c] + 1
            
    return dp[x]
    

n, m = list(map(int, input().split()))
heavy = [[] for _ in range(n+1)] # 무거운 배열
light = [[] for _ in range(n+1)] # 가벼운 배열
for _ in range(m):
    tmp = list(map(int, input().split()))
    heavy[tmp[1]].append(tmp[0])
    light[tmp[0]].append(tmp[1])

# 무거운거 dfs + dp를 통해서 자신보다 무거운 것들의 갯수 구하기
dp = [0] * (n + 1)
for i in range(1, n + 1):
    if not dp[i]:
        dfs(i, heavy)

ans = 0
for i, d in enumerate(dp):
    if d > n // 2:
        ans += 1

# 가벼운거 dfs + dp를 통해서 자신보다 가벼운 것들의 갯수 구하기
dp = [0] * (n + 1)
for i in range(1, n + 1):
    if not dp[i]:
        dfs(i, light)

for i, d in enumerate(dp):
    if d > n // 2:
        ans += 1
        
print(ans)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

왜 틀렸는지 봐주세요..ㅠㅠ

gozj3319 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말