2809: [Apio2012]dispatching

最新推荐文章于 2019-03-10 10:48:32 发布

MatouKariya

最新推荐文章于 2019-03-10 10:48:32 发布

阅读量631

点赞数

文章标签：左偏树

本文链接：https://blog.csdn.net/MatouKariya/article/details/45644675

版权

题意：给你一棵树，每个点有点权Ci,Li,选取尽量多的节点，使得SUM CI<=m,且最大化Lroot*size。

首先是枚举选取的子树的根，然后自然的想法是将它的所有子孙从小到大取，直到sum超过m。然而这样的做法会超时，所以我们考虑树形DP的思想，用儿子更新父亲，维护一个可并大根堆，将儿子和父亲的堆进行合并，如果sum超过m就删除堆顶，因为删除复杂度均摊，所以总复杂度是O（nlogn).
我这么弱，哪会写配对堆（Orz absi2011)，所以就写了个左偏树。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=100000+10;
vector<int> g[maxn];
int c[maxn],L[maxn],n,root[maxn],v[maxn],sz[maxn],l[maxn],r[maxn],dis[maxn],m,tot,bh[maxn];
long long sum[maxn],ans;
int top(int x)
{
  return v[x];
}
int merge(int x,int y)
{
  //cout<<x<<' '<<y<<endl;
  if(!x||!y) return x+y;
  if(v[x]<v[y]) swap(x,y);
  r[x]=merge(r[x],y);
  if(dis[l[x]]<dis[r[x]]) swap(l[x],r[x]);
  dis[x]=dis[r[x]]+1;
  return x;
}
void pop(int &x)
{
  x=merge(l[x],r[x]);
}
void dfs(int p)
{
  root[++tot]=p;bh[p]=tot;v[p]=c[p];dis[p]=l[p]=r[p]=0;sz[tot]=1;sum[tot]=c[p];
  for(int i=0;i<g[p].size();i++)
  {
    int v=g[p][i];
    dfs(v);sum[bh[p]]+=sum[bh[v]];sz[bh[p]]+=sz[bh[v]];
    root[bh[p]]=merge(root[bh[p]],root[bh[v]]);
  }
  while(sum[bh[p]]>m)
  {
    sum[bh[p]]-=top(root[bh[p]]);pop(root[bh[p]]);
    sz[bh[p]]--;
  }
  ans=max(ans,(long long)L[p]*sz[bh[p]]);
}
int main()
{
  //freopen("2809.in","r",stdin);
  //freopen("2809.out","w",stdout);
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++)
  {
    int x;scanf("%d%d%d",&x,&c[i],&L[i]);
    if(x) g[x].push_back(i);
  }
  dfs(1);
  printf("%lld\n",ans);
  return 0;
}