Timus Online Judge 1982 Electrification Plan

最后三周，打算练练图论，因为感觉这块已经丢掉了很久了。
今天上午AC了四道题目，都是简单题目，其实都没有什么好写的，还是记录一下吧，毕竟还是认真思考了的。

题目大意：有N个城市，编号为1~N，其中M个城市有电站。问：要让每一个城市都用上电的最少花费是多少？其中已经给出的M个有电站的城市的编号和城市两两之间要架设电路的固定花费。

拿到题目的时候一来就想成最短路，以为是每个城市到有电站城市的路径之和，可能是刚开始的时候题目没有读清楚吧。WA了之后再次读题才发现原来几乎是个裸的最小生成树。我们先把所有的有电站的城市先连接起来，这总共的花费记为0，然后再把其余的城市连接进来就行了。因为数据量小，用kruskal或者是prim都是可以的。

奉上代码，请多指教！

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
class edge{
    public:
        int u, v, cost;
        edge(){}
        edge(int u, int v, int cost) {
            this->u = u;
            this->v = v;
            this->cost = cost;
        }
        bool operator < (const edge &a) const{
            return this->cost < a.cost;
        }
};
int N, M, fa[101];
vector<edge> vec;
int find(int x) {
    return fa[x] = fa[x] == x ? x : find(fa[x]);
}
void unite(int x, int y) {
    int fax = find(x), fay = find(y);
    if (fax != fay) fa[fay] = fax;
}
bool same(int x, int y) {
    return find(x) == find(y);
}
int kruskal() {
    sort(vec.begin(), vec.end());
    int res = 0;
    for (int i = 0; i < vec.size(); ++i) {
        edge e = vec[i];
        if (!same(e.u, e.v)) {
            unite(e.u, e.v);
            res += e.cost;
        }
    }
    return res;
}
int main() {
    for (int i = 0; i < 101; ++i) fa[i] = i;
    cin >> N >> M;
    int val, t;
    for (int i = 0; i < M; ++i) {
        cin >> val;
        if (i == 0) t = val;
        unite(t, val);
    }
    for (int i = 1; i <= N; ++i) {
        for (int j = 1; j <= N; ++j) {
            cin >> val;
            if (i < j) vec.push_back(edge(i, j, val));
        }
    }
    cout << kruskal() << endl;
    return 0;
}