BZOJ4158 : [POI2007]Railway

论文题。

随便取个关键点，求出最短路树。

求出所有关键点组成的虚树，将两端都在虚树上的边保留。

对剩下的边求出最小生成树即可得到一组可行解。

#include<cstdio>
#include<algorithm>
const int N=5010,M=500010,inf=~0U>>1;
int n,m,p,i,x,f[N],id[N],ans,cnt,use[M];
int q[131072],in[N],d[N],pre[N];unsigned short h,t;
struct Edge{int v,w;Edge*nxt;}*g[N],pool[M<<1],*cur=pool,*I;
inline void add(int x,int y,int z){I=cur++;I->v=y;I->w=z;I->nxt=g[x];g[x]=I;}
struct E{int x,y,z;}e[M];
inline bool cmp(E a,E b){return a.z<b.z;}
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
int F(int x){return f[x]==x?x:f[x]=F(f[x]);}
int main(){
  for(read(n),read(m);i<m;i++){
    read(e[i].x),read(e[i].y),read(e[i].z);
    add(e[i].x,e[i].y,e[i].z);
    add(e[i].y,e[i].x,e[i].z);
  }
  for(read(p),i=1;i<=p;i++)read(id[i]);
  for(i=1;i<=n;i++)d[i]=inf;
  d[q[0]=id[1]]=0,in[id[1]]=1;
  while(h!=t+1)for(I=g[x=q[h++]],in[x]=0;I;I=I->nxt)if(d[x]+I->w<d[I->v]){
    d[I->v]=d[x]+I->w,pre[I->v]=x;
    if(!in[I->v]){
      in[I->v]=1;
      if(d[I->v]<d[q[h]])q[--h]=I->v;else q[++t]=I->v;
    }
  }
  for(in[id[1]]=i=1;i<=p;i++)for(x=id[i];!in[x];x=pre[x])in[x]=1;
  for(i=1;i<=n;i++)f[i]=i;
  for(std::sort(e,e+m,cmp),i=0;i<m;i++)if(in[e[i].x]&&in[e[i].y]&&F(e[i].x)!=F(e[i].y)){
    f[f[e[i].x]]=f[e[i].y];
    ans+=e[i].z,cnt++,use[i]=1;
  }
  for(printf("%d %d
",ans,cnt),i=0;i<m;i++)if(use[i])printf("%d %d
",e[i].x,e[i].y);
  return 0;
}