글 읽기 - 이게 왜틀린걸까요 ㅠㅠ

minjae200 5년 전 0

그냥 -1이면 root니까 root부터 DFS타고가서 for문한번도 안돌면 leafnode라고 판단했고 그것을 출력해주었습니다.

#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

const int MAX = 50;

vector<int> adj[MAX];
int leafnode = 0;
int erasenum;
bool visited[MAX];

void dfs(int from)
{
	int cnt = 0;
	visited[from] = true;

	for (auto p : adj[from]) {
		if (visited[p]) continue;
		dfs(p);
		cnt++;
	}
	if (cnt == 0) {
		//cout << from << endl;
		leafnode++;
	}
}

int main(void)
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	int N; cin >> N;
	int root;
	for (int i = 0; i < N; i++) {
		int num; cin >> num;
		if (num == -1) root = i;
		adj[num].push_back(i);
	}
	/*for (int i = 0; i < N; i++) {
		cout << i << " : ";
		for (int j = 0; j < adj[i].size(); j++) {
			cout << adj[i][j] << " ";
		}
		cout << endl;
	}*/
	memset(visited, false, sizeof(visited));
	cin >> erasenum;
	visited[erasenum] = true;
	dfs(root);
	cout << leafnode << endl;
	return 0;
}

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

문제

문제

출처

ICPC

이게 왜틀린걸까요 ㅠㅠ

minjae200 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말