题解链接:https://blog.csdn.net/shiqi_614/article/details/8105149
用树形dp是超时的,,
/* 先求出每个点可以跑的最长距离dp[i][0|1]表示向下|上可以跑到最大距离 再用单调队列求最大连续天数 */ #include<iostream> #include<cstring> #include<cstdio> using namespace std; #define maxn 1000005 #define ll long long struct Edge{int to,nxt,w;}edge[maxn<<1]; int dp[maxn][2],head[maxn],tot,n,m; void init(){ memset(head,-1,sizeof head); tot=0; } void addedge(int u,int v,int w){ edge[tot].to=v;edge[tot].w=w; edge[tot].nxt=head[u];head[u]=tot++; } void dfs0(int u,int pre){ dp[u][0]=0; for(int i=head[u];i!=-1;i=edge[i].nxt){ int v=edge[i].to; if(v!=pre){ dfs0(v,u); dp[u][0]=max(dp[u][0],dp[v][0]+edge[i].w); } } } void dfs1(int u,int pre){ long long Max=0,Sec=0,v1,v2;//u的大儿子下标,二儿子下标 for(int i=head[u];i!=-1;i=edge[i].nxt){//这个循环处理出u的大儿子 int v=edge[i].to; if(v==pre)continue; int tmp=edge[i].w+dp[v][0]; if(tmp>Max){//找到了更大的儿子树 Sec=Max,Max=tmp; v2=v1,v1=v; } else if(tmp==Max || tmp>Sec) Sec=tmp,v2=v; } //printf("%d %d\n",u,Max); if(u!=-1){//和u的上面进行比较 long long tmp=dp[u][1],v=-1; if(tmp>Max){//找到了更大的儿子树 Sec=Max,Max=tmp; v2=v1,v1=v; } else if(tmp==Max || tmp>Sec) Sec=tmp,v2=v; } for(int i=head[u];i!=-1;i=edge[i].nxt){//这个循环求dp进行递归 int v=edge[i].to; if(v==pre)continue; if(v==v1)dp[v][1]=Sec+edge[i].w; else dp[v][1]=Max+edge[i].w; dfs1(v,u); } //printf("%d %d\n",u,Max); } int d[maxn],qmax[maxn],qmin[maxn]; void solve() { int ans=0,i,j,front1,front2,rear1,rear2; front1=rear1=0; front2=rear2=0; for(int i=1;i<=n;i++)d[i]=max(dp[i][0],dp[i][1]); for(i=1,j=1;j<=n;j++) { while(rear1>front1&&d[qmax[rear1-1]]<=d[j]) rear1--; qmax[rear1++]=j; while(rear2>front2&&d[qmin[rear2-1]]>=d[j]) rear2--; qmin[rear2++]=j; if(d[qmax[front1]]-d[qmin[front2]]>m) { ans=max(ans,j-i); while(d[qmax[front1]]-d[qmin[front2]]>m) { i=min(qmax[front1],qmin[front2])+1; while(rear1>front1&&qmax[front1]<i) front1++; while(rear2>front2&&qmin[front2]<i) front2++; } } } ans=max(ans,j-i); printf("%d\n",ans); } int main(){ while(cin>>n>>m){ init(); int u,v,w; for(int v=2;v<=n;v++){ cin>>u>>w; addedge(u,v,w); addedge(v,u,w); } dfs0(1,0);//先求dp[i][0] dfs1(1,0);//再求dp[i][1] solve(); } }