从树中删除边缘的解决方案为某些测试用例提供了错误的答案

时间:2019-06-29 13:35:43

标签: c++ algorithm data-structures

问题陈述如下:

为您提供了n个节点和整数k的树。每个节点都包含一个存储在其中的整数。您需要从树中删除最少数量的边(可能为0),以便在删除该边之后,形成的每棵单独树的总和(树的总和是该树的所有顶点值的总和)小于或等于k。

输入格式

第一行:两个以空格分隔的整数和 下一行:以空格分隔的整数,其中整数表示存储在节点中的值 下一行:两个用空格分隔的整数,表示节点和之间存在边 输出格式 打印问题的整数答案。

约束 1 <= n <= 10 ^ 5

1 <= k <= 10 ^ 9

样本输入:

7 8

4 3 2 7 2 1 6

1 2

1 3

1 4

3 5

3 6

4 7

样本输出: 3

说明: 通过删除节点1和3之间的边缘,节点1和4之间的边缘以及节点4和7之间的边缘,我们可以在最少3次删除的情况下获得所需的条件。

我尝试了一种解决方案,其中我使用dfs方法解决了这一问题,并检查父节点值+子子树值的总和是否大于k,然后删除该边。

#include<stdio.h>
#include<bits/stdc++.h>
using namespace std;
int g[10004][10004];
int val[100005];
int ans=0;
int dfs(int u,int n,int k)
{
    int s=val[u];

    for(int i=0;i<n;i++)
    {   
        if(g[u][i]==1)
        {
            int c = dfs(i,n,k);

            if(c+s>k)
            {
               ans++; 
            }
            s+=c;
        }
    }
    return s;
}
int main()
{
    int n,k,r;
    cin>>n>>k;
    for(int i = 0; i < n; i ++)
    cin>>val[i];
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            g[i][j]=0;
        }
    }
    for(int i = 0; i < n - 1; i ++)
    {
        int u,v;
        cin>>u>>v;
        if(i==0)
        r=u;
        g[u-1][v-1]=1;

    }
    int c = dfs(r-1,n,k);
    cout<<ans;
    return 0;
}

对于某些测试用例,它可以正常工作,但对于其他测试用例却无法正常工作,我无法找到此类测试用例。请帮忙。

1 个答案:

答案 0 :(得分:1)

这个问题在正在进行的比赛中被问到。
这是比赛的link

请删除它,因为这对其他参与者不公平。