小R与怪病

Posted on Edited on In

题目大意：小镇上有 n 个人得了一种奇怪的病，他们两两熟悉并且可接触。现在有一种疫苗可以治愈这种疾病，小 R 作为镇长，希望可以利用小镇财政为他们治病。这种疫苗有种奇怪的特性，可以通过病人间的接触接种（即一个接种过的病人 i接触一个未接种过的病人 j，则后者也被接种，费用是 pij）。当然也可以通过医护人员直接接种（费用是 ai）。上述两种方法对不同人的费用是不同的，现在小R 拿到了医护人员提供的费用明细（每个人通过直接接种和其他人接触接种的费用），求助你为他提供一个所有人都接种的最小总费用。

连个超极源点，最小生成树板子题目

注意坑点：

  • 边的数组需要开到N*N
  • 并查集路径压缩
  • \(fa[fx] = fy\)不要搞错顺序
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 333, inf = 214748364;

int n, ycnt, num, res;
int fa[N];

struct yhzhyhm
{
    int x, y, z;
    bool operator < (const yhzhyhm &yhm) const
    {
        return z < yhm.z;
    }
}yh[N * N];

inline int yhm_find(int x)
{
    return fa[x] == x ? x : fa[x] = yhm_find(fa[x]);
}

signed main()
{
    int i, j; n = read();
    for (i = 1; i <= n; ++ i) fa[i] = i; fa[306] = 306;
    for (i = 1; i <= n; ++ i)
    {
        int val = read();
        yh[++ ycnt].x = 306, yh[ycnt].y = i, yh[ycnt].z = val;
    }
    for (i = 1; i <= n; ++ i) 
    {
        for (j = 1; j <= n; ++ j)
        {
            int val = read();
            if (i <= j) continue;
            yh[++ ycnt].x = i, yh[ycnt].y = j;
            yh[ycnt].z = val;
        }
    }

    sort(yh + 1, yh + 1 + ycnt);
    
    for (i = 1; i <= ycnt; ++ i)
    {
        int x = yh[i].x, y = yh[i].y, z = yh[i].z;
        int fx = yhm_find(x), fy = yhm_find(y);
        if (fx == fy) continue;
        
        fa[fx] = fy;
        res += yh[i].z;
        ++ num;
        if (num == n) break;
    }
    put(res);
    
    return 0;
}