[모각코] 210717 Today I Learned

<백준 18126번 - 너구리 구구>

 

내가 지금 .. 무언가 잘못된 길을 걷고 있다...

어려운 문제가 아닌 것 같은데 내가 지금 어렵게 풀고 있는 거다...

알고리즘 구상을 다시 해야 될 것 같다.

 

visited를 2차원으로 만들어서 각각의 1번 인덱스에 거리를 저장하고 그 값을 갱신한 후 가장 마지막 값을 출력하려고 했는데

지금 이거 쓰면서 생각해보니 이게 아니라.. 거리의 최댓값을 찾아야 되네...?

 

아 문제 좀 제대로 읽을 걸 완전 다르게 이해하고 문제 풀고 있었다.

지금까지 푼 코드라도 올릴까 했는데 아예 다른 알고리즘이라 ㅋㅋ

다 풀고 나면 그때 올려야겠다.....  😭

 

 

 

---

 

 

 

#210718

사실 어제 모각코 할 때는 귀찮아서 종이 없이 머리로만 구상했다. ㅋㅋ 오늘 다시 풀어보려고 종이에 쓰면서 풀었는데 예제는 맞았다. 제출하니까 틀렸지만...

스터디 때문에 풀어야 하는 문제가 더 남아있어서 다른 거 먼저 풀거나 장고 복습하고 다시 풀어봐야겠다.

 

일단 지금까지 푼 (틀린) 코드

import sys

def dfs(v):
    visited[v] = 1
    for i in range(len(adjList[v])):
        w = adjList[v][i][0]
        if not visited[w]:
            dist[w] = max(max(dist), adjList[v][i][1] + dist[v])
            dfs(w)

n = int(sys.stdin.readline())
visited = [0 for i in range(n+1)]
adjList = [[] for i in range(n+1)]
dist = [0 for i in range(n+1)]
for i in range(1, n):
    a, b, c = map(int, sys.stdin.readline().split())
    adjList[a].append([b, c])
    adjList[b].append([a, c])

dfs(1)
print(max(dist))

뭐가 틀린거지...... 조금만 고치면 맞을 거 같은데... :(