蓝桥杯 算法训练 结点选择

题解参考：https://blog.csdn.net/u013852115/article/details/79437734

https://www.cnblogs.com/A-Little-Nut/p/10394084.html

语法参考：https://www.cnblogs.com/zhuruibi/p/8983192.html

https://www.cnblogs.com/home123/p/7450758.html

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int dp[2][N]={0};
vector<vector<int> > ve;
void dfs(int now,int pre)//因为只有一个来处，所以用一个值来标识即可，不需用到vis数组
{
    for (int i=0;i<ve[now].size();i++)
    {
        int to=ve[now][i];
        if (to==pre) continue;
        dfs(to,now);
        dp[0][now]+=max(dp[0][to],dp[1][to]);//注意是+=!
        dp[1][now]+=dp[0][to];
    }
}
int main()
{
//    freopen("in.txt","r",stdin);
    int n;
    cin>>n;
    ve.resize(n+1);//此句不能少！！
    for (int i=1;i<=n;i++)//注意审题，这里需从0开始！！
    {
        int tmp;
        cin>>tmp;
        dp[1][i]=tmp;//tmp是第i个节点的权值不是第tmp个节点的权值！！
    }
    for (int i=0;i<n-1;i++)
    {
        int a,b;
        cin>>a>>b;
        ve[a].push_back(b);
        ve[b].push_back(a);
    }
    dfs(1,0);
    cout<<max(dp[0][1],dp[1][1])<<endl;

    return 0;
}