题意给出n-1条边,每个点有一个最远距离,为最长的一段区间,这个区间内的最远距离最大值和最小值的差不超过m。 这道题我最先跑最远距离用的是取子结点和父亲节点过来的最大值的方法,但是一直超时,我想应该是用的vector存边导致的吧(方便坑人。。)后来用另外一种方法求。求出最远距离后就是两个指针一起跑,然后要得到这两个指针区间的最大值和最小值,可以用线段树也可以用单调队列,我用的单调队列,代码如下
#include<stdio.h> #include<string.h> #include<iostream> #include<algorithm> #define INF 1e9 using namespace std; const int MAXN=1e6+10; struct node { int to,next,w; }s[MAXN*2]; int head[MAXN],cnt,dp[MAXN],visit[MAXN],n,m,qmin[MAXN],qmax[MAXN]; void init() { memset(head,-1,sizeof(head)); memset(visit,0,sizeof(visit)); memset(dp,0,sizeof(dp)); cnt=0; } void add(int a,int b,int w) { s[cnt].to=b; s[cnt].w=w; s[cnt].next=head[a]; head[a]=cnt++; s[cnt].to=a; s[cnt].w=w; s[cnt].next=head[b]; head[b]=cnt++; } void dfs(int u,int fa) { // printf("%d %d %d\n",u,dp[u],sum); for(int i=head[u];i+1;i=s[i].next) { if(fa!=s[i].to) { dp[s[i].to]=max(dp[u]+s[i].w,dp[s[i].to]); dfs(s[i].to,u); } } // printf("!%d ~",u); } void solve() { int ans=0,front1,front2,real1,real2,i,j; front1=front2=0; real1=real2=0; for(i=1,j=1;j<=n;j++) { while(real1>front1&&dp[qmin[real1-1]]>=dp[j]) real1--; qmin[real1++]=j; while(real2>front2&&dp[qmax[real2-1]]<=dp[j]) real2--; qmax[real2++]=j; if(dp[qmax[front2]]-dp[qmin[front1]]<=m) { ans=max(ans,j-i+1); } else { i++; while(qmax[front2]<i) front2++; while(qmin[front1]<i) front1++; } } printf("%d\n",ans); } int main() { int i,j; while(~scanf("%d %d",&n,&m)) { int a,b; init(); for(i=2;i<=n;i++) { scanf("%d %d",&a,&b); add(i,a,b); } dfs(1,-1); cnt=-1; int k=0; for(i=1;i<=n;i++) { if(dp[i]>cnt) { cnt=dp[i]; k=i; } } memset(dp,0,sizeof(dp)); dfs(k,-1); cnt=k=-1; for(i=1;i<=n;i++) { if(dp[i]>cnt) { cnt=dp[i]; k=i; } visit[i]=dp[i]; } memset(dp,0,sizeof(dp)); dfs(k,-1); for(i=1;i<=n;i++) dp[i]=max(visit[i],dp[i]); solve(); } }