dfs

Shortest Path

http://www.nowcoder.com/questionTerminal/f1d38ddf05124dc9898280431349fad9

解题思路:
只考虑一颗儿子全是叶子的子树时,那么最佳的匹配方案是这些叶子节点两两配对,多余的和根配对。直观感觉这个子树的边全被统计了,而且只统计了一次,如果叶子和非子树的点配对的话,这个子树的边仍然全被统计了,子树根到它父亲的边会被多次统计。
按照这个思路,所有边最多只会被统计一次。

现在证明这个方案是最优的。
如果一颗子树的节点数是奇数,多出一个和非子树的点匹配,所以这颗子树的根的父边一定会被统计,如果是偶数个点,按照上述思路一定不会统计,用其他方法可能会被统计。所以这个方法是最优的。

#include <bits/stdc++.h>
using namespace std;
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define me0(a) memset(a,0,sizeof(a))
#define me1(a) memset(a,-1,sizeof(a))
#define op freopen("in.txt","r",stdin);
#define read(a) scanf("%d",&a)
const int INF = 0x3f3f3f3f;
typedef long long LL;
#define mp(x,y) make_pair(x,y)
#define pii pair<int,int>
const int maxn = 2e5 + 100;
const int mod = 998244353;
int n;
vector<pii>g[maxn];
bool used[maxn];
LL ans=0;
void dfs(int u,int fa){
    int son=0;
    for(auto it:g[u]){
        int to=it.first,w=it.second;
        if(to==fa) continue;
        dfs(to,u);
        if(used[to]) continue;//已经配对了
        son++;ans+=w;
    }
    if(son&1) used[u]=true;//奇数个儿子,有一个需要和父节点进行配对
}

int main(){
    int t;
    read(t);
    while(t--){
        read(n);
        ans=0;
        fo(i,1,n) g[i].clear(),used[i]=false;
        fo(i,1,n-1){
            int u,v,w;
            read(u);read(v);read(w);
            g[u].push_back(mp(v,w));
            g[v].push_back(mp(u,w));
        }
        dfs(1,0);
        printf("%lld\n",ans);
    }
    return 0;
}
全部评论
tql
点赞 回复
分享
发布于 2020-04-08 09:57

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务