题解 CF682C 【Alyona and the Tree】

简单搜索题,我们每找到一组不满足题目给出条件的点和边就将其整个子树删除,然后最终答案加上该子树的大小即可。注意,搜索的时候如果当前的边权和sum已经为负了,应该将其改为0(可以想想为什么)

注:题目翻译有误

原文中的小于应该改为小于等于

#include<bits/stdc++.h>
#define maxn 100005
using namespace std;
inline char get(){
	static char buf[30000],*p1=buf,*p2=buf;
	return p1==p2 && (p2=(p1=buf)+fread(buf,1,30000,stdin),p1==p2)?EOF:*p1++;
}
inline int read(){
	register char c=get();register int f=1,_=0;
	while(c>'9' || c<'0')f=(c=='-')?-1:1,c=get();
	while(c<='9' && c>='0')_=(_<<3)+(_<<1)+(c^48),c=get();
	return _*f;
}
struct edge{
	int u,v,w,next;
}E[maxn<<1];
int p[maxn],eid;
inline void init(){
	for(register int i=0;i<maxn;i++)p[i]=-1;
	eid=0;
}
inline void insert(int u,int v,int w){
	E[eid].u=u;
	E[eid].v=v;
	E[eid].w=w;
	E[eid].next=p[u];
	p[u]=eid++;
}
inline void insert2(int u,int v,int w){
	insert(u,v,w);
	insert(v,u,w);
}
int n;
int a[maxn];
int fa[maxn],size[maxn];
int ans=0;
inline void get_size(int u,int fa){
	size[u]=1;
	for(register int i=p[u];~i;i=E[i].next){
		int v=E[i].v;
		if(v==fa)continue;
		get_size(v,u);
		size[u]+=size[v];
	}
}
inline void dfs(int u,int fa,int sum){
	if(sum>a[u]){
		ans+=size[u];
		//cout<<sum<<" "<<a[u]<<endl;
		return;
	}
	for(register int i=p[u];~i;i=E[i].next){
		int v=E[i].v;
		int w=E[i].w;
		if(v==fa)continue;
		dfs(v,u,max(sum+w,0));
	}
}
int main(){
	//freopen("1.txt","r",stdin);
	init();
	n=read();
	for(register int i=1;i<=n;i++)a[i]=read();
	fa[1]=-1;
	for(register int i=2;i<=n;i++){
		int p=read(),c=read();
		fa[i]=p;
		insert2(i,p,c);
	}
	get_size(1,-1);
	dfs(1,-1,0);
	cout<<ans<<endl;
	return 0;
}
posted @ 2019-03-02 10:40  迷失の风之旅人  阅读(109)  评论(0编辑  收藏  举报
123213123123