글 읽기 - 틀린이유를 알 수 있을까요?

parkwj93 5년 전 0

질문검색의 반례는 모두 맞다고 나옵니다.

어떤부분이 틀렸는지 잘 모르겠어요.ㅠㅠ

도움부탁드립니다.

#include<stdio.h>
#include<iostream>

using namespace std;
struct node {
	int parent;
	int child;
};
int n;
int a,b;
struct node arr[51];
int cnt = 0;
int pa;
void printMap() {
	for (int i = 0; i < n; i++) {
		cout << endl << i << "번째 부모" <<arr[i].parent<<"자식개수"<<arr[i].child;
	}
}
void solve() {
	for (int i = 0; i < n; i++) {
		if (arr[i].parent != -1) {
			if (arr[i].child == 0) {
				cnt++;
			}
		}
	}
}
void change(int s) {
	arr[s].parent = -1;
	if (arr[s].child == 0) {
		return;
	}
	for (int i = 0; i < n; i++) {
		if (arr[i].parent==s) {
			change(i);
		}
	}
}

int main() {
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> arr[i].parent;
		if (arr[i].parent != -1) {
			arr[arr[i].parent].child++;
		}
		if (arr[i].parent == -1) {
			pa = i;
		}
	}
	cin >> b;
	if (b == pa) {
		cout << "0";
	}
	else {
		arr[arr[b].parent].child--;
		//printMap();
		change(b);
		if (arr[pa].child == 0) {
			cout << "1";
		}
		else {
			solve();
			cout << cnt;
		}
	}
	return 0;
}

degurii 5년 전 2

자세히 안봐서 모르겠지만

-1 0

의 반례가 있네요

1을 출력해야합니다.

parkwj93 5년 전 0

댓글 감사합니다.

위의 반례를 추가하고

0번에 -1 온다고 생각했던 부분의 main 코드를 고치니까 맞았네요

감사합니다.

정답코드로 수정했습니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

틀린이유를 알 수 있을까요?

parkwj93 5년 전 0

degurii 5년 전 2

parkwj93 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말