UVa10600 ACM Contest and Blackout

1.题目描述：点击打开链接

2.解题思路：本题实际上是求最小生成树和次小生成树。做法在之前的例题中已经用过多次，不予赘述了。

3.代码：

#include<iostream>
#include<algorithm>
#include<cassert>
#include<string>
#include<sstream>
#include<set>
#include<bitset>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<cctype>
#include<list>
#include<complex>
#include<functional>
using namespace std;

#define me(s) memset(s,0,sizeof(s))
#define rep(i,n) for(int i=0;i<(n);i++)
#define pb push_back
typedef long long ll;
typedef pair <int,int> P;


const int N=100+10;
const int INF=1e9;
struct Edge
{
    int x,y,d;
    bool operator<(const Edge&rhs)const
    {
        return d<rhs.d;
    }
};

Edge e[N*N];
int pa[N];
int vis[N*N];

int findset(int x){return pa[x]==x?x:pa[x]=findset(pa[x]);}
int n,m;
int maxcost[N][N];
vector<int>nodes;
vector<int>g[N],C[N];

void dfs(int u,int fa,int facost)
{
    for(int i=0;i<nodes.size();i++)
    {
        int x=nodes[i];
        maxcost[x][u]=maxcost[u][x]=max(maxcost[x][fa],facost);
    }
    nodes.push_back(u);
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v!=fa)
            dfs(v,u,C[u][i]);
    }
}

int MST()
{
    int ans=0;
    int cnt=0;
    sort(e,e+m);
    me(vis);
    for(int i=0;i<n;i++)
    {
        pa[i]=i;
        g[i].clear();
        C[i].clear();
    }
    for(int i=0;i<m;i++)
    {
        int x=e[i].x,y=e[i].y,d=e[i].d;
        int u=findset(x),v=findset(y);
        if(u!=v)
        {
            vis[i]=1;
            pa[u]=v;
            g[x].push_back(y);C[x].push_back(d);
            g[y].push_back(x);C[y].push_back(d);
            ans+=d;
            if(++cnt==n-1)break;
        }
    }
    return ans;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        int u,v,d;
        me(g);me(e);me(maxcost);
        nodes.clear();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&d);
            u--,v--;
            e[i]=Edge{u,v,d};
        }
        int ans1=MST();
        dfs(0,-1,0);
        int ans2=INF;
        for(int i=0;i<m;i++)
            if(!vis[i])
        {
            int tmp=e[i].d;
            int x=e[i].x,y=e[i].y;
            ans2=min(ans2,ans1-maxcost[x][y]+tmp);
        }
        printf("%d %d\n",ans1,ans2);
    }
}