思维/图论 (CodeForces 1189D2 Add on a Tree: Revolution)

题目连接
思维量比较高的题目,首先,我们判断有解:
该题目有解当且仅当:

图中所有点的度均不等于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;
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值