neu1132（增量最小生成树，两点之间的最大边权）

最新推荐文章于 2022-09-01 15:08:01 发布

slmady

最新推荐文章于 2022-09-01 15:08:01 发布

阅读量2.3k

点赞数 1

文章标签：增量最小生成树

本文链接：https://blog.csdn.net/u013509299/article/details/39382319

版权

acm之图论专栏收录该内容

20 篇文章 0 订阅

订阅专栏

问题：给n-1条边构成一个最小生成树，然后q次询问，每次加入一条边，求新边和n-1条边构成的最小生成树。

解题思路：每次增加一条边，就会构成一个回路，去掉这个回路上权值最大的边，得到的就是最小生成树。

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<vector>
#include<cstring>
#include<string>

#define N 1005
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
#define eps 10e-6
using namespace std;
struct node
{
    int u,cost;
};
vector<node> G[N];
int dist[N][N],vis[N];
void dfs(int k,int cur,int cost)//求任意两点间的最大边权
{
    vis[cur] = 1;
    int len = G[cur].size();
    int i;
    for(i = 0; i < len; i++)
        if(!vis[ G[cur][i].u ])
        {
            dist[k][ G[cur][i].u ] = max(dist[k][ G[cur][i].u ], max(cost,G[cur][i].cost) );
            dfs(k,G[cur][i].u, max(cost,G[cur][i].cost) );
        }
}
int main()
{
    int n,cas = 1;
    while(scanf("%d",&n) != EOF)
    {
        int i,j,sum = 0;
        for(i = 0; i <= n; i++)     G[i].clear();
        for(i = 1; i < n; i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            node temp;
            temp.cost = w;
            temp.u = v;
            G[u].push_back(temp);
            temp.u = u;
            G[v].push_back(temp);
            sum+=w;
        }
        memset(dist,0,sizeof(dist));
        for(i = 1; i <= n; i++)
        {
            memset(vis,0,sizeof(vis));
            vis[i] = 1;
            dfs(i,i,0);
        }
        int q;
        printf("Test #%d\n",cas++);
        scanf("%d",&q);
        while(q--)
        {
            int x,y,w;
            scanf("%d%d%d",&x,&y,&w);
            int ans = sum - dist[x][y] + min(dist[x][y],w);
            printf("%d\n",ans);
        }
    }
    return 0;
}