bzoj1576 [Usaco2009 Jan]安全路经Travel（最短路径树+并查集）

bzoj1576 [Usaco2009 Jan]安全路经Travel

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1576

题意：

数据范围
n <= 1e5 m<=2e5 ti<=1000

题解：
原题意即求出每个点不走最短路径树上的父边能得到的最短路。
我们暂且称最短路径树上的边为树边，其他边为杂边。
那么，每一条杂边连接着树上的两个点 u，v，这条杂边权值为w。这两个点树上路径上的所有的点，都可以被这条杂边更新。
例如，对于点x，$dis_{new}[x]=dis[u]+dis[v]+w[i]-dis[x]$
对于一条杂边，前面那三个量是一定的。
每一个点，我们希望用最小的杂边更新他，更新过后就不再更新。
于是可以按w排序杂边，依次更新，更新过后的点就用并查集缩起来，以保证每个点只被更新一次。

数组开小WA一上午。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N=100005;
const int M=200005;
const int inf=0x3f3f3f3f;
int head[N],fa[N],anc[N],dis[N],dep[N],pre[N],ans[N],n,m,cnt=0,num=1;
bool ontree[2*M];
struct Edge
{
    int u,v,nxt,w;
    Edge(){}
    Edge(int u,int v,int nxt,int w):u(u),v(v),nxt(nxt),w(w){}
}E[2*M],d[2*M];
bool cmp(const Edge &A,const Edge &B){return A.w<B.w;}
struct node 
{
    int x,y;
    node(){}
    node(int x,int y):x(x),y(y){}
    bool friend operator<(const node &A,const node &B){return A.y>B.y;}
};
priority_queue<node> Q;
void build(int u,int v,int w) {num++; E[num]=Edge(u,v,head[u],w); head[u]=num;}
void dijkstra()
{
    memset(dis,0x3f,sizeof(dis)); dis[1]=0;
    Q.push(node(1,0));
    while(!Q.empty())
    {
        node pa=Q.top(); Q.pop();
        int u=pa.x; 
        for(int i=head[u];i;i=E[i].nxt)
        {
            if(dis[E[i].v]>dis[u]+E[i].w)
            {
                dis[E[i].v]=dis[u]+E[i].w;
                dep[E[i].v]=dep[u]+1;
                pre[E[i].v]=i; anc[E[i].v]=u;
                Q.push(node(E[i].v,dis[E[i].v]));
            }
        }
    }
}
int getfa(int x){return (fa[x]==x)?x:fa[x]=getfa(fa[x]);}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v,w; scanf("%d%d%d",&u,&v,&w);
        build(u,v,w); build(v,u,w);
    }
    dijkstra(); 
    for(int i=2;i<=n;i++) ontree[pre[i]]=1;
    memset(ans,0x3f,sizeof(ans)); for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=2;i<=num;i++) if(!ontree[i]&&!ontree[i^1]) d[++cnt]=Edge(E[i].u,E[i].v,0,dis[E[i].u]+dis[E[i].v]+E[i].w);
    sort(d+1,d+cnt+1,cmp);
    for(int i=1;i<=cnt;i++)
    {
        int u=d[i].u; int v=d[i].v;
        u=getfa(u); v=getfa(v);
        while(u!=v)
        {
            if(dep[u]<dep[v]) swap(u,v);
            ans[u]=min(ans[u],d[i].w-dis[u]);
            u=fa[u]=getfa(anc[u]); 
        }
    }
    for(int i=2;i<=n;i++)
    {
        if(ans[i]<inf) printf("%d\n",ans[i]);
        else printf("-1\n");
    }
    return 0;
}