题目连接
思维量比较高的题目,首先,我们判断有解:
该题目有解当且仅当:
图中所有点的度均不等于2
理由如下:
如果有一个度为2的点(则其一定不是叶子),我们发现,与之相连的两条边是连锁的,他们只能一起加减,所以,他们的值应当永远相同,不符合题意中:所有边权互不相同。
而对于没有度数2节点的图,我们都可以以下步骤找到解:
枚举每一条边:
对于每一个边的两个端点,有以下三种情况:
1.两个节点都是叶子。
2.一个节点的度大于等于3,另一个为叶子
3.两个节点都是度大于等于3的点。
第一种
只可能存在于n==2的情况,我们直接输出即可。
对于下面两个情况:
第二种:
设度大于等于3的节点为roota,另一个为rootb,roota到rootb的边权为W,
并设:任意取以rootb为根节点不包含roota的两个子树T1,T2;
T1
/
rootb ----- roota -- T2
\
T......
我们可以考虑,如果有三个操作:
从rootb到T1:加 W/2
从rootb到T2:加 W/2
从T1到T2:减 W/2
进行完这三个操作后:roota----rootb的边权符合要求,而树上其他边权不变。而题目中给的边权都是偶数!这就非常巧妙。
第三种:
类比第二中情况,只不过是找roota的子树T1,T2,rootb的子树T3,T4.然后:
从T1到T3:加 W/2
从T2到T4:加 W/2
从T1到T2:减 W/2
从T3到T4:减W/2
符合要求!
实现:
我们使用结构体和邻接表建图。使用结构图对边进行枚举:然后分情况讨论。
对于找子树,我们枚举端点的边,确定不是另一个端点后,开始dfs出度为1的点(即叶子),然后加入ans结构体中。
下面是ac代码:
#include <iostream>
#include <queue>
#include <vector>
#include <cstring>
#include <map>
#include <cmath>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <conio.h>
#define ll long long
#define INF 0x3f3f3f3f
#define N 1024
#define ll long long
using namespace std;
int he[N], ne[N * 2], ver[N * 2], e[N * 2];
int di[N];
int tot = 1;
int cnt=0;
int _cnt = 0;
struct Node
{
int x, y;
int w;
Node(){}
Node (int a, int b, int c):x(a), y(b), w(c){}
}eg[N], ans[100005];
int suba, subb;
void add(int x, int y, int w)
{
e[++tot] = w;
ver[tot] = y;
ne[tot] = he[x];
he[x] = tot;
}
int dfs(int cur, int fa)
{
if (di[cur] == 1) return cur;
int y = ver[he[cur]];
if (y == fa) y = ver[ne[he[cur]]];
return dfs(y, cur);
}
void fi(int k, int lin)
{
int gg = he[k];
int nf = ne[gg];
suba = ver[gg];
if (lin == suba)
{
suba = ver[nf];
nf = ne[nf];
}
if (ver[nf] == lin)
{
nf = ne[nf];
}
subb = ver[nf];
return;
}
int main()
{
int n;
scanf("%d", &n);
int x, y, w;
for (int i = 1; i < n; i++)
{
scanf("%d%d%d", &x, &y, &w);
eg[cnt++] = Node(x, y, w);
add(x, y, w);
add(y, x, w);
di[x]++;
di[y]++;
}
for (int i = 1; i <= n; i++)
{
if (di[i] == 2)
{
printf("NO\n");
return 0;
}
}
if (n == 2)
{
printf("YES\n1\n%d %d %d\n", x, y, w);
return 0;
}
for (int i = 0; i < cnt; i++)
{
int roa = eg[i].x;
int rob = eg[i].y;
// cout << roa << " " << rob << ":" << endl;
if (di[roa] == 1 && di[rob] > 1)
{
// cout << "s1:" << endl;
fi(rob, roa);
// cout << suba << " " << subb << endl;
int zha = dfs(suba, rob);
int zhb = dfs(subb, rob);
// cout << zha << " " << zhb << endl;
ans[_cnt++] = Node(roa, zha, eg[i].w / 2);
ans[_cnt++] = Node(roa, zhb, eg[i].w / 2);
ans[_cnt++] = Node(zha, zhb, -eg[i].w / 2);
}
if (di[roa] > 1 && di[rob] == 1)
{
// cout << "s2:" << endl;
fi(roa, rob);
// cout << suba << " " << subb << endl;
// cout << suba << " " << subb << endl;
int zha = dfs(suba, roa);
int zhb = dfs(subb, roa);
// cout << zha << " " << zhb << endl;
ans[_cnt++] = Node(rob, zha, eg[i].w / 2);
ans[_cnt++] = Node(rob, zhb, eg[i].w / 2);
ans[_cnt++] = Node(zha, zhb, -eg[i].w / 2);
}
if (di[roa] > 1 && di[rob] > 1)
{
// cout << "s3:" << endl;
fi(rob, roa);
int zha = dfs(suba, rob);
int zhb = dfs(subb, rob);
fi(roa, rob);
int zhc = dfs(suba, roa);
int zhd = dfs(subb, roa);
// cout << zha << " " << zhb <<" " << zhc << " " <<zhd << endl;
ans[_cnt++] = Node(zha, zhc, eg[i].w / 2);
ans[_cnt++] = Node(zhb, zhd, eg[i].w / 2);
ans[_cnt++] = Node(zha, zhb, -eg[i].w / 2);
ans[_cnt++] = Node(zhc, zhd, -eg[i].w / 2);
}
}
printf("YES\n");
printf("%d\n", _cnt);
for (int i = 0; i < _cnt; i++)
printf("%d %d %d\n", ans[i].x, ans[i].y, ans[i].w);
return 0;
}