PAT T1001 Battle Over Cities-Hard Version

按题意枚举每个点，建立缺少该点情况下的最小生成树，取权值最大的~

#include<bits/stdc++.h>
using namespace std;
const int maxn=;
const int inf=1e9;
int g[maxn][maxn];
int visit[maxn];
int d[maxn];
int N,M,x,y;
int flag;
int prim (int s) {
    fill (d,d+maxn,inf);
    d[s]=;
    int ans=;
    for (int i=;i<=N-;i++) {
        int u=-,min=inf;
        for (int j=;j<=N;j++)
        if (!visit[j]&&d[j]<min) {
            u=j;
            min=d[j];
        }
        if (u==-) return inf;
        visit[u]=;
        ans+=d[u];
        for (int v=;v<=N;v++)
        if (!visit[v]&&g[u][v]!=inf&&g[u][v]<d[v]) d[v]=g[u][v];
    }
    return ans;
}
int main () {
    scanf ("%d %d",&N,&M);
    for (int i=;i<=N;i++)
    for (int j=;j<=N;j++)
    g[i][j]=inf;
    for (int i=;i<M;i++) {
        scanf ("%d %d",&x,&y);
        scanf ("%d %d",&g[x][y],&flag);
        if (flag==) g[x][y]=;
        g[y][x]=g[x][y];
    }
    vector<int> vi;
    int Max=-;
    for (int i=;i<=N;i++) {
        fill (visit,visit+maxn,);
        visit[i]=;
        int j=i<N?i+:;
        int mst=prim(j);
        if (mst>Max) {
            vi.clear();
            vi.push_back(i);
            Max=mst;
        }
        else if (mst==Max) {
            vi.push_back(i);
        }
    }
    if (Max==) {
        printf ("");
        return ;
    }
    for (int i=;i<vi.size();i++) {
        if (i!=) printf (" ");
        printf ("%d",vi[i]);
    }
    return ;
}