2020杭电多校二 1001 Total Eclipse

http://acm.hdu.edu.cn/showproblem.php?pid=6763
题意:一个图,每次选一个联通块,里面的每个点亮度减1,直至全为0的最小次数。
思路:正着做是依次分裂图,不好写,那就倒着做,依次合并图。
2020杭电多校二 1001 Total Eclipse_第1张图片

#include
using namespace std;
const int maxn=100000+100;
typedef long long ll;

int T,n,m,a[maxn],p[maxn],r[maxn],ok[maxn];
vector<int> G[maxn];
ll sum;

int find(int x){return x==p[x]?x:p[x]=find(p[x]);}

bool cmp(int x,int y){return a[x]>a[y];}

int main()
{
    //freopen("input.in","r",stdin);
    cin>>T;
    while(T--)
    {
        cin>>n>>m;
        sum=0;
        for(int i=1;i<=n;i++)scanf("%d",&a[i]),G[i].clear(),p[i]=i,r[i]=i,ok[i]=0;
        int u,v;
        while(m--)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        sort(r+1,r+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
            int u=r[i];
            sum+=a[u];
            ok[u]=1;
            for(int v:G[u])
            {
                if(ok[v] && find(u)!=find(v))
                {
                    sum-=a[u];
                    p[find(v)]=find(u);
                }
            }
        }
        printf("%lld\n",sum);
    }
}

你可能感兴趣的:(MST)