洛谷P2680运输计划——二分+lca+树上差分

题目：https://www.luogu.org/problem/P2680

分析：二分答案+倍增+树上差分 
题目实质：给定一棵树和一些点对，将一条边的权值变为0，使点对最大距离最小并求出最小距离。 
1、因为是求最大距离的最小值，所以很自然地想到二分答案法。 
2、怎样求点对距离？需要求出lca。求lca有三个方法：一是倍增；二是树链剖分；三是Tarjan。 
dfs求出树上每个点到根节点的距离d。对于点对(xi,yi)，dis[i]=d[xi]+d[yi]-2*d[lca(xi,yi)]。 
3、怎样二分答案，需要用到树上差分。

设当前二分出来可能答案为x。
首先从点对来看：对于dis[i]<=x，则忽略。对于dis[i]>x的点对，统计到num，并且需要删掉路径i上的一条边。 
接下来思考删除最长距离路径中的哪条边？从边来看，如果删掉一条边能使所有的点对距离都小于等于x，那么这条边的权值 w[i]>=max(dis[i])-x。因此要删除的边的权值满足 w[i]>=max(dis[i])-x。 
再接下来需要统计满足  w[i]>=max(dis[i])-x  的边在dis[i]>x的点对路径上的出现次数。这里用到树上差分（树上前缀和）。

对于点对(xi,yi)，设数组p，如果dis[i]>x，将p[xi]++, p[yi]++, p[lca(xi,yi)]–=2。

接着我们统计路径i上点t的关于数组p的前缀和qz[t]=sum( qt[ son[t] ] )。

qz[t]就是结点t和father[t]所连边出现的次数。 
（以后有时间再补上树上差分的图）

遍历，如果存在qz[t]>=num，说明找到了要去掉的边，二分check返回真，x值再往大调整，否则说明没有找到，则x往小调整。

先给出代码一（树上倍增求lca），noip2015官方第20个测试点TLE，采取了读入优化，耗时从1128ms降到1056ms

注意：浴谷数据水，可以全过，但用noip2015官方数据，则会卡一个点

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int MaxN=300005,MaxM=300005;
int n,m,cns,head[MaxN];
struct Edge{
	int nxt,to,val;
}edge[MaxN*2];
struct Road{
	int u,v,dis,lca;
}road[MaxM];
void add(int from,int to,int val){
	edge[++cns].nxt=head[from];
	edge[cns].to=to;
	edge[cns].val=val;
	head[from]=cns;
}
inline int get(){//读入优化 
    char c=getchar();
    int x=0;
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9'){
        x=(x<<3)+(x<<1)+c-'0';
        c=getchar();
    }
    return x;
}
int fa[MaxN],dep[MaxN],dis[MaxN],f[MaxN][21];
int cunt,num[MaxN];
void build(int x,int fx){
	fa[x]=fx;
	num[++cunt]=x;
	for(int i=1;i<=20;i++){//顺序。打表 
		f[x][i]=f[f[x][i-1]][i-1];
	}
	for(int i=head[x];i;i=edge[i].nxt){
		int to=edge[i].to;
		if(edge[i].to!=fx){
			f[to][0]=x;
			dep[to]=dep[x]+1;
			dis[to]=dis[x]+edge[i].val;
			build(to,x);
		}
	}		
}
int lca(int x,int y){
    if(dep[x]<dep[y]){//用x表示深度大的结点 
    	int tmp=x;x=y;y=tmp;
    }
    if(dep[x]>dep[y]){
    	for(int i=20;i>=0;i--){//逆序。查表 
	    	if(dep[ f[x][i] ] >= dep[y] ){//细节：有等号
	    		x=f[x][i];	
	    	}
	    }   
    }
	if(x==y)return x;//细节：重要
    for(int i=20;i>=0;i--)
        if(f[x][i]!=f[y][i]){
            x=f[x][i];
			y=f[y][i];
        }
    //细节：前面的for结束后x!=y
    return f[x][0];	
}
bool cmp(const Road &a,const Road &b){
	return a.dis>b.dis;
}
int temp[MaxN];
int ans;
bool check(int mid){
    int cnt=0,ans=0;
    memset(temp,0,sizeof(temp));
    for(int i=1;i<=m;i++){
        if(road[i].dis>mid){
            temp[road[i].u]++;temp[road[i].v]++;
			temp[road[i].lca]-=2;//树上差分           
            cnt++;//统计超过mid的路径条数 
        }
    }
    ans=road[1].dis-mid;
    if(cnt==0) return true;
    for(int i=n;i>=1;i--) temp[fa[num[i]]]+=temp[num[i]];//妙 
    for(int i=1;i<=n;i++) if(temp[i]==cnt&&dis[i]-dis[f[i][0]]>=ans) return true;
    return false;
}
int summ,x,y,z;
int main(){
	n=get();m=get();
	for(int i=1;i<n;i++){
		x=get();y=get();z=get();
		add(x,y,z);add(y,x,z);
	}
	dep[1]=1;dis[1]=0;
	build(1,0);
	for(int i=1;i<=m;i++){
		road[i].u=get();road[i].v=get();
		road[i].lca=lca(road[i].u,road[i].v);
		road[i].dis=dis[road[i].u]+dis[road[i].v]-2*dis[road[i].lca];
	}
	sort(road+1,road+m+1,cmp);
	int left=0,right=road[1].dis;
    int mid;
    while(left<right){
        mid=(left+right)>>1;
        if(check(mid)) right=mid;
        else left=mid+1;
    }
    printf("%d",left);
    return 0; 
}