兽径管理(生成树)

兽径管理

题目描述

约翰农场的牛群希望能够在 N 个(1<=N<=200) 草地之间任意移动。草地的编号由 1到 N。草地之间有树林
隔开。牛群希望能够选择草地间的路径，使牛群能够从任一 片草地移动到任一片其它草地。 牛群可在路径上
双向通行。

牛群并不能创造路径，但是他们会保有及利用已经发现的野兽所走出来的路径(以 下简称兽径)。每星期他们会
选择并管理一些或全部已知的兽径当作通路。

牛群每星期初会发现一条新的兽径。他们接着必须决定管理哪些兽径来组成该周牛 群移动的通路，使得牛群得
以从任一草地移动到任一草地。牛群只能使用当周有被 管理的兽径做为通路。

牛群希望他们管理的兽径长度和为最小。牛群可以从所有他们知道的所有兽径中挑 选出一些来管理。牛群可以
挑选的兽径与它之前是否曾被管理无关。

在每周开始的时候，牛群会描述他们新发现的兽径。如果可能的话，请找出可从任 何一草地通达另一草地的一
组需管理的兽径，使其兽径长度和最小。

题目分析

从题目中能够从任一 片草地移动到任一片其它草地可以推测出求解的是最小生成树，但是可选边时逐渐增加的，如果正向每集一条边求一次最小生成树时间复杂度会较大，这时候我们可以逆向考虑

• 最后一周x拥有最多的可选择的边，可以求出其最小生成树tree
• 第x天的边如果是tree所需要的，对于x-1天就需要重新求一边，反之应与第x天得到的tree相同
• 若第x天都无法得到一个生成树，则选择更少的也无法得到

关键在于记录生成树所用边，用数组统一存放输出即可

code

#include<bits/stdc++.h>

using namespace std;

const int N = 20010, M = 2 * N;

int n, m, k, t;
bool st[N];
int p[N], ans[N], ed[N];

struct Edge
{
    int a, b, w;
    int id;
    bool operator < (const Edge &t) const
    {
        return w < t.w;
    }
}e[M];

int findx(int x)
{
    if(p[x] != x) p[x] = findx(p[x]);
    return p[x];
}

int kruskal()
{
    int res = 0, cnt = 0;
    memset(st, 0, sizeof st);
    for(int i = 1; i <= n; i ++)p[i] = i;

    for(int i = 0; i < m; i ++)
    {
        if(ed[e[i].id]) continue;
        int a = findx(e[i].a), b = findx(e[i].b), w = e[i].w;
        int nu = e[i].id;
        if(a != b)
        {
            p[a] = b;
            st[nu] = true;
            res += w;
            cnt ++;
        }
    }

    if(cnt != n - 1) res = -1;
    return res;
}

int main()
{
    cin >> n >> m;

    for(int i = 0; i < m; i ++)
    {
        int u, v, w;
        cin >> u >> v >> w;
        e[i] = {u, v, w, i};
    }
    sort(e, e + m);

    ans[m - 1] = kruskal();
    for(int i = m - 2; i >= 0; i --)
    {
        ed[i + 1] = true;
        if(st[i + 1]) 
        {
            ans[i] = kruskal();
        }
        else ans[i] = ans[i + 1];
        
        if(ans[i] == -1)
        {
            for(int j = 0; j < i; j ++) ans[j] = -1;
            break;
        }
    }

    for(int i = 0; i < m; i ++) cout << ans[i] << "\n";

    return 0;
}