bzoj2500幸福的道路 树形dp+单调队列

2500: 幸福的道路

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 434  Solved: 170
[Submit][Status][Discuss]

Description

 
小T与小L终于决定走在一起,他们不想浪费在一起的每一分每一秒,所以他们决定每天早上一同晨练来享受在一起的时光.
他们画出了晨练路线的草图,眼尖的小T发现可以用树来描绘这个草图.
他们不愿枯燥的每天从同一个地方开始他们的锻炼,所以他们准备给起点标号后顺序地从每个起点开始(第一天从起点一开始,第二天从起点二开始……). 而且他们给每条道路定上一个幸福的值.很显然他们每次出发都想走幸福值和最长的路线(即从起点到树上的某一点路径中最长的一条).
他们不愿再经历之前的大起大落,所以决定连续几天的幸福值波动不能超过M(即一段连续的区间并且区间的最大值最小值之差不超过M).他们想知道要是这样的话他们最多能连续锻炼多少天(hint:不一定从第一天一直开始连续锻炼)?
现在,他们把这个艰巨的任务交给你了!

Input

第一行包含两个整数N, M(M<=10^9).
第二至第N行,每行两个数字Fi , Di, 第i行表示第i个节点的父亲是Fi,且道路的幸福值是Di.

Output

最长的连续锻炼天数

Sample Input

3 2
1 1
1 3

Sample Output

3
数据范围:
50%的数据N<=1000
80%的数据N<=100 000
100%的数据N<=1000 000

这其实是两个题强行合在一起啊。。
首先是对于每个节点求它在树上的最长路,可以树形dp 2次,一次从儿子转移一次从父亲转移
求最长连续区间的话维护两个单调队列max min就好

 #include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define N 1000005
#define ll long long
using namespace std;
int n,m,tot,w[N],hd[N],ans=;
ll f[N],g[N],a[N];int mx[N],mn[N];
struct edge{int v,w,next;}e[N<<];
char gc(){
static char s[],*p1,*p2;
if(p1==p2)p2=(p1=s)+fread(s,,,stdin);
if(p1==p2)return EOF;
return *p1++;
}
int read(){
int x=;char ch=gc();
while(ch>''||ch<'')ch=gc();
while(ch<=''&&ch>='')x=x*+ch-'',ch=gc();
return x;
}
void adde(int u,int v,int w){
e[++tot].v=v;
e[tot].next=hd[u];
e[tot].w=w;
hd[u]=tot;
}
void dfs1(int u,int fa){
for(int i=hd[u];i;i=e[i].next){
int v=e[i].v;
if(v==fa)continue;
dfs1(v,u);
f[u]=max(f[u],f[v]+e[i].w);
}
}
void dfs2(int u,int fa){
ll mx1=,mx2=;
for(int i=hd[u];i;i=e[i].next){
int v=e[i].v;if(v==fa)continue;
if(f[v]+e[i].w>mx1)mx2=mx1,mx1=f[v]+e[i].w;
else if(f[v]+e[i].w>mx2)mx2=f[v]+e[i].w;
g[v]=g[u]+e[i].w;
}
for(int i=hd[u];i;i=e[i].next){
int v=e[i].v;if(v==fa)continue;
if(f[v]+e[i].w==mx1)g[v]=max(g[v],mx2+e[i].w);
else if(mx1+e[i].w>g[v])g[v]=mx1+e[i].w;
dfs2(v,u);
}
}
void solve(){
for(register int i=;i<=n;i++)a[i]=max(g[i],f[i]);
int l1=,l2=,r2=,r1=,t=;
for(register int i=;i<=n;i++){
while(l1<=r1&&a[i]>=a[mx[r1]])r1--;
while(l2<=r2&&a[i]<=a[mn[r2]])r2--;
mx[++r1]=i;mn[++r2]=i;
while(a[mx[l1]]-a[mn[l2]]>m){
if(mx[l1]<=mn[l2])t=mx[l1]+,l1++;
else t=mn[l2]+,l2++;
}
ans=max(ans,i-t+);
}
}
int main(){
n=read();m=read();
for(register int i=;i<=n;i++){
int v=read(),w=read();
adde(i,v,w);adde(v,i,w);
}dfs1(,);dfs2(,);
solve();printf("%d",ans);
return ;
}
上一篇:使用xUnit为.net core程序进行单元测试(1)


下一篇:PAT (Advanced Level) Practice(更新中)