結果

問題 No.748 yuki国のお財布事情
ユーザー saksak
提出日時 2020-07-24 00:01:36
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 133 ms / 2,000 ms
コード長 2,415 bytes
コンパイル時間 2,506 ms
コンパイル使用メモリ 214,304 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-23 23:18:47
合計ジャッジ時間 5,690 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 16 ms
5,376 KB
testcase_14 AC 25 ms
5,376 KB
testcase_15 AC 18 ms
5,376 KB
testcase_16 AC 55 ms
5,376 KB
testcase_17 AC 103 ms
5,632 KB
testcase_18 AC 115 ms
6,528 KB
testcase_19 AC 133 ms
6,944 KB
testcase_20 AC 113 ms
6,528 KB
testcase_21 AC 107 ms
6,400 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 89 ms
5,760 KB
testcase_26 AC 110 ms
6,528 KB
testcase_27 AC 103 ms
6,272 KB
testcase_28 AC 87 ms
5,760 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<ll, ll> p_ll;

template<class T>
void debug(T itr1, T itr2) { auto now = itr1; while(now<itr2) { cout << *now << " "; now++; } cout << endl; }
#define repr(i,from,to) for (ll i=(ll)from; i<(ll)to; i++)
#define all(vec) vec.begin(), vec.end()
#define rep(i,N) repr(i,0,N)
#define per(i,N) for (int i=(int)N-1; i>=0; i--)

const ll MOD = pow(10,9)+7;
const ll LLINF = pow(2,61)-1;
const int INF = pow(2,30)-1;

vector<ll> fac;
void c_fac(int x=pow(10,6)+10) { fac.resize(x,true); rep(i,x) fac[i] = i ? (fac[i-1]*i)%MOD : 1; }
ll inv(ll a, ll m=MOD) { ll b = m, x = 1, y = 0; while (b!=0) { int d = a/b; a -= b*d; swap(a,b); x -= y*d; swap(x,y); } return (x+m)%m; }
ll nck(ll n, ll k) { return fac[n]*inv(fac[k]*fac[n-k]%MOD)%MOD; }
ll gcd(ll a, ll b) { if (a<b) swap(a,b); return b==0 ? a : gcd(b, a%b); }
ll lcm(ll a, ll b) { return a/gcd(a,b)*b; }

// ----------------------------------------------------------------------
// ----------------------------------------------------------------------

struct UnionFind {
  vector<int> par, s; 
  UnionFind(int N) : par(N), s(N) { rep(i,N) { par[i] = i; s[i] = 1; } }
  int root(int x) { return par[x]==x ? x : par[x] = root(par[x]); }
  int size(int x) { return par[x]==x ? s[x] : s[x] = size(root(x)); }
  void unite(int x, int y) { int rx=root(x), ry=root(y); if (rx!=ry) { s[rx] += s[ry]; par[ry] = rx; } }
  bool same(int x, int y) {  int rx=root(x), ry=root(y); return rx==ry; }
};

struct edge { ll from; ll to; ll d; };
vector<edge> e;
vector<ll> used;

ll Kruskal(int N) {
  ll result = 0;
  UnionFind tree(N);
  for (auto x: used) {
    tree.unite(e[x].from, e[x].to);
    result += e[x].d;
  }
  sort(all(e), [](const edge &x, const edge &y){return x.d<y.d;});
  for (auto x: e) {
    if (tree.same(x.from, x.to)) continue;
    result += x.d;
    tree.unite(x.from,x.to);
  }
  return result;
}

// ----------------------------------------------------------------------
// ----------------------------------------------------------------------

int main() {
  ll N, M, K; cin >> N >> M >> K;
  e.resize(M);
  rep(i,M) {
    cin >> e[i].from >> e[i].to >> e[i].d;
    e[i].from--; e[i].to--;
  }
  used.resize(K); rep(i,K) { cin >> used[i]; used[i]--; }
  
  ll sum = 0; rep(i,M) sum += e[i].d;
  ll d = Kruskal(N);
  ll result = sum - d;
  cout << result << endl;
  return 0;
}
0