NOIP提高组 树上路径

Description

给出一棵树，求出k，使k最小，且在树中存在路径p，使得k>=S且k<=E。(k为路径p上的边的权值和)，若无解输出-1。

Data Constraint

树中点的数量<=$10^5$,|E-S|<=10^6

对于所有数据满足1<=Wi<=1000,|E|、|S|<=10^9

Solution

我们考虑一条可能路径在以i为根的树中的情况。

1、这条路径不经过点i。

这种情况我们在往i的子树处理时就会考虑到，所以对答案并没有影响。

2、这条路径的一个端点是i。

这种情况我们维护一个队列d，表示以i为根的子树的每个点到i的距离。假设某个点到i的距离>=S且<=E，那么我们更新答案，并将点到i的距离>=E的点踢出队列。

3、路径的两个端点分别在i的不同子树中。

假设现在根为0，根连向的点依次是1、2、3，现在我遍历了1、2两颗子树，刚遍历了3子树，要将1、2和3子树的信息合并，显然在遍历2后我就已经将1和2合并了，所以现在1、2是在一个有序队列（设为f），3在另一个（设为g）。对于f中每一个值x，我们都要找一个g中找一个值y，使x+y>=s且x+y尽量小。所以我们在g中维护一个指针j，j刚开始为队尾。然后顺序枚举f，j指针往前找到一个与x相加

所以，这个算法的复杂度理论上为O($N^2$),但很多不会到达理论上界，除了像菊花图这种图，实际情况中表现良好。

代码

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=200005;
int first[maxn],last[maxn],x,y,z,next[maxn],num,size[maxn],mx[maxn];
int n,i,t,j,k,l;
ll ans,value[maxn],d[maxn],b[maxn],s,e;
void lian(int x,int y,ll z){
    last[++num]=y;value[num]=z;next[num]=first[x];first[x]=num;
}
void dg1(int x,int y){
    int t;size[x]++;
    for (t=first[x];t;t=next[t])
        if (last[t]!=y){
            dg1(last[t],x);size[x]+=size[last[t]];mx[x]=max(mx[x],size[last[t]]);
        }
    mx[x]=max(mx[x],n-size[x]);
}
void qsort(int l,int mid,int r){
    int i,j,t,k;
    for (i=l;i<=r;i++)
        b[i]=d[i];
    i=l;j=mid+1;t=l-1;
    while (i<=mid && j<=r)
        if (b[i]<b[j]) d[++t]=b[i++];
        else d[++t]=b[j++]; 
    while (i<=mid) d[++t]=b[i++];
    while (j<=r) d[++t]=b[j++];
}
void dg(int x,int y){
    d[++d[0]]=0;
    int t,k=d[0]+1,l,i,j;
    for (t=first[x];t;t=next[t])
        if (last[t]!=y){
            l=d[0]+1;
            dg(last[t],x);
            for (j=l;j<=d[0];j++){
                d[j]+=value[t];
                if (d[j]>=s) ans=min(ans,d[j]);
                if (d[j]>e){
                    d[0]=j-1;
                    break;
                }
            }
            j=d[0];
            if (j<l)continue;
            for (i=k;i<l;i++){
                while (d[i]+d[j]>=s && j>l) j--;
                if (d[i]+d[j]<s && j<d[0]) j++;
                if (d[i]+d[j]>=s) ans=min(ans,d[i]+d[j]);
                if (j==l && d[i]+d[j]>=s) break;
            }
            if (k<l) qsort(k,l-1,d[0]);
        }
}
int main(){
//  freopen("data.in","r",stdin);
    scanf("%d%d%d",&n,&s,&e);
    for (i=1;i<n;i++)
        scanf("%d%d%d",&x,&y,&z),lian(x,y,z),lian(y,x,z);
    dg1(1,0);
    ans=e+1;
    for (i=1;i<=n;i++)
        if (mx[i]<=n/2){
            dg(i,0);
            break;
        }
    if (ans!=e+1) printf("%lld\n",ans);
    else printf("-1");
}