21-seongwon030

[seongwon030]

🔗 문제 링크

트리

✔️ 소요된 시간

1시간

✨ 수도 코드

과정

리프 노드란 자식이 없는 노드이므로 삭제노드가 정해졌을 때 해당 노드를 기준으로 해서 자식 노드를 재귀적으로 없애면 된다.

노드 삭제

1. 삭제노드의 값을 노드 번호와 겹치지 않는 값으로 설정한다.
2. 노드 개수만큼 반복하며 노드의 값이 이전에 설정한의미 없는 값이라면 1번으로 돌아가 같은 과정을 반복한다.
3. 위 과정을 반복하면서 삭제노드의 자식노드까지 모두 없앤다.

def dfs(node):
  k[node] = -2
  for i in range(n):
    if node == k[i]:
      dfs(i)

dfs(del_node)

리프노드의 개수

모든 노드를 순회하며 값이 의미 없는 값이 아니어야 하고 노드 또한 노드 리스트에 없어야 한다.

cnt = 0
for j in range(n):
  if k[j] != -2 and j not in k:
    cnt+=1

print(cnt)

📚 새롭게 알게된 내용

dfs에 대해 파 볼 예정입니다.

[yuyu0830]

풀이를 보면서 굳이 DFS를 통해서 트리를 제거할 필요가 있을까? 라고 생각했습니다. 트리를 DFS로 제거하는 부분과 아래에 트리를 탐색할 때 j not in k 이 부분이 조금 비효율 적인 것 같아서 저는 다른 방식으로 풀어봤어요 (n이 최대 50이기 때문에 효율을 따질 문제가 아니긴 합니다)

저는 트리를 받은 뒤 루트부터 DFS로 탐색을 하되, 탐색 도중 현재 탐색할 노드가 삭제할 노드 번호라면 탐색을 중지하는 방향으로 트리 순회 한번만 하면 되게끔 했습니다. 짜고 보니 복잡도 자체는 크게 차이가 없는 것 같아요..

#include "iostream"
#include "vector"

using namespace std;

vector<int> v[51];

int main() {
    int n, root; cin >> n;

    for (int i = 0; i < n; i++) {
        int t; cin >> t;
        if (t == -1) root = i;
        else v[t].push_back(i);
    }   

    int del, ans = 0; cin >> del;

    vector<int> s;
    if (root != del) s.push_back(root);

    while (!s.empty()) {
        int cur = s.back();
        s.pop_back();

        int cnt = 0;
        
        for (int i : v[cur]) {
            if (i == del) continue;
            s.push_back(i);
            cnt++;
        }
        
        if (!cnt) ans++;
    }

    printf("%d\n", ans);
}

[seongwon030]

문제 풀 때 n이 크지 않아서 모든 노드를 탐색하도록 했는데 n이 커질수록 노드탐색을 중지하는 방법이 훨씬 효율적이겠네요. 덕분에 참고해서 다른 문제에도 적용할 수 있을 것 같아요!

[InSange]

주어진 인덱스 값에서 해당하는 노드 인덱스 값들을 dfs로 연쇄시켜 먼저 제거후 차례대로 탐색하는군요. 이렇게 접근하는 방식도 있다는 것을 보고 갑니다.