D. Sum in the tree(树形+贪心)

题目链接;http://codeforces.com/contest/1099/problem/D

题目大意:给出一棵树,每个节点到根节点的路径上经过的所有点的权值之和,其深度为偶数的节点的信息全部擦除了,也就是用-1表示,让你求最终所有点权之和(要求最小)

具体思路:对于每一个节点,这个点到根节点的权值最小的时候是他的所有的根节点的最小值,然后一路更新上去就可以了,最后求值得时候,我们求父亲节点和子节点之间的差就可以了, 如果差值为负值就是非法情况。

AC代码:

#include<bits/stdc++.h>
using namespace std;
# define ll long long
# define inf 0x3f3f3f3f
const int maxn = 1e5+100;
const int mod = 1e9+7;
int head[maxn],num,dp[maxn][3],flag=1,vis[maxn];
struct node
{
    int nex;
    int to;
} edge[maxn*2];
ll sum;
void addedge(int fr,int to)
{
    edge[num].to=to;
    edge[num].nex=head[fr];
    head[fr]=num++;
}
void dfs1(int fr,int rt)
{
    if(dp[fr][0]==-1)
    {
        int minn=inf;
        for(int i=head[fr]; i!=-1; i=edge[i].nex)
        {
            int to=edge[i].to;
            if(to==rt)
                continue;
            minn=min(minn,dp[to][0]);
        }
        dp[fr][0]=minn;
        //cout<<fr<<" "<<dp[fr][0]<<endl;
    }
    for(int i=head[fr]; i!=-1; i=edge[i].nex)
    {
        int to=edge[i].to;
        if(to==rt)
            continue;
        dfs1(to,fr);
        if(dp[to][0]==inf)
            dp[to][0]=dp[fr][0];
    }
}
void dfs2(int fr,int rt)
{
    if(flag==0)
        return ;
    for(int i=head[fr]; i!=-1; i=edge[i].nex)
    {
        int to=edge[i].to;
        if(to==rt)
            continue;
            dfs2(to,fr);
        sum+=(dp[to][0]-dp[fr][0]);
        if(dp[to][0]-dp[fr][0]<0)
        {
            flag=0;
        }

    }
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        head[i]=-1;
    }
    int tmp;
    for(int i=2; i<=n; i++)
    {
        scanf("%d",&tmp);
        addedge(tmp,i);
        addedge(i,tmp);
    }
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&dp[i][0]);
    }
    dfs1(1,-1);
    dfs2(1,-1);
    sum+=dp[1][0];
    for(int i=1; i<=n; i++)
    {
     //   cout<<dp[i][0]<<endl;
    }
//        for(int i=n-1; i>=1; i--)
//        {
//            sum+=(dp[i+1][0]-dp[i][0]);
//            if(dp[i+1][0]-dp[i][0]<0)flag=0;
//        }
    if(flag==0)
        sum=-1;
    printf("%lld\n",sum);
    return 0;
}
/*
5
1 2 2 3
1 -1 2 3 -1
3
*/

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值