感觉十分对过了样例但是WA的代码求调

#include<bits/stdc++.h>
using namespace std;
const int T = 110;
vector<int> F[T];
vector<int> a[T];
int N,Q;
vector<int> son[4];
void dfs(int k,int fa)
{
    if(son[k].size()==1)
    {
        return;
    }
    if(N>=3)
    {
        int tmp[2];
        int count = 0;
        for(int j = 0;j<son[k].size();j++)
        {
            if(son[k][j]!=fa)
            {
                dfs(son[k][j],k);
                tmp[count] = j;
                count++;
            }
        }
        F[k][1] = max(F[son[k][tmp[0]]][0]+a[k][tmp[0]],F[son[k][tmp[1]]][0]+a[k][tmp[1]]);
        F[k][2] = max(F[son[k][tmp[0]]][1]+a[k][tmp[0]],F[son[k][tmp[1]]][1]+a[k][tmp[1]]);
        for(int l = 2;l<=N-1;l++)
            for(int t = 0;t<=l-2;t++)
                F[k][l] = max(F[k][l],F[son[k][tmp[0]]][l-t-2]+F[son[k][tmp[1]]][t]+a[k][tmp[0]]+a[k][tmp[1]]);
    }
    else if(N==2)
        F[1][1] = a[1][0];
}
int main()
{
    scanf("%d%d",&N,&Q);
    for(int i = 0;i<N+2;i++)
        for(int j = 0;j<N+2;j++)
            F[i].push_back(0);
    for(int i = 1;i<=N-1;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        a[x].push_back(z);
        a[y].push_back(z);
        son[x].push_back(y);
        son[y].push_back(x);
    }
    dfs(1,-1);
    printf("%d",F[1][Q]);
    return 0;
}

这里我用F[x][y]表示以x为节点的子树在保留y根枝条的情况下能得到的最大苹果数量,a[x][y]表示x节点的第y根枝条的苹果数量。状态转移方程是这样的:如果要保留的枝条数y>=2,有三种选择:只保留左枝条,然后左树保留y-1根枝条;只保留右枝条,然后右树保留y-1根枝条;左右两根枝条都保存,左树保留t根枝条(t从0到y-2),右树保留y-t-2根枝条。

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-02 17:28
25届每天都在焦虑找工作的事情0offer情绪一直很低落硬撑着面了一个岗位岗位有应酬的成分面试的时候hr给我出各种场景题问的问题比较犀利&nbsp;有点压力面的感觉感觉有点回答不上来本来就压抑的情绪瞬间爆发了呢一瞬间特别想哭觉得自己特别没用没绷住掉眼泪了事后想想觉得自己挺有病的&nbsp;真的破大防了
喜欢唱跳rap小刺猬...:我觉得没关系吧,之前有一次面试leader给我压力面,我顶住了压力,结果入职的时候发现组里氛围很差,果断跑路。其实从面试就能大概看出组的情况,面试体验好的组倒是不一定好,但是面试体验不好的组。。。就很难说
面试尴尬现场
点赞 评论 收藏
分享
点赞 评论 收藏
分享
07-01 23:23
郑州大学 Java
否极泰来来来来:牛客迟早有高三的
点赞 评论 收藏
分享
就前几天旅游的时候,打开抖音就经常刷到这类视频:以前是高学历学生、老师、主持人,现在做着团播、擦边主播的工作,以及那些经过精心包装的“职业转型”故事——从铺天盖地的VLOG到所谓的“04年夜场工作日记”,这些内容在初中升学、高考放榜等关键时间节点持续发酵。可以说非常直接且精准地在潜移默化地影响着心智尚未成熟的青少年,使其对特殊行业逐渐脱敏。那我就想问了:某些传播公司、平台运营者甚至某些夜场的老板,你们究竟在传递怎样的价值观?点开那些视频,评论区里也是呈现明显的两极分化:一种是​​经济下行论​​:“现在就业市场已经艰难到这种程度了吗?”​​一种是事实反驳派​​:这些创作者往往拥有名校背景,从事着...
牛客刘北:被环境教育的,为了能拿到足够的钱养活自己,不甘心也得甘心,现在的短视频传播的思想的确很扭曲,但是很明显,互联网玩上一年你就能全款提A6,但你全心全意不吃不喝工作一年未必能提A6,但是在高考中考出现这个的确很扭曲,在向大家传播“不上学,玩互联网也可以轻松年入百万”,不是人变了,是社会在变
预测一下26届秋招形势
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务