zoukankan      html  css  js  c++  java
  • bzoj 2809: [Apio2012]dispatching【dfs序+主席树】

    可并堆就可以,但是想复健一下主席树。
    考虑枚举管理者,然后选忍者的时候在子树中贪心的从小到大选。做成dfs序就是选区间内和小于等于k的最多点。可以用主席树,查询的时候在主席树上二分即可
    这里注意,为了方便起见,离散化的时候即使值相同也离散成不同值,这样可以保证主席树叶子结点的size最大为1,方便二分

    #include<iostream>
    #include<cstdio>
    #include<map>
    #include<algorithm>
    using namespace std;
    const int N=200005;
    int n,m,h[N],cnt,a[N],dfn,rt[N],tot,g[N],l[N],r[N],ss,sk,mp[N];
    long long c[N],d[N],ans;
    struct qwe
    {
    	int ne,to;
    }e[N];
    struct zxs
    {
    	int l,r,ls,rs,s;
    	long long sum;
    }t[3000005];
    int read()
    {
    	int r=0,f=1;
    	char p=getchar();
    	while(p>'9'||p<'0')
    	{
    		if(p=='-')
    			f=-1;
    		p=getchar();
    	}
    	while(p>='0'&&p<='9')
    	{
    		r=r*10+p-48;
    		p=getchar();
    	}
    	return r*f;
    }
    inline bool cmp(const int &a,const int &b)
    {
    	return c[a]<c[b];
    }
    inline void add(int u,int v)
    {
    	cnt++;
    	e[cnt].ne=h[u];
    	e[cnt].to=v;
    	h[u]=cnt;
    }
    inline void dfs(int u)
    {
    	a[++dfn]=u;
    	l[u]=dfn;
    	for(int i=h[u];i;i=e[i].ne)
    		dfs(e[i].to);
    	r[u]=dfn;
    }
    void build(int &ro,int l,int r)
    {
    	ro=++tot;
    	t[ro].l=l,t[ro].r=r;
    	if(l==r)
    		return;
    	int mid=(l+r)>>1;
    	build(t[ro].ls,l,mid);
    	build(t[ro].rs,mid+1,r);
    }
    void ins(int &ro,int pr,int v,int w)
    {
    	ro=++tot;
    	t[ro]=t[pr];
    	t[ro].s++;
    	t[ro].sum+=w;
    	if(t[ro].l==t[ro].r)
    		return;
    	int mid=(t[ro].l+t[ro].r)>>1;
    	if(v<=mid)
    		ins(t[ro].ls,t[pr].ls,v,w);
    	else
    		ins(t[ro].rs,t[pr].rs,v,w);
    }
    int ques(int l,int r,long long k)
    {
    	if(t[l].l==t[l].r)
    	{
    		if(t[r].sum-t[l].sum!=0&&sk+t[r].sum-t[l].sum<=m)
    			return ss+1;
    		else
    			return ss;
    	}
    	long long now=t[t[r].ls].sum-t[t[l].ls].sum;
    	if(now>=k)
    		return ques(t[l].ls,t[r].ls,k);
    	else
    	{
    		ss+=t[t[r].ls].s-t[t[l].ls].s;
    		sk+=now;
    		return ques(t[l].rs,t[r].rs,k-now);
    	}
    }
    int main()
    {
    	n=read(),m=read();
    	for(int i=1;i<=n;i++)
    	{
    		int b=read();g[i]=i,c[i]=read(),d[i]=read();
    		add(b,i);
    	}
    	sort(g+1,g+1+n,cmp);
    	for(int i=1;i<=n;i++)
    		mp[g[i]]=i;
    	dfs(1);
    	build(rt[0],1,n);
    	for(int i=1;i<=n;i++)
    		ins(rt[i],rt[i-1],mp[a[i]],c[a[i]]);
    	for(int i=1;i<=n;i++)
    	{
    		ss=0,sk=0;
    		ans=max(ans,d[i]*ques(rt[l[i]-1],rt[r[i]],m));
    	}
    	printf("%lld
    ",ans);
    	return 0;
    }
    
  • 相关阅读:
    加法问题,编程入门课
    三.Python变量,常量,注释
    二、python介绍
    一. python 安装
    Syntax behind sorted(key=lambda :)
    Java动态数组
    Package name does not correspond to the file path (IntelliJ IDEA)
    理解Java构造器中的"this"
    The Constructor with No Arguments
    160229-01、web页面常用功能js实现
  • 原文地址:https://www.cnblogs.com/lokiii/p/8854211.html
Copyright © 2011-2022 走看看