글 읽기 - 이코드가 왜 틀린건지 도저히 모르겠습니다

smucs13 3년 전 0

클릭해주셔서 감사합니다.

제 머릿속에 나올 수 있는 최선의 방법인데

왜틀린건지 도저히 모르겠습니다 도와주시면 감사합니다

// https://www.acmicpc.net/problem/1068
// 리프노드(자식의 수가 0인 노드)의 개수를 구하기

#include<bits/stdc++.h>
using namespace std;

const int MAX = 51;
int N, ans;
vector<int> tree[MAX];
bool visited[MAX];
int root, deleteNode;
int parent[MAX];

void bfs() {
	queue<int> q;
	q.push(root);

	while (!q.empty()) {
		int cur = q.front();
		q.pop();
		if (tree[cur].size() == 0) {
			ans++;
			continue;
		}
		for (int i = 0; i < tree[cur].size(); i++) {
			int next = tree[cur][i];
			q.push(next);
		}
	}
}

void solve() {
	if (deleteNode == root) return;
    // 삭제될 노드포함 그 이하는 전부 tree에 추가하지 않기
	for (int i = 0; i < N; i++) {
		if (parent[i] == -1 || i == deleteNode || parent[i] == deleteNode) continue;
		tree[parent[i]].push_back(i);
	}

	bfs();
	cout << ans << '\n';
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr); cout.tie(nullptr);

	cin >> N;
	for (int i = 0; i < N; i++) {
		int num;
		cin >> num;
		if (num == -1) root = i;
		parent[i] = num;
	}
	cin >> deleteNode;
	solve();
	return 0;
}	

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

문제

문제

출처

ICPC

이코드가 왜 틀린건지 도저히 모르겠습니다

smucs13 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말