【运输计划】【二分】【树上倍增】

---

又来重新写了一遍运输计划
不得不说noip2015的分量还是很重的orz
比上次写多了5分，可是最后一个测试点怎么也得要2.4s不开心（打滚qwq）

---

贴代码吧~

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=3e5+5;
int n,m,plan[N][4],a,b,c,fa[N][20],wor[N][20],dp[N],ma;
int head[N],next[N*2],tov[N*2],tot,zhi[N*2];
int tr[N],al,flag;
bool vis[N],yes;
void build(int x,int y,int z)
{tot++;next[tot]=head[x];tov[tot]=y;head[x]=tot;zhi[tot]=z;}
void dfs(int x) //预遍历 
{
    vis[x]=true;
    int u=head[x],v=tov[u];
    while(u){
        if(!vis[v])
        {
            dp[v]=dp[x]+1;dfs(v);
            wor[v][0]=zhi[u];
            fa[v][0]=x;
        }
        u=next[u],v=tov[u];
    }
}
void LCA(int k,int x,int y)//求lca 
{
    if(dp[x]<dp[y])swap(x,y);
    int dis=dp[x]-dp[y],hh=0;
    for(int i=19;i>=0;i--)
        if(dis&(1<<i)){hh+=wor[x][i];x=fa[x][i];}
    if(x==y)
    {
        plan[k][0]=x;plan[k][3]=hh;
        return;
    }
    for(int i=19;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
        {hh+=wor[x][i]+wor[y][i];x=fa[x][i];y=fa[y][i];}
    plan[k][0]=fa[x][0];
    plan[k][3]=hh+wor[x][0]+wor[y][0];
}
void pre() //预main 
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        build(a,b,c);build(b,a,c);
    }
    dp[1]=1;dfs(1);
    for(int i=1;i<20;i++)
        for(int j=1;j<=n;j++)
        {fa[j][i]=fa[fa[j][i-1]][i-1];wor[j][i]=wor[fa[j][i-1]][i-1]+wor[j][i-1];}
    for(int i=1;i<=m;i++)  //处理出所有线路的相关量 
    {
        scanf("%d%d",&plan[i][1],&plan[i][2]);
        LCA(i,plan[i][1],plan[i][2]);
        ma=max(ma,plan[i][3]);
    }
}
void dfs2(int x,int fa) //check 
{
    if(yes)return;
    int u=head[x],v=tov[u];
    while(u){
        if(v!=fa)
        {
            dfs2(v,x);
            tr[x]+=tr[v];
        }
        u=next[u],v=tov[u];
    }
    if(tr[x]==al&&wor[x][0]>=flag)yes=true;
}
bool check(int x)
{
    memset(tr,0,sizeof(tr));al=0;yes=false;
    for(int i=1;i<=m;i++)
    if(plan[i][3]>x)
    {al++;tr[plan[i][1]]++;tr[plan[i][2]]++;tr[plan[i][0]]-=2;}
    flag=ma-x;dfs2(1,1);
    return yes;
}
void worrk()
{
    int le=0,ri=ma,mid=ma;
    while(le<ri)
    {
        mid=(le+ri)>>1;
        if(check(mid))ri=mid;
        else le=mid+1;
    }
    if(le==ri)mid=le;
    printf("%d\n",mid);
}
int main()
{
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
    pre();
    worrk();
}