인프런 커뮤니티 질문&답변

jwk님의 프로필 이미지
jwk

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

2-R

트리 문제 질문드려요

작성

·

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점은 제게 큰 힘이 됩니다.

감사합니다.

jwk님의 프로필 이미지
jwk

작성한 질문수

질문하기