题解:上树的超级钢琴 我们考虑dfs的时候直接由父亲转移而来 那么我们对于每个节点则维护是这点到根路径上的分布情况 那么转化为序列上的问题 这也就和超级钢琴本质上是一样的了
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
#include <set>
#include <map>
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
const int MAXN=5e5+10;
const double eps=1e-8;
#define ll long long
const ll inf=1e18;
using namespace std;
struct edge{int t;edge*next;}e[MAXN<<1],*h[MAXN],*o=e;
void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;}
ll read(){
ll x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return x*f;
}
typedef struct Node{
int id,l,r,pos;ll ans;
friend bool operator <(Node aa,Node bb){return aa.ans<bb.ans;}
}Node;
priority_queue<Node>que;
typedef struct node{
int l,r,pos;ll sum,flag;
}node;
node d[MAXN*31];
int cnt,n,m,L,R,dep[MAXN],rt[MAXN],a[MAXN];
void up(int x,int l,int r){
d[x].sum=d[d[x].l].sum+d[x].flag,d[x].pos=d[d[x].l].pos;
if(!d[x].pos)d[x].pos=l;
if(d[d[x].r].sum+d[x].flag>d[x].sum){d[x].sum=d[d[x].r].sum+d[x].flag,d[x].pos=d[d[x].r].pos;
if(!d[x].pos)d[x].pos=r;
}
}
void update(int &x,int y,int l,int r,int ql,int qr,int vul){
x=++cnt;d[x]=d[y];
if(!d[x].pos)d[x].pos=l;
if(ql<=l&&r<=qr){d[x].sum+=vul;d[x].flag+=vul;return ;}
int mid=(l+r)>>1;
if(ql<=mid)update(d[x].l,d[y].l,l,mid,ql,qr,vul);
if(qr>mid)update(d[x].r,d[y].r,mid+1,r,ql,qr,vul);
up(x,l,r);
}
ll maxx;int id;
void querty(int x,int l,int r,int ql,int qr,ll tag){
if(!x){
if(tag>maxx)maxx=tag,id=l;
return ;
}
if(ql<=l&&r<=qr){
if(d[x].sum+tag>maxx)maxx=d[x].sum+tag,id=d[x].pos;
return ;
}
int mid=(l+r)>>1;
if(ql<=mid)querty(d[x].l,l,mid,ql,qr,tag+d[x].flag);
if(qr>mid)querty(d[x].r,mid+1,r,ql,qr,tag+d[x].flag);
}
void dfs(int x,int deep,int fa){
dep[x]=deep+1;update(rt[x],rt[fa],1,n,1,dep[x],a[x]);
link(x){
if(j->t!=fa){
dfs(j->t,deep+1,x);
}
}
}
void slove(){
inc(i,1,n){
if(dep[i]<L)continue;
maxx=-inf;querty(rt[i],1,n,max(1,dep[i]-R+1),dep[i]-L+1,0);
que.push((Node){i,max(1,dep[i]-R+1),dep[i]-L+1,id,maxx});
}
ll ans=0;
inc(i,1,m){
Node t=que.top();que.pop();ans+=t.ans;
if(t.pos>t.l)maxx=-inf,querty(rt[t.id],1,n,t.l,t.pos-1,0),que.push((Node){t.id,t.l,t.pos-1,id,maxx});
if(t.pos<t.r)maxx=-inf,querty(rt[t.id],1,n,t.pos+1,t.r,0),que.push((Node){t.id,t.pos+1,t.r,id,maxx});
}
printf("%lld
",ans);
}
int main(){
n=read();int t;
inc(i,1,n)t=read(),add(i,t),add(t,i);
inc(i,1,n)a[i]=read();
m=read();L=read();R=read();
dfs(1,0,0);
slove();
}
4458: GTY的OJ
Time Limit: 40 Sec Memory Limit: 512 MBSubmit: 167 Solved: 72
[Submit][Status][Discuss]
Description
身为IOI金牌的gtyzs有自己的一个OJ，名曰GOJ。GOJ上的题目可谓是高质量而又经典，他在他的OJ里面定义了一个树形的分类目录，且两个相同级别的目录是不会重叠的。比如图论的大目录下可能分为最短路，最小生成树，网络流等低一级的分类目录，这些目录下可能还有更低一级的目录，以此类推。现在gtyzs接到了一个任务，要他为SDTSC出题。他准备在自己的OJ题库中找出M道题作为SDTSC的试题，而他深知SDTSC的童鞋们个个都是神犇，所以gtyzs认为自己出的这M道题中，每道题都应该属于不少于L种分类目录；可他又怕自己的题没有人会做，所以每道题也应该属于不超过R种分类目录，并且这些分类目录应该是连续的，不能出现断层。对了，最重要的是，不存在一道题，它属于两个分类目录且这两个目录不是包含关系。（比如不存在一道题它既是一道DP题，又是一道网络流题）gtyzs怕被骂，所以他希望不存在任意的一对题目(u,v)，使得u所属的分类目录与v完全相同。举例来说，gtyzs不能出两道同样的都是动态规划，背包的题，但是却可以出一道属于动态规划，背包和01背包的题和一道属于背包，01背包的题，当然也可以出一个属于图论的题和一个属于数论的题。（注意：一道题所属的分类目录不一定从根开始，如加粗部分所示）
为了让自己的题目变得更加靠谱，他给每一个分类目录都定了一个靠谱值ai，这个值可正可负。一道题的靠谱度为其从属的分类目录靠谱值的加和。我们假设动态规划的靠谱值为10，插头DP的靠谱值为－5，则一道动态规划插头DP的题的靠谱值就是5。gtyzs希望自己出的这M道题，在满足上述前提条件下，靠谱度总和最大。gtyzs当然会做啦，于是你就看到了这个题。
Input
Output
一行一个整数，表示最大的靠谱度。
Sample Input
0 1 1 2 2 3 3
2 3 4 1 2 3 4
3 3 3
Sample Output