80pts 求调
查看原帖
80pts 求调
361461
Creadron楼主2024/7/15 11:25

本人用的 LCA+二分做法,LCA 用倍增求。

交上去后#9,#10,#11,#13 TLE 了

经过调试我发现#9(n,m<=2000)我在自己机子上要跑20多秒,而且时间主要花在了求 LCA 上。

本人代码:

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=3e5+5,maxn_log=25+5;
int n,m,maxw,maxd,head[maxn],cntm=1,lg[maxn],dep[maxn],f[maxn][maxn_log],id[maxn],cnt,c[maxn],dis[maxn],w[maxn];
struct Edge{int u,v,w,next;} edge[maxn*2];
struct Path{int u,v,lca,w;} p[maxn];
int read(){
	int x=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-48; ch=getchar();}
	return x*f;
}
void add(int u,int v,int w){
	edge[cntm].u=u; edge[cntm].v=v; edge[cntm].w=w;
	edge[cntm].next=head[u]; head[u]=cntm++;
}
void getDepth(int x,int fa){
	f[x][0]=fa; dep[x]=dep[fa]+1; id[++cnt]=x;
	for(int i=1;i<=lg[dep[x]];i++)f[x][i]=f[f[x][i-1]][i-1];
	for(int i=head[x];i;i=edge[i].next)if(edge[i].v!=fa){
		int y=edge[i].v,z=edge[i].w;
		dis[y]=dis[x]+z; w[y]=z;
		getDepth(y,x);
	}
}
int getLCA(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	while(dep[x]>dep[y]){
		x=f[x][lg[dep[y]-dep[x]]];
	} if(x==y)return x;
	for(int i=lg[dep[x]]-1;i>=0;i--){
		if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	} return f[x][0];
}
bool check(int x){
	int tot=0;
	for(int i=0;i<=n;i++)c[i]=0;
	for(int i=1;i<=m;i++){
		if(p[i].w>x)tot++,c[p[i].u]++,c[p[i].v]++,c[p[i].lca]-=2;
	}
	for(int i=n;i>=1;i--)c[f[id[i]][0]]+=c[id[i]];
	for(int i=n;i>=1;i--){
		int t=id[i];
		if(c[t]==tot&&w[t]>=maxd-x)return true;
	} return false;
}
int search(int l,int r){
	int ans;
	while(l<=r){
		int mid=(l+r)>>1;
		if(check(mid))r=mid-1,ans=mid;
		else l=mid+1;
	} return ans;
}
signed main(){
//	freopen("P2680_9.in","r",stdin);
// 	freopen("P2680_9.out","w",stdout);
	n=read(); m=read();
	for(int i=1;i<=n;i++)lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	for(int i=1;i<n;i++){
		int u=read(),v=read(),w=read();
		add(u,v,w); add(v,u,w);
		maxw=max(maxw,w);
	} getDepth(1,0);
	for(int i=1;i<=m;i++){
		p[i].u=read(); p[i].v=read();
		p[i].lca=getLCA(p[i].u,p[i].v);
		p[i].w=dis[p[i].u]+dis[p[i].v]-dis[p[i].lca]*2;
		maxd=max(maxd,p[i].w);
	} cout<<search(maxd-maxw,maxd+1);
	return 0;
}

2024/7/15 11:25
加载中...