萌新倍增 WA 25 pts 求调【悬关】
查看原帖
萌新倍增 WA 25 pts 求调【悬关】
637796
Xy_top楼主2023/8/7 22:08
#include<iostream>
#define int long long
using namespace std;
int n,m,k;
int d[300005],f[300005][19];
int dep[300005],pre[300005],sum[300005],val[300005];
struct Edge{int u,v,w,nxt;}a[600005];
struct Query{int l,u,v,w;}q[300005];
void add(int u,int v,int w){
	a[++k]={u,v,w,pre[u]};
	pre[u]=k;
}
void dfs(int u){
	for(int i=1;i<=18;i++)f[u][i]=f[f[u][i-1]][i-1];
	for(int i=pre[u];i;i=a[i].nxt){
		int v=a[i].v;
		if(v==f[u][0])continue;
		dep[v]=dep[u]+1;
		f[v][0]=u;
		sum[v]=sum[u]+a[i].w;
		dfs(v);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=18;i>=0;i--)if(dep[f[x][i]]>=dep[y])x=f[x][i];
	if(x==y)return x;
	for(int i=18;i>=0;i--)if(f[x][i]!=f[y][i]){
		x=f[x][i];
		y=f[y][i];
	}
	return f[x][0];
}
void fun(int u){
	for(int i=pre[u];i;i=a[i].nxt){
		int v=a[i].v;
		if(v==f[u][0])continue;
		fun(v);
		d[u]+=d[v];
	}
}
int jump(int x,int d){
	for(int i=18;i>=0;i--)if(dep[f[x][i]]>=d)x=f[x][i];
	return x;
}
bool check(int x){
	for(int i=1;i<=n;i++)d[i]=0;
	for(int i=1;i<=m;i++){
		if(q[i].w>x){
			d[q[i].u]++;
			d[q[i].v]++;
			d[q[i].l]-=2;
		}
	}
	fun(1);
	int mx=0,node=0;
	for(int i=1;i<=n;i++){
		if(d[i]){
			if(mx<val[i]){
				mx=val[i];
				node=i;
			}
			mx=max(mx,val[i]);
		}
	}
	int mxval=0;
	if(mx){
		for(int i=1;i<=m;i++){
			if(dep[node]>dep[q[i].l]&&(jump(q[i].u,dep[node])==node||jump(q[i].v,dep[node])==node))mxval=max(mxval,q[i].w-val[node]);
			else mxval=max(mxval,q[i].w);
		}
	}else return true;
	return mxval<=x;
}
signed main(){
	dep[1]=1;
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<n;i++){
		int u,v,w;
		scanf("%lld%lld%lld",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1);
	for(int i=1;i<=m;i++){
		scanf("%lld%lld",&q[i].u,&q[i].v);
		q[i].l=lca(q[i].u,q[i].v);
		q[i].w=sum[q[i].u]+sum[q[i].v]-2*sum[q[i].l];
	}
	for(int i=1;i<=k;i++){
		if(f[a[i].u][0]==a[i].v)val[a[i].u]=a[i].w;
		else val[a[i].v]=a[i].w;
	}
	int l=0,r=90000000000,mid;
	while(l<=r){
		mid=l+r>>1;
		if(check(mid))r=mid-1;
		else l=mid+1;
	}
	cout<<r+1;
	return 0;
}
2023/8/7 22:08
加载中...