https://www.acmicpc.net/problem/11725
11725번: 트리의 부모 찾기
루트 없는 트리가 주어진다. 이때, 트리의 루트를 1이라고 정했을 때, 각 노드의 부모를 구하는 프로그램을 작성하시오.
www.acmicpc.net
연결된 정점 1개만 자신의 부모
따로 부모 배열을 생성하여 각 요소의 부모 (정점)을 저장
탐색하며 nxt 가 부모인 경우 패스 아닌 경우 탐색
#include <bits/stdc++.h>
using namespace std;
int n;
vector<int> vec[100001];
int p[100001];
void dfs(int root){
stack<int> s;
s.push(root);
while(!s.empty()){
int cur = s.top();
s.pop();
for(int i= 0; i<vec[cur].size(); i++){
int nxt = vec[cur][i];
if(p[cur] == nxt) continue;
s.push(nxt);
p[nxt] = cur;
}
}
}
int main(void){
ios::sync_with_stdio(0);
cin.tie(0);
cin>>n;
for(int i=0; i<n-1; i++){
int a,b;
cin>>a>>b;
vec[a].push_back(b);
vec[b].push_back(a);
}
dfs(1);
for(int i=2; i<=n; i++){
cout<<p[i]<<"\n";
}
}
'Algorithm > BOJ' 카테고리의 다른 글
[c++] 백준 10844 쉬운 계단 수 (0) | 2021.01.21 |
---|---|
[c++] 백준 4963 섬의 개수 (0) | 2021.01.19 |
[c++] 백준 7785 회사에 있는 사람 (0) | 2021.01.18 |
[c++] 백준 11724 연결 요소의 개수 (0) | 2021.01.17 |
[c++] 백준 2606 바이러스 (0) | 2021.01.17 |