neu1132(增量最小生成树,两点之间的最大边权)

来源:互联网 发布:js window document 编辑:程序博客网 时间:2024/04/28 18:24

问题:给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-6using 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;}

0 0
原创粉丝点击