結果

問題 No.748 yuki国のお財布事情
ユーザー Konton7Konton7
提出日時 2020-03-06 15:08:22
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 126 ms / 2,000 ms
コード長 2,570 bytes
コンパイル時間 2,352 ms
コンパイル使用メモリ 211,308 KB
実行使用メモリ 6,776 KB
最終ジャッジ日時 2024-10-14 02:49:05
合計ジャッジ時間 5,373 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 1 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 16 ms
5,248 KB
testcase_14 AC 26 ms
5,248 KB
testcase_15 AC 17 ms
5,248 KB
testcase_16 AC 50 ms
5,248 KB
testcase_17 AC 105 ms
6,176 KB
testcase_18 AC 115 ms
6,068 KB
testcase_19 AC 126 ms
6,756 KB
testcase_20 AC 109 ms
6,456 KB
testcase_21 AC 110 ms
6,472 KB
testcase_22 AC 1 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 2 ms
5,248 KB
testcase_25 AC 93 ms
6,128 KB
testcase_26 AC 109 ms
6,776 KB
testcase_27 AC 107 ms
6,728 KB
testcase_28 AC 85 ms
5,956 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
using ll = long long;
using VI = vector<int>;
using VL = vector<ll>;
using PII = std::pair<int, int>;
using PLL = std::pair<ll, ll>;

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define repr(i, n) for (int i = (int)(n)-1; i >= 0; i--)
#define rep2(i, s, n) for (int i = (s); i < (int)(n); i++)
#define allpt(v) (v).begin(), (v).end()
#define allpt_r(v) (v).rbegin(), (v).rend()

const int mod = 1e9 + 7;
const string wsp = " ";
const string tb = "\t";
const string rt = "\n";

template <typename T>
void show1dvec(vector<T> v)
{
    int n = v.size() - 1;
    rep(i, n) cout << v[i] << wsp;
    cout << v[n] << rt;
    return;
}

template <typename T>
void show2dvec(vector<vector<T>> v)
{
    int n = v.size();
    rep(i, n) show1dvec(v[i]);
}

class Unionfind
{
    vector<int> p;

public:
    int find(int x);
    void unite(int x, int y);
    Unionfind(int n);
};

Unionfind::Unionfind(int n)
{
    for (int i = 0; i < n; i++)
    {
        p.push_back(i);
    }
}
int Unionfind::find(int x)
{
    while (p[x] != x)
    {
        p[x] = p[p[x]];
        x = p[x];
    }
    return x;
}

void Unionfind::unite(int x, int y)
{
    x = Unionfind::find(x);
    y = Unionfind::find(y);
    if (x != y)
    {
        p[x] = y;
    }
}

struct road
{
    int cost;
    int from;
    int to;
};

int main()
{

#ifdef DEBUG
    cout << "DEBUG MODE" << endl;
    ifstream in("input.txt"); //for debug
    cin.rdbuf(in.rdbuf());    //for debug
#endif

    int n, m, k, a, b, c;
    ll total_cost = 0, remain_cost = 0;
    cin >> n >> m >> k;
    vector<road> roads(m), sp_roads, ord_roads;
    vector<bool> sp_or_ord(m, false);
    Unionfind cities(n);

    rep(i, m)
    {
        cin >> a >> b >> c;
        roads[i] = (road){c, a - 1, b - 1};
        total_cost += c;
    }
    rep(i, k)
    {
        cin >> c;
        sp_or_ord[c - 1] = true;
    }
    rep(i, m)
    {
        if (sp_or_ord[i])
            sp_roads.push_back(roads[i]);
        else
            ord_roads.push_back(roads[i]);
    }
    rep(i, k)
    {
        cities.unite(sp_roads[i].from, sp_roads[i].to);
        remain_cost += sp_roads[i].cost;
    }


    sort(allpt(ord_roads), [](const road &a, const road &b) { if (a.cost != b.cost) return a.cost < b.cost; else return a.from < b.from; });

    for (auto z : ord_roads)
    {
        if (cities.find(z.from) != cities.find(z.to))
        {
            cities.unite(z.from, z.to);
            remain_cost += z.cost;
        }
    }

    cout << total_cost - remain_cost << rt;
    return 0;
}
0