Submission #1034814


Source Code Expand

#include <bits/stdc++.h>

using namespace std;

#define REP(i,a,b) for(int i=a;i<(int)b;i++)
#define rep(i,n) REP(i,0,n)
#define all(c) (c).begin(), (c).end()
#define zero(a) memset(a, 0, sizeof a)
#define minus(a) memset(a, -1, sizeof a)
#define watch(a) { cout << #a << " = " << a << endl; }
template<class T1, class T2> inline bool minimize(T1 &a, T2 b) { return b < a && (a = b, 1); }
template<class T1, class T2> inline bool maximize(T1 &a, T2 b) { return a < b && (a = b, 1); }
template<class T> void operator>> (istream& ist, vector<T>& vs) { for(auto& e: vs) cin >> e; }

typedef long long ll;
int const inf = 1<<29;

namespace tree {
struct union_find {
  vector<int> par, rank, size;
  int compnum;

  union_find(int N) {
    compnum = N;
    par.resize(N), rank.resize(N), size.resize(N);
    for(int i=0; i<N; i++) {
      par[i] = i;
      rank[i] = 0;
      size[i] = 1;
    }
  }

  int root(int x) {
    return par[x] == x ? x : par[x] = root(par[x]);
  }

  void unite(int x, int y) {
    x = root(x), y = root(y);
    if(x == y) return;
    if(rank[x] < rank[y]) {
      par[x] = y, size[y] += size[x];
    } else {
      par[y] = x, size[x] += size[y];
      if(rank[x] == rank[y]) rank[x]++;
    }
    compnum--;
  }

  int operator[](int x) { return root(x); }
  void operator()(int x, int y) { return unite(x, y); }

  bool same(int x, int y) { return root(x) == root(y); }
  int size_of(int x) { return size[root(x)]; }
  int num_of_comps() { return compnum; }
};
}

int main() {

  int N, M; cin >> N >> M;
  vector<tuple<int, int, int>> es;
  rep(i, M) {
    int a, b, c; scanf("%d%d%d", &a, &b, &c);
    a--, b--;
    es.emplace_back(c, a, b);
  }

  sort(es.begin(), es.end());

  int Q; cin >> Q;
  rep(i, Q) {
    int s, t; cin >> s >> t; s--, t--;
    tree::union_find uf(N);
    uf.unite(s, t);
    ll ans = 0;
    rep(k, M) {
      int a, b, c; tie(c, a, b) = es[k];
      if(!uf.same(a, b)) {
        uf.unite(a, b);
        ans += c;
      }
    }

    cout << ans << endl;
  }
  
  return 0;
}

Submission Info

Submission Time
Task A - Graph
User motxx
Language C++14 (GCC 5.4.1)
Score 200
Code Size 2115 Byte
Status TLE
Exec Time 3155 ms
Memory 6512 KB

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:63:45: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     int a, b, c; scanf("%d%d%d", &a, &b, &c);
                                             ^

Judge Result

Set Name Sample subtask1 subtask2 All
Score / Max Score 0 / 0 200 / 200 0 / 300 0 / 200
Status
AC × 2
AC × 12
AC × 16
TLE × 5
AC × 16
TLE × 13
Set Name Test Cases
Sample sample_1.txt, sample_2.txt
subtask1 sample_2.txt, subtask_1_1.txt, subtask_1_10.txt, subtask_1_11.txt, subtask_1_2.txt, subtask_1_3.txt, subtask_1_4.txt, subtask_1_5.txt, subtask_1_6.txt, subtask_1_7.txt, subtask_1_8.txt, subtask_1_9.txt
subtask2 sample_1.txt, sample_2.txt, subtask_1_1.txt, subtask_1_10.txt, subtask_1_11.txt, subtask_1_2.txt, subtask_1_3.txt, subtask_1_4.txt, subtask_1_5.txt, subtask_1_6.txt, subtask_1_7.txt, subtask_1_8.txt, subtask_1_9.txt, subtask_2_1.txt, subtask_2_2.txt, subtask_2_3.txt, subtask_2_4.txt, subtask_2_5.txt, subtask_2_6.txt, subtask_2_7.txt, subtask_2_8.txt
All sample_1.txt, sample_2.txt, subtask_1_1.txt, subtask_1_10.txt, subtask_1_11.txt, subtask_1_2.txt, subtask_1_3.txt, subtask_1_4.txt, subtask_1_5.txt, subtask_1_6.txt, subtask_1_7.txt, subtask_1_8.txt, subtask_1_9.txt, subtask_2_1.txt, subtask_2_2.txt, subtask_2_3.txt, subtask_2_4.txt, subtask_2_5.txt, subtask_2_6.txt, subtask_2_7.txt, subtask_2_8.txt, subtask_3_1.txt, subtask_3_2.txt, subtask_3_3.txt, subtask_3_4.txt, subtask_3_5.txt, subtask_3_6.txt, subtask_3_7.txt, subtask_3_8.txt
Case Name Status Exec Time Memory
sample_1.txt AC 3 ms 256 KB
sample_2.txt AC 3 ms 256 KB
subtask_1_1.txt AC 3 ms 256 KB
subtask_1_10.txt AC 74 ms 3444 KB
subtask_1_11.txt AC 3 ms 256 KB
subtask_1_2.txt AC 17 ms 1148 KB
subtask_1_3.txt AC 147 ms 6512 KB
subtask_1_4.txt AC 4 ms 384 KB
subtask_1_5.txt AC 5 ms 384 KB
subtask_1_6.txt AC 38 ms 1912 KB
subtask_1_7.txt AC 146 ms 6512 KB
subtask_1_8.txt AC 4 ms 384 KB
subtask_1_9.txt AC 6 ms 512 KB
subtask_2_1.txt TLE 3154 ms 6512 KB
subtask_2_2.txt TLE 3154 ms 6512 KB
subtask_2_3.txt TLE 3155 ms 6512 KB
subtask_2_4.txt TLE 3155 ms 6512 KB
subtask_2_5.txt AC 260 ms 384 KB
subtask_2_6.txt AC 575 ms 704 KB
subtask_2_7.txt AC 1530 ms 1912 KB
subtask_2_8.txt TLE 3155 ms 6512 KB
subtask_3_1.txt TLE 3155 ms 6512 KB
subtask_3_2.txt TLE 3154 ms 6512 KB
subtask_3_3.txt TLE 3154 ms 896 KB
subtask_3_4.txt TLE 3154 ms 704 KB
subtask_3_5.txt TLE 3154 ms 3444 KB
subtask_3_6.txt TLE 3154 ms 6512 KB
subtask_3_7.txt TLE 3154 ms 6512 KB
subtask_3_8.txt TLE 3155 ms 6512 KB