poj2342(summerIv) 树状DP水题

参考: 点击打开链接

需要特别注意的是这种题目一定要递归找到根节点再从根节点进行dfs,上面的链接中的代码就是因为没有进行这一步所以TLE

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=6000+50;
int dp[maxn][2],root,father[maxn],vis[maxn];///二维数组不要两边都开这么大很容易MLE
int employee,l,k;

void dfs(int node)
{
    vis[node]=1;
    for(int i=1;i<=employee;i++)
    {
        if(!vis[i]&&father[i]==node)
        {
            dfs(i);
        dp[node][0]=dp[node][0]+max(dp[i][1],dp[i][0]);
        dp[node][1]=dp[node][1]+dp[i][0];
        }
    }
}

int main()
{
    while(scanf("%d",&employee)!=EOF)
    {
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        memset(father,0,sizeof(father));
        for(int i=1;i<=employee;i++)
        scanf("%d",&dp[i][1]);
    while(scanf("%d%d",&l,&k))
    {
        if(l==0&&k==0) break;
        father[l]=k;
    }
    root=k;
        while(father[root])
            root=father[root];
        dfs(root);
    printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值