Mùa hè oi ả ở Wisconsin đã khiến cho lũ bò phải đi tìm nước để làm dịu đi cơn khát. Các đường ống dẫn nước của nông dân John đã dẫn nước lạnh vào 1 tập \(N\) nhánh (đánh số từ \(1...N\)) từ một cái bơm đặt ở chuồng bò.
Khi nước lạnh chảy qua các ống, sức nóng mùa hè sẽ làm nước ấm lên. Bessie muốn tìm chỗ có nước lạnh nhất để cô bò có thể tận hưởng mùa hè một cách thoải mái nhất.
Bessie đã vẽ sơ đồ toàn bộ các nhánh ống nước và nhận ra rằng nó là một đồ thị dạng cây với gốc là chuồng bò và ở các điểm nút ống thì có chính xác \(2\) nhánh con đi ra từ nút đó. Một điều ngạc nhiên là các nhánh ống này đều có độ dài là \(1\).
Cho bản đồ các ống nước, hãy cho biết khoảng cách từ chuồng bò tới tất cả các nút ống và ở các phần cuối đường ống.
"Phần cuối" của một đường ống, có thể là đi vào một nút ống hoặc là bị bịt, được gọi theo số thứ tự của đường ống. Bản đồ có \(C\) nút ống, được mô tả bằng \(3\) số nguyên: là "phần cuối" của ống \(E_{i}\) và \(2\) ống nhánh đi ra từ đó là \(B_{1i}\) và \(B_{2i}\). Đường ống số \(1\) nối với chuồng bò; khoảng cách từ phần cuối của đường ống này tới chuồng bò là \(1\).
Input
- Dòng 1: 2 số nguyên cách nhau bởi dấu cách: \(N\) và \(C\)
- Dòng \(2...C+1\): Dòng \(i+1\) mô tả nút ống \(i\) với ba số nguyên cách nhau bởi dấu cách: \(E_{i},B_{1i}\), và \(B_{2i}\).
Output
- Dòng \(1...N\): Dòng \(i\) chứa \(1\) số nguyên là khoảng cách từ chuồng tới "phần cuối" của ống thứ \(i\).
Constraints
- \(3 \leq N \leq 99999\), \(N\) lẻ
- \(1 \leq C \leq N\)
- \(1 \leq E_{i} \leq N\)
- \(2 \leq B_{1i}, B_{2i} \leq N\)
Example
Test 1
Input
5 2
3 5 4
1 2 3
Output
1
2
2
3
3
Note
Dữ liệu ở trên mô tả bản đồ ống nước sau:
+-––––––-+
| Chuồng |
+-––––––-+
| 1
*
2 / \ 3
*
4 / \ 5
Ống 1 luôn cách chuồng 1 đoạn là 1. Ống 2 và 3 nối với ống 1 nên khoảng cách sẽ là 2. Ống 4 và 5 nối với ống 3 nên khoảng cách sẽ là 3.
```
Bình luận
from collections import deque
def find_distances(N, C, edges):
graph = [[] for _ in range(N + 1)]
for i in range(C):
E_i, B1_i, B2_i = edges[i]
graph[E_i].append(B1_i)
graph[E_i].append(B2_i)
graph[B1_i].append(E_i)
graph[B2_i].append(E_i)
distances = [-1] * (N + 1)
distances[1] = 1
queue = deque([1])
N,C=map(int,input().split())
edges=[]
for i in range(C):
a=list(map(int,input().split()))
edges.append(a)
distances = find_distances(N, C, edges)
print("\n".join(map(str, distances)))
khó phết
code python
1 bình luận nữa