[11725] 트리의 부모 찾기
BOJ/Tree2018. 5. 20. 01:52
[11725] 트리의 부모 찾기 : http://boj.kr/11725
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | #include <cstdio> #include <vector> #include <queue> using namespace std; int main() { int n; scanf("%d", &n); vector<int> v[100001]; for(int i=1; i<n; i++) { int a, b; scanf("%d%d", &a, &b); v[a].push_back(b); v[b].push_back(a); } vector<int> p(n+1, 0); queue<int> q; q.push(1); p[1] = -1; while(!q.empty()) { int parent = q.front(); q.pop(); for(int i=0; i<v[parent].size(); i++) { int child = v[parent][i]; if(!p[child]) { q.push(child); p[child] = parent; } } } for(int i=2; i<=n; i++) printf("%d\n", p[i]); return 0; } | cs |
>> 부모를 찾지 못했다면 큐에 넣고 부모 저장. BFS.
'BOJ > Tree' 카테고리의 다른 글
[4803] 트리 (0) | 2018.05.21 |
---|---|
[1991] 트리 순회 (0) | 2018.05.20 |