인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

Inflearn Community Q&A

jwk's profile image
jwk

asked

10-Week C++ Coding Test | Algorithm Coding Test

2-R

트리 문제 질문드려요

Written on

·

212

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;
}

안녕하세요

이 코드는 어떤 점에서 예외가 생겨서 틀리는 걸까요?? 혹시 루트 노드만 남았을 때가 예외일까요??

c++코딩-테스트

Answer 1

0

kundol님의 프로필 이미지
kundol
Instructor

안녕하세요 ㅎㅎ

이 코드는 어떤 점에서 예외가 생겨서 틀리는 걸까요?? 혹시 루트 노드만 남았을 때가 예외일까요??

>> 네 맞습니다. a -> b 라는 트리가 있고 b를 지운다고 했을 때 a만 남게 되는데 이 코드는 해당 부분을 처리하지 못합니다.

 

또 질문 있으시면 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다.

감사합니다.

jwk's profile image
jwk

asked

Ask a question