[BZOJ 2500]幸福的道路 树形dp+单调队列+二分答案

时间:2022-01-22 21:17:22

考试的时候打了个树链剖分,而且还审错题了,以为是每天找所有点的最长路,原来是每天起点的树上最长路径再搞事情。。

先用dfs处理出来每个节点以他为根的子树的最长链和次长链。(后面会用到)

然后用类似dp的方法把每个节点的最长路径求出来。

下面是具体解释,请思考

以一个节点为例(w为它与父亲节点道路的权值)

一、如果它父亲节点的最长路径不过它

那么它最长路径等于它父亲最长路径+w(自己画图即可理解,往上走的)

它的次长路径等于它的最长链(只能往下走)

二、如果过它

那么它的最长路径有两种可能

①它的最长链

②它父亲的次长路径+w

如果①优,那么它最长路径为①,次长路径为max(它的次长链,②)

如果②优,那么它最长路径为②,次长路径为①

很好理解吧。。。

然后让求最大值与最小值差在m之内的最长子串。

据他们说有O(n)的做法,不过我打了O(nlogn)的打法,跑进了1s,反正O(能过)就行。

我的nlogn是二分答案+单调队列。二分答案去验证,挨个移动,看是否满足条件。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define pos(i,a,b) for(int i=(a);i<=(b);i++)
#define pos2(i,a,b) for(int i=(a);i>=(b);i--)
int n,m;
#define N 1001000
struct haha
{
	int to,next,w;
};
int read()
{
    int x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9')
        ch=getchar();
    while(ch>='0'&&ch<='9')
        x=x*10+ch-'0',ch=getchar();
    return x;
}
haha edge[N*4];
int head[N],cnt=1;
void add(int u,int v,int w)
{
	edge[cnt].w=w;
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
struct qian
{
	int fir,firch;      int sec,secch;
	int roadfir,roadsec;int roadfirch,roadsecch;
}cun[N];
int fa[N];
void dfs(int now)
{
	for(int i=head[now];i;i=edge[i].next)
	{
		int to=edge[i].to;
		int w=edge[i].w;
		dfs(to);
		if(cun[now].sec<cun[to].fir+w)
		{
			cun[now].sec=cun[to].fir+w;
			cun[now].secch=to;
			if(cun[now].sec>cun[now].fir)
			{
				swap(cun[now].sec,cun[now].fir);
				swap(cun[now].secch,cun[now].firch);
			}
		}
	}
}
void dp(int now,int w)
{
	if(now==1)
	{
		cun[now].roadfir=cun[now].fir;
		cun[now].roadsec=cun[now].sec;
		cun[now].roadfirch=cun[now].firch;
		cun[now].roadsecch=cun[now].secch;
	}
	else
	{
		if(cun[fa[now]].roadfirch!=now)
		{
			cun[now].roadfir=cun[fa[now]].roadfir+w;
			cun[now].roadfirch=fa[now];
			cun[now].roadsec=cun[now].fir;
		}
		else
		{
			if(cun[now].fir>cun[fa[now]].roadsec+w)
			{
				cun[now].roadfir=cun[now].fir;
				cun[now].roadfirch=cun[now].firch;
				cun[now].roadsec=max(cun[now].sec,cun[fa[now]].roadsec+w);
			}
			else
			{
				cun[now].roadfir=cun[fa[now]].roadsec+w;
				cun[now].roadfirch=fa[now];
				cun[now].roadsec=cun[now].fir;
			}
		}
	}
	for(int i=head[now];i;i=edge[i].next)
	{
		int to=edge[i].to;
		int ww=edge[i].w;
		dp(to,ww);
	}
}
int f[N];
int qmax[N],hed,tail;
int qmin[N],beg,ed;
bool find(int val){
    hed=tail=0;
    beg=ed=0;
    for(int i=1;i<=n;i++){
        while(hed<tail&&i-qmax[hed]>=val) hed++;
        while(beg<ed&&i-qmin[beg]>=val) beg++;
        while(hed<tail&&f[qmax[tail-1]]<f[i])   tail--;qmax[tail++]=i;
        while(beg<ed&&f[qmin[ed-1]]>f[i]) ed--;qmin[ed++]=i;
        if(i>=val&&f[qmax[hed]]-f[qmin[beg]]<=m) return true;
    }
    return false;
}
int main()
{
    //freopen("race.in","r",stdin);
    //freopen("race.out","w",stdout);
	n=read();m=read();
	pos(i,2,n)
	{
		int x,y;
		x=read();y=read();
		add(x,i,y);
		fa[i]=x;
	}
	dfs(1);
	dp(1,0);

    for(int i=1;i<=n;i++)
        cout<<cun[i].roadfir<<" "<<cun[i].roadsec<<endl;
	while(1);
    for(int i=1;i<=n;i++)
	   f[i]=cun[i].roadfir;
	int l=1,r=n+1;
	while(l<r-1){
        int mid=l+r>>1;
        if(find(mid))l=mid;
        else r=mid;
    }
    int l=1,r=n+1;
    while(l<r-1)
    {
        int mid=(l+r)>>1;
        if(find(mid))
          l=mid;
        else
          r=mid;
    }   

	cout<<l;
	//while(1);
	return 0;
}