【最短路树】BZOJ 3694 —— 最短路

题目传送门(权限题警告)

显然可以发现,将1到i路径上的最后一条路切断后,需要重新找到一条从i的子树出发的最短路径重新回到最短路树上去.
因此考虑一条边什么时候会被计算在答案中.
设一条边u->v权值为val,只会可能对u,v到 lca(u,v) 之间的点产生影响.记录源点1到节点i的距离为dep[i],那么就可以把答案更新为 dep[u]+dep[v]+val-dep[i].
可以预处理出dep,然后对于每一条边都暴力爬升至lca更新答案.
时间复杂度:$O(nm)$ 但是数据可能比较水,没有生成最坏的数据卡,所以针对随机数据是O(mlogn)

这样做显然不能过掉 (实际能过) ,于是开始优化.首先对于边,按照dep[u]+dep[v]+val排序,因为dep[i]是随着点的不同有变化,不用管.
那么显然每个点有效的答案只会被第一次到它的边更新.然后利用并查集,可以维护出一个点u上次更新到的点.然后每次跳转到那里更新,注意判断是否直接越过了lca.
时间复杂度:$O(mlogm)$

只写了暴力,并查集代码以后补

#include<cstdio>
#include<algorithm>
using namespace std;

const int MAXN=4005;
const int MAXM=100005;
const int INF=0x3f3f3f3f;

int n,m,ncnt;
int p[MAXN],dep[MAXN],vis[MAXN];
int ans[MAXN];
int head[MAXN],ecnt;
struct edge{
    int v,val,nxt;
}E[MAXM];
struct node{
    int u,v,val;
}G[MAXM];

void addedge(int u,int v,int val){
    E[++ecnt]=(edge){v,val,head[u]};
    head[u]=ecnt;
}

void dfs(int u,int fa=0){
    p[u]=fa;
    for(int i=head[u];i;i=E[i].nxt){
        int v=E[i].v,val=E[i].val;
        if(v==fa) continue;
        dep[v]=dep[u]+val;
        dfs(v,u);
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,val,op;
        scanf("%d%d%d%d",&u,&v,&val,&op);
        if(op) addedge(u,v,val),addedge(v,u,val);
        else G[++ncnt]=(node){u,v,val};
    }
    dfs(1);
    for(int i=1;i<=ncnt;i++) G[i].val+=dep[G[i].u]+dep[G[i].v];
    for(int i=1;i<=n;i++) ans[i]=INF;
    for(int i=1;i<=ncnt;i++){
        int x=G[i].u,y=G[i].v;
        while(x!=y){
            if(dep[x]<dep[y]) swap(x,y);
            ans[x]=min(ans[x],G[i].val-dep[x]);
            x=p[x];
        }
    }
    for(int i=2;i<=n;i++)
        printf("%s%d",i==2?"":" ",ans[i]==INF?-1:ans[i]);
}