算法实现——树的最大连通分支问题

问题描述

我们每次在进行权重的加和之前可以事先判断一下权重是否为负数:

·w[k]<0:直接跳过这个顶点k
·w[k]>=0:假设sum(k)是以k为根节点的树的最大连通子图的权值，那么有递推关系：
sum(k)=∑{sum(i)}+w(k)
i是k的子节点，并且满足sum(i)>0

我们所求的问题就是最大连通分支的权值max{sum(k)}中的k

代码

#include <iostream>
#include <vector>
using namespace std;
const int maxn=1005;
vector<int>G[maxn];//图
int w[maxn];//边权
bool visit[maxn]={0};//标记访问
int res=0;//记录最大联通子图权值
int tree(int u)
{
    visit[u]=1;//标记访问
    int sum=w[u];//以该节点为根节点的树的最大联通子图权值
    int v,temp;//子节点编号和临时变量
    for(int i=0;i<G[u].size();i++)
    {
        v=G[u][i];
        if(!visit[v])
        {
            temp=tree(v);
            if(temp>0) sum+=temp;
        }
    }
    res=max(res,sum);//迭代选择最大值
    return sum;
}
int  main()
{
    int n,u,v;
    cin>>n;
    for(int i=0;i<n;i++) cin>>w[i];
    for(int i=0;i<n-1;i++)
    {//构图
        cin>>u>>v;
        G[u-1].push_back(v-1);
        G[v-1].push_back(u-1);
    }
    int res=tree(0);
    cout<<res<<endl;
    return 0;
}

算法分析

每个结点都只访问了一次，所以时间复杂度为O(n)