A7301 | 霓虹线路
时间限制1s
内存限制128MB
通过 / 提交0/0
题目描述
雾港城的道路网络是一棵树，共有 $n$ 个路口（编号 $1\sim n$）与 $n-1$ 条道路。学宫打算给每条道路安装一种颜色的霓虹灯。
为了避免同一路口附近的霓虹混淆，规定：
- 对任意路口 $v$，与 $v$ 相连的所有道路颜色必须 两两不同。
现在给定可用颜色总数为 $k$（颜色编号为 $1\sim k$）。请判断是否能完成安装；若能，输出任意一种可行的边染色方案。
为了避免同一路口附近的霓虹混淆，规定：
- 对任意路口 $v$，与 $v$ 相连的所有道路颜色必须 两两不同。
现在给定可用颜色总数为 $k$（颜色编号为 $1\sim k$）。请判断是否能完成安装；若能，输出任意一种可行的边染色方案。
输入格式
第一行两个整数 $n,k$。
接下来 $n-1$ 行，每行两个整数 $u,v$，表示树上的一条无向边（输入顺序记为第 $i$ 条边）。
接下来 $n-1$ 行，每行两个整数 $u,v$，表示树上的一条无向边（输入顺序记为第 $i$ 条边）。
输出格式
若无解，输出一行：
```
NO
```
若有解，输出两行：
第一行：
```
YES
```
第二行输出 $n-1$ 个整数 $c_1,c_2,\dots,c_{n-1}$，其中 $c_i$ 表示第 $i$ 条边的颜色（$1\le c_i\le k$）。要求对每个点，相邻边颜色两两不同。
```
NO
```
若有解，输出两行：
第一行：
```
YES
```
第二行输出 $n-1$ 个整数 $c_1,c_2,\dots,c_{n-1}$，其中 $c_i$ 表示第 $i$ 条边的颜色（$1\le c_i\le k$）。要求对每个点，相邻边颜色两两不同。
输入输出样例
输入 #1
5 3 1 2 2 3 2 4 4 5
输出 #1
YES 1 2 3 1
- $1\le n\le 2\times 10^5$
- $1\le k\le 2\times 10^5$
- $1\le k\le 2\times 10^5$
C++ 编辑器
输入
输出
可保存默认模板；新题优先使用已保存模板。
当前快捷键仅展示，暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好，我是作答助手。可以问思路、复杂度、样例含义或代码报错原因；不会直接给出完整 AC 代码。
确定要清空代码吗？