BZOJ - 2809 dispatching 主席树+dfs序

在一个忍者的帮派里,一些忍者们被选中派遣给顾客,然后依据自己的工作获取报偿。在这个帮派里,有一名忍者被称之为 Master。除了 Master以外,每名忍者都有且仅有一个上级。为保密,同时增强忍者们的领导力,所有与他们工作相关的指令总是由上级发送给他的直接下属,而不允许通过其他的方式发送。现在你要招募一批忍者,并把它们派遣给顾客。你需要为每个被派遣的忍者 支付一定的薪水,同时使得支付的薪水总额不超过你的预算。另外,为了发送指令,你需要选择一名忍者作为管理者,要求这个管理者可以向所有被派遣的忍者 发送指令,在发送指令时,任何忍者(不管是否被派遣)都可以作为消息的传递 人。管理者自己可以被派遣,也可以不被派遣。当然,如果管理者没有被排遣,就不需要支付管理者的薪水。你的目标是在预算内使顾客的满意度最大。这里定义顾客的满意度为派遣的忍者总数乘以管理者的领导力水平,其中每个忍者的领导力水平也是一定的。写一个程序,给定每一个忍者 i的上级 Bi,薪水Ci,领导力L i,以及支付给忍者们的薪水总预算 M,输出在预算内满足上述要求时顾客满意度的最大值。
 
$1≤N≤100000$ 忍者的个数;
$1≤M≤1000000000$ 薪水总预算; 
$0≤B_i < i$ 忍者的上级的编号;
$1≤C_i≤M$   忍者的薪水;
$1≤L_i≤1000000000$   忍者的领导力水平。
 
题解:
  问题等价于,每一个点最多能在子树上取多少点,花费小于M?
  显然的是,我们的策略一定是从最小的点开始取,直到超过$M$为止,这显然是贪心的过程
  然后得枚举所有点,总复杂度$O(n^2)$
  然后考虑优化,
  我们把问题"询问一个点,其子树上之和大于$M$的$x$个权值最小的点"
  转化成"询问一个区间,查询一个区间上和小于等于$M$的$x$个最小的元素"
  显然,后者是主席树可以解决的,单次操作的复杂度就从$O(n)$变成了$O(log(n))$
  然后我们用$dfs$序将树拍扁,变成序列即可解决问题,就完成了优化
  具体过程就是
  $dfs(master)$得到$dfs$序,离散化花费
  主席树保存两个值,一个是点数,一个是花费,单点更新,区间求和
  按照$dfs$序列插入权值,构造n颗树
  最后,遍历每个点,每个点等价于一个区间询问,枚举所有点就能得到最大值
  (这道题网上的主席树题解代码真是乱写,反正我是没懂他们是怎么写的,AC是AC了,莫名其妙多此一举的操作一大堆)
#include <bits/stdc++.h>
#define nd seg[now]
#define ndp seg[pre]
#define mid ((s+t)>>1)
#define ll long long
using namespace std;
const int maxn=1e5+10;
const int maxm=1e6+10;
const int INF=0x3f3f3f3f;
int casn,n,k;
ll m;
struct node2{
	int to,next;
}e[maxn];
int	head[maxn],nume;
int tin[maxn],tout[maxn];
ll cost[maxn],pos[maxn],power[maxn];
int rt[maxn],size,cnt,pre[maxn],dfn[maxn];
int cmp(int a,int b){
	return cost[a]<cost[b];
}
inline void add(int a,int b){
	e[++nume]=(node2){b,head[a]};
	head[a]=nume;
}
void dfs(int now){
	tin[now]=++cnt;
	dfn[cnt]=now;
	for(int i=head[now];i;i=e[i].next){
		dfs(e[i].to);
	}
	tout[now]=cnt;
}
struct node{
	int l,r;ll sum,cnt;
}seg[maxn*20];
void maketree(int s=1,int t=n,int &now=rt[0]){
	now=++size;nd=(node){s,t,0,0};
	if(s==t) return ;
	maketree(s,mid,nd.l);maketree(mid+1,t,nd.r);
}
void update(int &now,int pre,int k,ll cost,int s=1,int t=n){
	now=++size;nd=ndp,nd.sum+=cost,nd.cnt++;
	if(s==t) return ;
	if(k<=mid)update(nd.l,ndp.l,k,cost,s,mid);
	else update(nd.r,ndp.r,k,cost,mid+1,t);
}
ll query(int ndl,int ndr,ll k,int s=1,int t=n){
	if(seg[ndr].sum-seg[ndl].sum<=k) return seg[ndr].cnt-seg[ndl].cnt;
	if(s==t) return min(seg[ndr].cnt-seg[ndl].cnt,k/pos[s]);
	ll sum=seg[seg[ndr].l].sum-seg[seg[ndl].l].sum;
	if(k>=sum) return query(seg[ndl].r,seg[ndr].r,k-sum,mid+1,t)+seg[seg[ndr].l].cnt-seg[seg[ndl].l].cnt;
	else return query(seg[ndl].l,seg[ndr].l,k,s,mid);
}
#undef mid
int main(){
	scanf("%d%lld",&k,&m);
	int master;
	for(int i=1;i<=k;i++){
		scanf("%d%lld%lld",pre+i,cost+i,power+i);
		if(pre[i]==0)master=i;
		else add(pre[i],i);
		pos[i]=cost[i];
	}
	sort(pos+1,pos+1+k);
	n=unique(pos+1,pos+1+k)-(pos+1);
	dfs(master);
	maketree();
	for(int i=1;i<=k;i++){
		int id=lower_bound(pos+1,pos+1+n,cost[dfn[i]])-pos;
		update(rt[i],rt[i-1],id,cost[dfn[i]]);
	}
	ll ans=0;
	for(int i=1;i<=k;i++){
		ans=max(ans,power[i]*query(rt[tin[i]-1],rt[tout[i]],m));
	}
	printf("%lld\n",ans);
  return 0;
}

  

 

 
  
posted @ 2018-06-20 19:04  nervending  阅读(180)  评论(0编辑  收藏  举报