글 읽기 - 반례좀 부탁드립니다. 혹은 코드 어느 부분이 잘못 되었는지 알려주세요 ㅠ

rla6431078 6년 전 0

게시판의 모든 예시가 다 되는데 0퍼에서 틀렸데요 ㅠ

#include <cstdio>
#include <cstring>
#include <stack>

int main(){
	int n, arr[50], result, erase,i;
	bool isLeaf[50];
	std::stack<int> s;

	scanf("%d", &n); result = n;
	for (i = 0; i < n; i++){
		scanf("%d", &arr[i]);
		if (arr[i] >= 0 && isLeaf[arr[i]]){
			isLeaf[arr[i]] = false;
			result--;
		}
	}

	scanf("%d", &erase);
	s.push(erase);
	for (i = 0; i < n; i++){
		if (i != erase&&arr[i] == arr[erase])
			break;
	}
	if (i == n&&arr[erase]!=-1)
		result++;
	while (!s.empty()){
		erase = s.top();		
		s.pop();
		if (isLeaf[erase])
			result--;
		for (i = 0; i < n; i++){
			if (arr[i] == erase){
				s.push(i);
			}
		}
	}
	printf("%d", result);
}