CF1436D Bandit in a City（二分，貪心）

傳送門

---

解題思路

先二分答案，然後做一個在樹上從葉子節點向上的貪心，讓每個葉子儘可能取滿。

當然也可以直接樹形dp，O(n)。

注意如果WA on test 5，說明爆long long了，因爲二分mid可以達到1e14，然後貪心過程中累加可能爆掉。

這時候設置一個inf，如果當前sum>inf,sum=inf即可。

AC代碼

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<bitset>
#include<stack>
using namespace std;
template<class T>inline void read(T &x)
{
    x=0;register char c=getchar();register bool f=0;
    while(!isdigit(c))f^=c=='-',c=getchar();
    while(isdigit(c))x=(x<<3)+(x<<1)+(c^48),c=getchar();
    if(f)x=-x;
}
template<class T>inline void print(T x)
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)print(x/10);
    putchar('0'+x%10);
}
const int maxn=4e5+5;
const long long inf=2e15;
int p[maxn],cnt,fa[maxn],in[maxn],n,yyy;
long long w[maxn],l,r;
struct node{
	int v,next;
}e[maxn];
void insert(int u,int v){
	cnt++;
	e[cnt].v=v;
	e[cnt].next=p[u];
	p[u]=cnt;
}
long long dfs(int u,long long x){
	if(!in[u]){
		if(x-w[u]>=0) return x-w[u];
		yyy=1;
		return 0;
	}
	long long sum=0;
	for(int i=p[u];i!=-1;i=e[i].next){
		int v=e[i].v;
		sum+=dfs(v,x);
		if(sum>=inf) sum=inf;
	}
	if(sum-w[u]>=0) return sum-w[u];
	yyy=1;
	return 0;
}
bool check(long long x){
	yyy=0;
	return (dfs(1,x)>=0)&&(yyy==0);
}
int main(){
	ios::sync_with_stdio(false);
	memset(p,-1,sizeof(p));
	cin>>n;
	for(int i=2;i<=n;i++) cin>>fa[i],in[fa[i]]=1,insert(fa[i],i);
	for(int i=1;i<=n;i++) cin>>w[i],r+=w[i];
	while(l<r){
		long long mid=(l+r)/2;
		if(check(mid)) r=mid;
		else l=mid+1;
	}
	cout<<l<<endl;
	return 0;
}