2023河南萌新联赛第（五）场：郑州轻工业大学 -E 换根DP

2023河南萌新联赛第（五）场：郑州轻工业大学 -E 换根DP

传送带

题目描述

解题思路

可以看到$w$只有$1，2$俩个值，而所求的$w(u,v)$是边权的最大公约数，所以若路径上有$w=1$则后面的全为1，只需要找到$w=2$的连通块并记录下来，再计算即可。（具体看样例注释

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+4;
int n,u,v,w;
vector<int> tr[N];   //记录w=2的连通块
int flag[N];         //判断是否已经算过
int coun[N];         //计算连通块的大小
int cnt;
void dfs(int x,int fa)
{
    flag[x]=cnt; //标记连通块的位置
    for(auto i:tr[x])//遍历儿子
    {
        if(i==fa) continue ;
        dfs(i,x);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        if(w==2)//只记录w=2的
        {
            tr[u].push_back(v);
            tr[v].push_back(u);
        }
    }
    for(int i=1;i<=n;i++)
    {
        cnt++;
        if(!flag[i])
            dfs(i,i);
    }
     for(int i=1;i<=n;i++){
         coun[flag[i]]++;
     }
    int ans=1e9+7;
    for(int i=1;i<=n;i++)
    {
        ans=min(ans,(coun[flag[i]]-1)*2+n-coun[flag[i]]);
        该节点所在连通块大小减去它本身 算权值（w=2)，加上其余全职为1的
    }
    printf("%d",ans);
}