작성
·
184
0
vector<int> tree[51];
int cnt, a;
void go(int node)
{
if (tree[node].size() == 0)
{
cnt++;
return;
}
for (auto& leaf : tree[node])
{
if (m == leaf) continue;
go(leaf);
}
return;
}
int main(void)
{
ios_base::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
cin >> n;
for (int i = 0; i < n; i++)
{
int tRoot;
cin >> tRoot;
if (tRoot == -1)
{
a = i;
continue;
}
tree[tRoot].push_back(i);
}
cin >> m;
if (m == a)
{
cout << 0;
return 0;
}
go(a);
cout << cnt;
return 0;
}
안녕하세요
이 코드는 어떤 점에서 예외가 생겨서 틀리는 걸까요?? 혹시 루트 노드만 남았을 때가 예외일까요??
답변 1
0
안녕하세요 ㅎㅎ
이 코드는 어떤 점에서 예외가 생겨서 틀리는 걸까요?? 혹시 루트 노드만 남았을 때가 예외일까요??
>> 네 맞습니다. a -> b 라는 트리가 있고 b를 지운다고 했을 때 a만 남게 되는데 이 코드는 해당 부분을 처리하지 못합니다.
또 질문 있으시면 질문 부탁드립니다.
좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다.
감사합니다.