hdu 4126（最小生成树+dp）

题目大意：

   给你一张连通图，共有Q次更改，每次在原图的基础上，增加一条边的权值，求每次更改后最小生成树的权的总和sum,并输出sum/Q；

题解：

<span style="font-family:FangSong_GB2312;font-size:14px;"><strong>#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=3010;
struct edge{
  int x,next;
}edg[N*2];                  //记录MST边
int totl;
int front[N];
int dis[N][N],cnt,dp[N][N]; //dis记录 任意两点间距离， dp求解i到j子树的最短距离
bool is_MST[N][N];          //标记MST边
int n,m;
//--------------------求MST（prim）-------------------
const int inf=1000000000;
bool vis[N];
int e[N],fa[N];
int prim() {
  int i,j,u,v,w,ans=0;
  for(i=0;i<n;i++) e[i]=dis[0][i],fa[i]=0;
  memset(vis,0,sizeof(vis)); vis[0]=1;
  for(i=1;i<n;i++) {
    w=inf; u=-1;
    for(j=0;j<n;j++) if(!vis[j]&&w>e[j]) w=e[j],u=j;
    if(u==-1) return -1;
    vis[u]=1; ans+=w;
    is_MST[u][fa[u]]=is_MST[fa[u]][u]=true;
    edg[totl].x=u;edg[totl].next=front[fa[u]];
    front[fa[u]]=totl++;
    edg[totl].x=fa[u];edg[totl].next=front[u];
    front[u]=totl++;
    for(v=0;v<n;v++) if(!vis[v])
      if(e[v]>dis[u][v]) {
          e[v]=dis[u][v];
          fa[v]=u;
      }
  }
  return ans;
}
int now_node;
//----------------求i到j子树的最短距离dp------------------------
int dfs1(int o,int f){
   if (is_MST[now_node][o]==false) dp[now_node][o]=dis[now_node][o];
   else dp[now_node][o]=inf;
   for (int i=front[o];i!=-1;i=edg[i].next){
     if (is_MST[o][edg[i].x]&&f!=edg[i].x){
       dp[now_node][o]=min(dfs1(edg[i].x,o),dp[now_node][o]);
     }
   }
   return dp[now_node][o];
}
int best[N][N];
//-----------------求i，j的最佳替换边-------------
int dfs2(int o,int f){
  best[now_node][o]=dp[o][now_node];
  for (int i=front[o];i!=-1;i=edg[i].next){
    if (is_MST[o][edg[i].x]&&f!=edg[i].x){
       best[now_node][o]=min(dfs2(edg[i].x,o),best[now_node][o]);
    }
  }
  return best[now_node][o];
}
int main (){
  //freopen("test.in","r",stdin);
  while (~scanf("%d%d",&n,&m)){
    if (n==0&&m==0) break;
    //-----------更新数组--------------------
    totl=0;
    for(int i=0;i<=n;i++)
      for(int j=0;j<=n;j++) {
        dis[i][j]=dp[i][j]=best[i][j]=inf;
        is_MST[i][j]=0;
    }
    memset(front,-1,sizeof(front));
    for (int i=0;i<m;i++){
      int a,b,w;scanf("%d%d%d",&a,&b,&w);
      dis[a][b]=dis[b][a]=w;
    }
    double MST_val=prim();
    for (int i=0;i<n;i++){
      now_node=i;
      dfs1(i,-1);
    }
    for (int i=0;i<n;i++){
      now_node=i;
      dfs2(i,-1);
    }
    double sum=0;
    //-------------询问-----------------
    int Q;scanf("%d",&Q);
    for (int i=0;i<Q;i++){
      int a,b,w;
      scanf("%d%d%d",&a,&b,&w);
      if (is_MST[a][b]==false){
        continue;
      }
      if (best[a][b]>=w){
        sum+=w-dis[a][b];
      }
      else {
        sum+=best[a][b]-dis[a][b];
      }
    }
    printf("%.4lf\n",sum/Q+MST_val);
  }
  return 0;
}
</strong></span>