24-InSange

[InSange]

🔗 문제 링크

트리

✔️ 소요된 시간

4시간

✨ 수도 코드

문제 개요

노드의 개수 N개가 주어지고 각 노드의 부모를 가리키는 인덱스 번호가 N개 주어진다.
루트 노드의 부모는 -1이다.
그리고 마지막으로 지우고자 하는 노드의 번호가 주어지며 해당 노드의 자식 노드도 다 없어진다.

접근 방식

각 노드에는 자식들의 인덱스를 저장할 배열과 부모 노드의 주소를 저장할 변수를 지니고 있으며 추가로 리프노드인지 확인할 수 있는 불리언 변수도 지니고 있다.
해당 노드는 구조체로 이루어져 있으며 각 노드들에 대한 주소는 배열 nodes에 저장된다.
지우고자 하는 노드를 입력받으면 해당 노드가 루트 노드인지 파악하여 루트 노드가 아닐 경우 해당 노드의 부모 노드를 업데이트 시켜준다.

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

struct Node
{
	Node()
	{
		parent = nullptr;
		leafNode = true;
	}

	vector<int> childs;
	Node* parent;
	bool leafNode;
};

vector<Node*> nodes;
int N, ans = 0, rIndex;

void NodeUpdate(Node* curNode)
{
	for (auto child : curNode->childs)
	{
		if (nodes[child] != nullptr)
		{
			curNode->leafNode = false;
			return;
		}
	}
	curNode->leafNode = true;
}

void Solve()
{
	cin >> N;

	for (int i = 0; i < N; i++)
	{
		Node* node = new Node();

		nodes.push_back(node);
	}
	int root_index = 0;

	for(int i = 0; i < N; i++)
	{
		int p;
		cin >> p;

		if (p == -1)
		{
			root_index = i;
			continue;
		}
		nodes[i]->parent = nodes[p];
		nodes[i]->parent->childs.push_back(i);
		if (nodes[i]->parent)
		{
			NodeUpdate(nodes[i]->parent);
		}
	}

	cin >> rIndex;
	if (rIndex == root_index)	// 제거 된게 루트라면
	{
		cout << ans << "\n";
		return;
	}
	Node* parentNode = nodes[rIndex]->parent;
	delete nodes[rIndex];
	nodes[rIndex] = nullptr;
	NodeUpdate(parentNode);

	queue<Node*> q;
	q.push(nodes[root_index]);

	while (!q.empty())
	{
		Node* cur = q.front();
		q.pop();

		if (cur->leafNode) ans++;
		for (auto index : cur->childs)
		{
			if (nodes[index] != nullptr)
			{
				q.push(nodes[index]);
			}
		}
	}

	cout << ans;
}

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

	Solve();

	return 0;
}

처음에 접근할 때 루트 노드는 맨 처음 것이 아닐 수 있다는 점과 지워지는 노드가 루트 노드일 수 있다는 점을 간과하여 틀리게 되었고 해당 부분을 파악후 예외 처리해준 뒤 통과할 수 있었다.

후기

구조체를 활용하여 노드형식으로 풀이하는 방식이 오랜만이었다.
특히 포인터를 처리하는 부분에서 헷갈리는 부분도 발생하는 문제가 있었다.
자주 사용하는 방식임에도 알고리즘에서 막히다보니 자주 연습해야겠다는 생각이 들었다.

📚 새롭게 알게된 내용

[yuyu0830]

진짜 트리를 만들어서 풀었네요.. 문제 자체는 어려운 문제는 아니지만 루트 제거 같이 의외로 귀찮은 조건들이 있는 문제로 기억합니다..

확실히 자료구조는 자주 사용해줘야하는 것 같아요.. 까먹기 너무 쉬움

[seongwon030]

코드가 기능하는 단위가 눈에 잘 보여서 좋았어요. 명명도 잘 되어 있어 구조체로 선언된 변수들이 어떤 역할을 하는지 쉽게 이해할 수 있었습니다.