結果

問題 No.748 yuki国のお財布事情
ユーザー hide1214hide1214
提出日時 2018-10-19 22:50:16
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 2,432 bytes
コンパイル時間 1,241 ms
コンパイル使用メモリ 106,168 KB
実行使用メモリ 13,896 KB
最終ジャッジ日時 2023-08-12 01:57:58
合計ジャッジ時間 3,339 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
7,548 KB
testcase_01 AC 3 ms
7,524 KB
testcase_02 AC 3 ms
7,540 KB
testcase_03 AC 2 ms
7,528 KB
testcase_04 AC 2 ms
7,688 KB
testcase_05 AC 2 ms
7,420 KB
testcase_06 AC 2 ms
7,544 KB
testcase_07 AC 2 ms
7,476 KB
testcase_08 AC 2 ms
7,528 KB
testcase_09 AC 2 ms
7,432 KB
testcase_10 AC 2 ms
7,660 KB
testcase_11 AC 2 ms
7,472 KB
testcase_12 AC 2 ms
7,480 KB
testcase_13 AC 8 ms
8,208 KB
testcase_14 AC 12 ms
8,220 KB
testcase_15 AC 9 ms
7,984 KB
testcase_16 AC 20 ms
9,000 KB
testcase_17 AC 44 ms
12,140 KB
testcase_18 AC 49 ms
11,264 KB
testcase_19 AC 50 ms
11,128 KB
testcase_20 AC 43 ms
10,512 KB
testcase_21 AC 50 ms
12,008 KB
testcase_22 AC 2 ms
4,380 KB
testcase_23 AC 3 ms
7,520 KB
testcase_24 AC 2 ms
7,416 KB
testcase_25 AC 42 ms
10,696 KB
testcase_26 AC 53 ms
13,896 KB
testcase_27 AC 51 ms
12,980 KB
testcase_28 AC 32 ms
9,208 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<queue>
#include<stack>
#include<string>
#include<cmath>
#include<algorithm>
#include<bitset>
#include<map>
#include<unordered_map>
#include<functional>
#include<set>
#include<numeric>

#pragma region
using namespace std;
#define FOR(i,r,n) for(ll i = (ll)(r); i < (ll)(n); i++)
#define rep(i,n) FOR(i,0LL,n)
#define RFOR(i,r,n) for(ll i=(ll)(n-1);i>=r;i--)
#define rrep(i,n) RFOR(i,0LL,n)
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define COUNT(a,y,x) upper_bound(all(a), y) - lower_bound(all(a), x)
#define UNIQUE(a) sort(all(a)); a.erase(unique(all(a)), a.end())
#define SUM(a) accumulate(all(a),0LL)
#define pb push_back
#define endl '\n'
typedef long long int ll;
typedef vector<ll> vll;
typedef vector<vll> vvll;
typedef pair<ll, ll> pll;
typedef vector<pll> vpll;
typedef vector<string> vs;
typedef map<ll, ll> MAP;
const ll inf = 2222222222222222222LL;
const ll mod = 1000000007LL;

ll n = 0, m = 0, ans = 0, cnt = 0, tmp = 0, ma = -inf, mi = inf;
string s;
bool ok = true, flag = false;
ll dx[5] = { 1,-1,0,0,0 }, dy[5] = { 0,0,1,-1,0 };
ll ddx[9] = { 1,-1,0,0,1,1,-1,-1,0 }, ddy[9] = { 0,0,1,-1,1,-1,1,-1,0 };
#pragma endregion
#define MAX 222222

struct UnionFind {
  vll rank, par, data;

  void set(ll n) {
	rank.resize(n + 1, 0);
	par.resize(n + 1, 0);
	data.resize(n + 1, 1);
	rep(i, n + 1) par[i] = i;
  }

  ll root(ll x) {
	if (x != par[x]) par[x] = root(par[x]);
	return par[x];
  }

  ll size(ll x) {
	return data[root(x)];
  }

  void unite(ll a, ll b) {
	ll x = root(a);
	ll y = root(b);
	if (x == y) return;
	if (rank[x] > rank[y]) {
	  par[y] = x;
	  data[x] += data[y];
	}
	else {
	  par[x] = y;
	  data[y] += data[x];
	  if (rank[x] == rank[y]) rank[y]++;
	}
  }

  bool same(ll a, ll b) {
	return root(a) == root(b);
  }
};

ll a[MAX], b[MAX], c[MAX], d[MAX];

int main(void) {
  ios::sync_with_stdio(false);
  cin.tie(0);


  ll k;
  vector<pair<ll, pll>> v;
  UnionFind u;
  cin >> n >> m >> k;
  u.set(n);
  rep(i, m) {
	cin >> a[i] >> b[i] >> c[i];
  }
  rep(i, k) {
	cin >> tmp;
	tmp--;
	d[tmp]++;
	u.unite(a[tmp], b[tmp]);
  }
  rep(i, m) {
	if (d[i]) continue;
	v.pb(make_pair(c[i], pll(a[i], b[i])));
  }
  sort(all(v));
  rep(i, v.size()) {
	if (u.same(v[i].second.first, v[i].second.second)) {
	  ans += v[i].first;
	}
	else {
	  u.unite(v[i].second.first, v[i].second.second);
	}
  }
  cout << ans << endl;


  return 0;
}
0