感觉十分对过了样例但是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根枝条。