[BZOJ2125][仙人掌]最短路

BZOJ2125

建出圆方树,圆方边的权值为圆点到环上最高点在dfs树上的点的距离,这样就可以记录下环上两个点的距离
并在方点处记录一下环的总长,因为环上两点间有两条路径可走,要分类讨论
然后就是树上求dis的问题,先求lca
然后lca如果是圆点就直接求距离
如果是方点就分类讨论一下就好了

Code:

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    int res=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
    while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
    return res*f;
}
const int N=2e4+5;
struct E{
    int vis[N<<1],c[N<<1],head[N<<1],nxt[N<<1],cnt=0;
    inline void add(int x,int y,int z){vis[++cnt]=y;nxt[cnt]=head[x];head[x]=cnt;c[cnt]=z;}
}tar,tr;
int tot,len[N],d[N],stk[N],lst[N],dfn[N],low[N],top,pt[N],sign=0;
inline void build(int v,int k){
    ++tot;int x;len[tot]=d[stk[top]]-d[v]+lst[stk[top]];
    do{
     x=stk[top--];
     int x1=d[x]-d[v],x2=len[tot]-x1;
     tr.add(tot,x,min(x1,x2));pt[x]=(x1<=x2);
    }while(x!=k);
    tr.add(v,tot,0);
}
void tarjan(int v,int f){
    dfn[v]=low[v]=++sign;stk[++top]=v;
    for(int i=tar.head[v];i;i=tar.nxt[i]){
        int y=tar.vis[i];
        if(y==f) continue;
        if(!dfn[y]){
            d[y]=d[v]+tar.c[i];tarjan(y,v);
            if(low[y]>dfn[v]) --top,tr.add(v,y,tar.c[i]);
            else if(low[y]==dfn[v]) build(v,y);
            low[v]=min(low[v],low[y]);
        }
        else low[v]=min(low[v],dfn[y]),lst[v]=tar.c[i];
    }
}
int fa[N][16],sum[N][16],dep[N];
void dfs(int v){
    for(int i=tr.head[v];i;i=tr.nxt[i]){
        int y=tr.vis[i];
        fa[y][0]=v;dep[y]=dep[v]+1;sum[y][0]=tr.c[i];
        dfs(y);
    }
}
int n;
inline int lca(int u,int v){
    if(dep[u]<dep[v]) swap(u,v);
    int t=dep[u]-dep[v],res=0;
    for(int i=15;~i;i--) if(t&(1<<i)) res+=sum[u][i],u=fa[u][i];
    if(u==v) return res;
    for(int i=15;~i;i--) if(fa[u][i]!=fa[v][i]) res+=sum[u][i]+sum[v][i],u=fa[u][i],v=fa[v][i];
    if(fa[u][0]<=n) return sum[u][0]+sum[v][0]+res;
    int x1=sum[u][0],x2=sum[v][0],mn;
    if(pt[u]==pt[v]) mn=min(abs(x1-x2),len[fa[u][0]]-abs(x1-x2));
    else mn=min(x1+x2,len[fa[u][0]]-x1-x2);
    return res+mn;
}
int m,q;
int main(){
    n=read();m=read();q=read();tot=n;
    for(int x,y,z,i=1;i<=m;i++){
        x=read(),y=read(),z=read();
        tar.add(x,y,z),tar.add(y,x,z);
    }
    tarjan(1,0);dfs(1);
    for(int j=1;j<=15;j++)
        for(int i=1;i<=tot;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1],sum[i][j]=sum[i][j-1]+sum[fa[i][j-1]][j-1];
    for(int x,y,i=1;i<=q;i++) x=read(),y=read(),cout<<lca(x,y)<<"\n";
    return 0;
}
发布了298 篇原创文章 · 获赞 88 · 访问量 1万+
展开阅读全文

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

©️2019 CSDN 皮肤主题: 书香水墨 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览