Baekjoon #1068

Baekjoon #1068

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
#include <bits/stdc++.h>
using namespace std;
vector<int> tree[51];
int N, del, answer;

void dfs(int idx) {
    bool isOk = false;
    for(auto next: tree[idx]) {
        isOk = true;
        dfs(next);
    }
    if(!isOk) answer++;
}

int main() {
    ios::sync_with_stdio(false); 
    cin.tie(NULL);
    
    cin >> N;
    int root = 0;
    for(int i = 0; i < N; i++) {
        int tmp;
        cin >> tmp;
        if(tmp == -1) {
            root = i;
            continue;
        }
        tree[tmp].push_back(i);
    }
    
    cin >> del;
    tree[del].clear();
    for(int i = 0; i < N; i++) {
        for(int j = 0; j < tree[i].size(); j++) {
            if(del == tree[i][j]) tree[i].erase(tree[i].begin() + j);
        }
    }
    
    if(del != root) dfs(root);
    cout << answer;
}

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.