[APIO2012]派遣 洛谷P1552 bzoj2809 codevs1763

http://www.codevs.cn/problem/1763/

https://www.lydsy.com/JudgeOnline/problem.php?id=2809

https://www.luogu.org/problemnew/show/P1552

http://210.33.19.101/problem/2182

用线段树合并，值域线段树维护可重集合，线段树每个节点维护一个sum表示当前集合中，所有在范围[l,r]（这个点的值域区间）内的数的和；每个点从其子节点合并；查询就根据左子节点sum与当前最大允许的和决定往左还是往右走

话说稍微有点卡空间。。。（加了内存回收，然而理论上内存回收应该并不能改进最坏空间复杂度）

看了题解，发现只要暴力启发式合并/可并堆，合并完成后暴力删除最大值直到合法就行了。。。（因为每个点只会被删一次）

又想了想，只要暴力普通堆启发式合并就行了。。。。40行就行。。。

 #pragma GCC optimize(3)//不开优化可能被卡常
 #include<cstdio>
 #include<algorithm>
 #include<queue>
 #include<vector>
 #define pb push_back
 using namespace std;
 typedef long long LL;
 int n,m;
 vector<int> ch[];
 int c[],l[],fa[];
 LL sum[];
 priority_queue<int> q[];
 LL ans;
 void dfs(int u)
 {
     int i,v;
     q[u].push(c[u]);sum[u]+=c[u];
     for(i=;i<ch[u].size();i++)
     {
         v=ch[u][i];
         dfs(v);
         if(q[v].size()>q[u].size())    swap(q[u],q[v]);
         while(!q[v].empty())    q[u].push(q[v].top()),q[v].pop();
         sum[u]+=sum[v];
     }
     while(sum[u]>m)    sum[u]-=q[u].top(),q[u].pop();
     ans=max(ans,l[u]*LL(q[u].size()));
 }
 int main()
 {
     int i,a;
     scanf("%d%d",&n,&m);
     for(i=;i<=n;i++)
     {
         scanf("%d%d%d",&a,&c[i],&l[i]);
         fa[i]=a;ch[a].pb(i);
     }
     dfs();
     printf("%lld",ans);
     return ;
 }

空间因为pq里面用了vector之类的好像会被卡。。。可以改一下

 #pragma GCC optimize(3)//不开优化可能被卡常
 #include<cstdio>
 #include<algorithm>
 #include<queue>
 #include<vector>
 #define pb push_back
 using namespace std;
 typedef long long LL;
 int n,m;
 vector<int> ch[];
 int c[],l[],fa[];
 LL sum[];
 priority_queue<int> *q[];
 LL ans;
 void dfs(int u)
 {
     int i,v;q[u]=new priority_queue<int>();
     q[u]->push(c[u]);sum[u]+=c[u];
     for(i=;i<ch[u].size();i++)
     {
         v=ch[u][i];
         dfs(v);
         if(q[v]->size()>q[u]->size())    swap(q[u],q[v]);
         while(!q[v]->empty())    q[u]->push(q[v]->top()),q[v]->pop();
         sum[u]+=sum[v];delete q[v];
     }
     while(sum[u]>m)    sum[u]-=q[u]->top(),q[u]->pop();
     ans=max(ans,l[u]*LL(q[u]->size()));
 }
 int main()
 {
     int i,a;
     scanf("%d%d",&n,&m);
     for(i=;i<=n;i++)
     {
         scanf("%d%d%d",&a,&c[i],&l[i]);
         fa[i]=a;ch[a].pb(i);
     }
     dfs();
     printf("%lld",ans);
     return ;
 }

错误记录：答案没有开longlong所以无限WA

 #include<cstdio>
 #include<algorithm>
 #include<vector>
 #define pb push_back
 using namespace std;
 typedef long long LL;
 namespace S
 {
 #define mid (l+((r-l)>>1))
 #define N 3000000
 int dat[N],lc[N],rc[N];LL sum[N];
 int L;
 int st[N+],top;
 void init()
 {
     int i;
     for(i=;i<N;i++) st[++top]=i;
 }
 int getnode()
 {
     int t=st[top--];dat[t]=sum[t]=lc[t]=rc[t]=;
     return t;
 }
 void delnode(int x) {st[++top]=x;}
 void _addx(int l,int r,int &num)
 {
     if(!num)    num=getnode();
     if(l==r)    {dat[num]++;sum[num]+=l;return;}
     if(L<=mid)   _addx(l,mid,lc[num]);
     else    _addx(mid+,r,rc[num]);
     dat[num]=dat[lc[num]]+dat[rc[num]];
     sum[num]=sum[lc[num]]+sum[rc[num]];
 }
 void addx(int p,int &num)
 {
     L=p;_addx(,1e9,num);
 }
 int merge(int a,int b)
 {
     if(!a||!b)  return a+b;
     lc[a]=merge(lc[a],lc[b]);
     rc[a]=merge(rc[a],rc[b]);
     dat[a]+=dat[b];
     sum[a]+=sum[b];
     delnode(b);
     return a;
 }
 int find(int l,int r,int num,LL msum)
 {
     if(l==r)    return msum/l;
     if(sum[lc[num]]>=msum)   return find(l,mid,lc[num],msum);
     else    return find(mid+,r,rc[num],msum-sum[lc[num]])+dat[lc[num]];
 }
 #undef mid
 }
 vector<int> ch[];
 int n,m;
 int c[],l[],fa[];
 int rt[];LL ans;
 void dfs(int u)
 {
     S::addx(c[u],rt[u]);
     for(int i=,v;i<ch[u].size();i++)
     {
         v=ch[u][i];
         dfs(v);
         rt[u]=S::merge(rt[u],rt[v]);
     }
     ans=max(ans,LL(l[u])*S::find(,1e9,rt[u],m));
 }
 int main()
 {
     int i,a;S::init();
     scanf("%d%d",&n,&m);
     for(i=;i<=n;i++)
     {
         scanf("%d%d%d",&a,&c[i],&l[i]);
         fa[i]=a;ch[a].pb(i);
     }
     dfs();
     printf("%lld",ans);
     return ;
 }