zoukankan      html  css  js  c++  java
  • 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;
    }
    
  • 相关阅读:
    进化中的技术小白
    mybatis的学习5______使用注解实现CURD
    mybatis的学习4______分页的实现
    Mybatis的学习3______使用log4j打印日志信息
    mybatis的学习2_____配置文件的详解和代码的优化
    Java代码实现邮件的上传
    JavaWeb小作业 用户表单数据的获取
    Java的学习JSP____4
    Javaweb的学习session___3
    Javaweb的学习(servlet和request)___2
  • 原文地址:https://www.cnblogs.com/yinyuqin/p/15525992.html
Copyright © 2011-2022 走看看