萌新求调
查看原帖
萌新求调
339299
osfly楼主2024/8/13 16:43

思路是求出每个点到守卫的最短距离然后最大生成树然后树剖求链上最小值,25分,剩下全RE,不知道RE在哪里。

#include<bits/stdc++.h>
using namespace std;

#define int long long

const int N=2e6+10,M=5e6+10;
const int INF=5e16;

struct ldh
{
	int u,v,w;
}myx[M];

struct graph
{
	struct edge
	{
		int u,v,w,nxt;
	}e[M<<1];
	int head[N],tot;
	void add(int u,int v,int w)
	{
		e[++tot]=edge{u,v,w,head[u]};
		head[u]=tot;
	}
}g1,g2;

int n,m,k,q;

int dis[N],vis[N];
void dijkstra()
{
	priority_queue<pair<int,int> > q;
	for(int i=1;i<=n;i++) dis[i]=INF;
	q.push({0,0});
	while(!q.empty())
	{
		int u=q.top().second;
		q.pop();
		if(vis[u]) continue;
		vis[u]=1;
		for(int i=g1.head[u];i;i=g1.e[i].nxt)
		{
			int v=g1.e[i].v,w=g1.e[i].w;
			if(dis[v]>dis[u]+w)
			{
				dis[v]=dis[u]+w;
				if(!vis[v]) q.push({-dis[v],v});
			}
		}
	}
}

int f[N];
int find(int x)
{
	return f[x]==x?f[x]:f[x]=find(f[x]);
}
void kruskal()
{
	sort(myx+1,myx+1+m,[](ldh a,ldh b){return min(dis[a.u],dis[a.v])>min(dis[b.u],dis[b.v]);});
	int cnt=0;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=m;i++)
	{
		int fu=find(myx[i].u),fv=find(myx[i].v),w=min(dis[myx[i].u],dis[myx[i].v]);
		if(fu!=fv)
		{
			cnt++,f[fu]=fv;
			g2.add(myx[i].u,myx[i].v,w);
			g2.add(myx[i].v,myx[i].u,w);
		}
	}
}

int dep[N],fa[N],siz[N],son[N],a[N];
void dfs1(int u,int f)
{
	fa[u]=f,dep[u]=dep[f]+1,siz[u]=1;
	for(int i=g2.head[u];i;i=g2.e[i].nxt)
	{
		int v=g2.e[i].v,w=g2.e[i].w;
		if(v==f) continue;
		a[v]=w;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[v]>siz[son[u]]) son[u]=v;
	}
}

int top[N],w[N],dfn[N],cur;
void dfs2(int u,int t)
{
	top[u]=t,dfn[u]=++cur,w[cur]=a[u];
	if(!son[u]) return ;
	dfs2(son[u],t);
	for(int i=g2.head[u];i;i=g2.e[i].nxt)
	{
		int v=g2.e[i].v;
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	}
}

int st[N][30],lg[N];
void ST()
{
	lg[0]=-1;
	for(int i=1;i<=n;i++) lg[i]=lg[i>>1]+1,st[i][0]=w[i];
	for(int j=1;j<=20;j++)
		for(int i=1;i+(1<<j)-1<=n;i++)
			st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
int query(int l,int r)
{
	int p=lg[r-l+1];
	return min(st[l][p],st[r-(1<<p)+1][p]);
}

int qlink(int x,int y)
{
	int res=INF;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		res=min(res,query(dfn[top[x]],dfn[x]));
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	res=min(res,query(dfn[x]+1,dfn[y]));
	return res;
}

signed main()
{
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%lld%lld%lld",&myx[i].u,&myx[i].v,&myx[i].w);
		g1.add(myx[i].u,myx[i].v,myx[i].w);
		g1.add(myx[i].v,myx[i].u,myx[i].w);
	}
	scanf("%lld",&k);
	for(int i=1;i<=k;i++)
	{
		int x;
		scanf("%lld",&x);
		g1.add(0,x,0);
	}
	dijkstra(),kruskal(),dfs1(1,0),dfs2(1,1),ST();
	scanf("%lld",&q);
	while(q--)
	{
		int s,t,ans;
		scanf("%lld%lld",&s,&t);
		printf("%lld\n",max(0ll,qlink(s,t)-1));
	}
	return 0;
}
2024/8/13 16:43
加载中...