BZOJ2125: 最短路【圆方树+仙人掌】

2125: 最短路

仙人掌上求最短路。

将仙人掌转化为圆方树,圆点与方点连的边长为圆点到方点父节点的最短路,Tarjan可以求出。

考虑前缀和+LCA,如果LCA为圆点,直接算就可以了。

如果为方点,分类讨论是否经过返祖边,就可以了。

#include<cmath>
#include<cstdio>
#include<vector>
using namespace std;
const int MAXN=1e4+5,MAXE=2*MAXN;
struct Edge{
	int tot,lnk[MAXN*2],son[MAXE<<1],nxt[MAXE<<1],W[MAXE<<1];
	void Add(int x,int y,int w){son[++tot]=y;W[tot]=w;nxt[tot]=lnk[x];lnk[x]=tot;}
}E,S;
int n,m,q,Fa[MAXN*2][30],Num,Dst[MAXN*2],cnt,Dfn[MAXN],Low[MAXN],cst[MAXN*2],Sum[MAXN*2],Siz[MAXN],Dep[MAXN*2];
bool Plc[MAXN*2];
#include<cctype>
int read(){
	int ret=0;char ch=getchar();bool f=1;
	for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
	for(; isdigit(ch);ch=getchar()) ret=ret*10+ch-48;
	return f?ret:-ret;
}
vector<int> a;
void Work(int x,int y,int EW){
	a.clear();++Num;
	for(int i=y;i!=x;i=Fa[i][0]) a.push_back(i);a.push_back(x);
	Siz[Num]=EW+Dst[y]-Dst[x];
	for(int i=0;i<a.size();i++)
	if(Dst[a[i]]-Dst[x]<Siz[Num]-(Dst[a[i]]-Dst[x])) Plc[a[i]]=1,S.Add(a[i],n+Num,Dst[a[i]]-Dst[x]),S.Add(n+Num,a[i],Dst[a[i]]-Dst[x]);
	else Plc[a[i]]=0,S.Add(a[i],n+Num,Siz[Num]-(Dst[a[i]]-Dst[x])),S.Add(n+Num,a[i],Siz[Num]-(Dst[a[i]]-Dst[x]));
}
void Tarjan(int x,int fa){
	Dfn[x]=Low[x]=++cnt;Fa[x][0]=fa;
	for(int j=E.lnk[x];j;j=E.nxt[j])
	if(E.son[j]!=fa){
		if(!Dfn[E.son[j]]) Dst[E.son[j]]=Dst[x]+E.W[j],Tarjan(E.son[j],x),Low[x]=min(Low[x],Low[E.son[j]]);
		else Low[x]=min(Low[x],Dfn[E.son[j]]);
		if(Dfn[x]<Low[E.son[j]]) S.Add(x,E.son[j],E.W[j]);
	}
	for(int j=E.lnk[x];j;j=E.nxt[j])
	if(Fa[E.son[j]][0]!=x&&Dfn[E.son[j]]>Dfn[x]) Work(x,E.son[j],E.W[j]);
}
void DFS(int x,int fa){
	Fa[x][0]=fa;Dep[x]=Dep[fa]+1;
	for(int j=S.lnk[x];j;j=S.nxt[j])
	if(S.son[j]!=fa) Sum[S.son[j]]=Sum[x]+S.W[j],cst[S.son[j]]=S.W[j],DFS(S.son[j],x);
}
void INIT(){
	for(int j=1;(1<<j)<=n+Num;j++)
	for(int i=1;i<=n+Num;i++) Fa[i][j]=Fa[Fa[i][j-1]][j-1];
}
int LCA(int &p,int &q){
	if(Dep[p]<Dep[q]) swap(p,q);
	int Del=Dep[p]-Dep[q];
	for(int j=0;(1<<j)<=Del;j++) if(Del&(1<<j)) p=Fa[p][j];
	if(p==q) return p;
	for(int j=log2(n);j>=0;j--)
	if(Fa[p][j]!=Fa[q][j]) p=Fa[p][j],q=Fa[q][j];
	return Fa[p][0];
}
int main(){
	n=read(),m=read(),q=read();
	for(int i=1,x,y,w;i<=m;i++) x=read(),y=read(),w=read(),E.Add(x,y,w),E.Add(y,x,w);
	for(int i=1;i<=n;i++) if(!Dfn[i]) Tarjan(i,0);
	DFS(1,0);INIT();
	for(int i=1;i<=q;i++){
		int xx=read(),yy=read(),x=xx,y=yy,fa=LCA(x,y);
		if(fa<=n) printf("%d\n",Sum[xx]+Sum[yy]-2*Sum[fa]);
		else{
			int Ans=0;int dis=0;
			if(Plc[x]==Plc[y]) dis=cst[x]-cst[y];
			else dis=min(cst[x]+cst[y],Siz[fa-n]-cst[x]-cst[y]);
			if(dis<0) dis=-dis;
			Ans=Sum[xx]+Sum[yy]-Sum[x]-Sum[y]+dis;
			printf("%d\n",Ans);
		}
	}
	return 0;
}
发布了306 篇原创文章 · 获赞 104 · 访问量 8万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 像素格子 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览