似乎用牛客的人越来越少了mua.
嗯,从今天开始写点题解呐呐呐...
就从陪队友做的cf题开始吧,emmm,马上到学校了,加油...
这题挺简单的,因为它是要删叶子节点嘛,那你从根节点走下来的一定是只有正数才保留,且一定要满足我这个节点的元素的值>从根节点走下来那条路径的和,这里,我们使用树上前缀和即可解决/vov.如此记录前缀和以及子树大小就可以了吧...
然后代码如下:
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
vector<int>v[N];
int a[N],w[N],sz[N],sum[N],ans;//a是各点的权值 w是i号节点上面那边的边权 sz是子树大小 sum是从1开始1的前缀和.
void dfs1(int u)
{
sz[u]=1;
for(int i=0;i<v[u].size();i++)
{
int son=v[u][i];
dfs1(son);
sz[u]+=sz[son];
}
}
void dfs2(int u,int fa)
{
sum[u]+=sum[fa]+w[u];
if(sum[fa]<0) sum[u]=w[u];
if(a[u]<sum[u]) {ans+=sz[u];return;}
for(int i=0;i<v[u].size();i++)
{
int son=v[u][i];
dfs2(son,u);
}
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=2;i<=n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
w[i]=y;
v[x].push_back(i);
}
dfs1(1);//dfs1出来子树大小.
dfs2(1,0);//dfs2出来前缀以及ans.
printf("%d\n",ans);
return 0;
}发现自己代码越来越简洁了2333...

京公网安备 11010502036488号