题目梗概
给出一棵有边权的树,并有m个从x到y的路径。
将一条边的边权改为0,使所有路径的最大值最小。
解题思路
因为要使最大值最小,所以我们二分答案。
对于大于答案的路径,我们要修改它。
那么就是把这些路径的交集中最大的边改为0。
求路径交集差分一下就可以了,在差分的时候只需要LCA就可以了。
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline char nc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
if (l==r) return EOF;return *l++;
}
inline int _read(){
int num=0;char ch=nc();
while(ch<'0'||ch>'9') ch=nc();
while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=nc();
return num;
}
const int maxn=300005;
int tot,son[maxn*2],w[maxn*2],nxt[maxn*2],lnk[maxn],lg;
int dis[maxn],f[maxn][20],n,m,dep[maxn],X[maxn],Y[maxn],c[maxn],fa[maxn],L,R,mid,tag[maxn],sum,num;
inline void add(int x,int y,int z){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;w[tot]=z;}
inline int max(int x,int y){if (x>y) return x;return y;}
void DFS(int x,int fa){
dep[x]=dep[fa]+1;f[x][0]=fa;
for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa){
dis[son[j]]=dis[x]+w[j];
DFS(son[j],x);
}
}
void work(){
for (int j=1;j<=lg;j++)
for (int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
}
int LCA(int x,int y){
if (dep[x]<dep[y]) swap(x,y);
for (int i=lg;i>=0;i--) if (dep[f[x][i]]>=dep[y]) x=f[x][i];
if (x==y) return x;
for (int i=lg;i>=0;i--) if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}
void query(int x,int fa,int lst){
for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa) query(son[j],x,w[j]),tag[x]+=tag[son[j]];
if (tag[x]==sum) num=max(num,lst);
}
bool check(int x){
memset(tag,0,sizeof(tag));int now=0;sum=0;num=0;
for (int i=1;i<=m;i++)if (c[i]>x) tag[X[i]]++,tag[Y[i]]++,tag[fa[i]]-=2,now=max(now,c[i]),sum++;
query(1,0,0);
return now-num<=x;
}
int main(){
freopen("exam.in","r",stdin);
freopen("exam.out","w",stdout);
n=_read();m=_read();lg=log2(n);
for (int i=1;i<n;i++){int x=_read(),y=_read(),z=_read();add(x,y,z);add(y,x,z);}
DFS(1,0);work();
for (int i=1;i<=m;i++){
X[i]=_read();Y[i]=_read();
fa[i]=LCA(X[i],Y[i]);c[i]=dis[X[i]]+dis[Y[i]]-2*dis[fa[i]];
R=max(R,c[i]);
}
while(L<=R){
mid=L+(R-L>>1);
if (check(mid)) R=mid-1;else L=mid+1;
}
printf("%d\n",L);
return 0;
}