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

안녕하세요 큰돌님!

2-R에서 선생님께서는 child라는 변수를 만들어 기저사례를 정하셨는데 저는 a[here].size() 즉 배열의 크기가 0이면 return 1을 하도록 기저사례를 정의했습니다.

그렇게 했더니 틀렸습니다가 뜨더군요ㅠㅠㅠ 배열의 크기로 확인하면 안되는 걸까요..?

#include <bits/stdc++.h>
using namespace std;
vector<int> a[54];
int n, tmp, del, root;

int dfs(int here) {
	int ret = 0;
	//이 부분입니다! 
	if(a[here].size() == 0) return 1;
	
	for(int there : a[here]) {
		if(there == del) continue;
		ret += dfs(there);
	}

	return ret;
}
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	
	cin >> n;
	for(int i = 0; i < n; i++) {
		cin >> tmp;
		
		if(tmp == -1) root = i;
		else a[tmp].push_back(i);
	}
	cin >> del;
	
	if(del == root) {
		cout << 0 << "\n";
		return 0;
	}
	cout << dfs(root) << "\n";
	return 0;
}