Submission #2369159


Source Code Expand

//#include <bits/stdc++.h>

#include <iostream>
#include <algorithm>

#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <unordered_set>
#include <unordered_map>
#include <stack>
#include <queue>
#include <deque>
#include <cstring>
#include <string>
#include <utility>
#include <array>
#include <complex>
#include <valarray>

#include <cassert>
#include <cmath>
#include <functional>
#include <iomanip>
#include <chrono>
#include <random>
#include <numeric>


using namespace std;
#define int long long

typedef long long ll;
typedef unsigned long long ull;
//typedef unsigned __int128 HASH;
typedef pair<int,int> pii;
typedef pair<ll, ll> pll;
typedef pair<ull, ull> pullull;
typedef pair<ll,int> plli;
typedef pair<long double, int> pdbi;
typedef pair<int,pii> pipii;
typedef pair<ll,pll> plpll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<vi> vvi;
typedef vector<vvi> vvvi;
typedef vector<pii> vpii;
typedef vector<vector<int>> mat;

#define rep(i,n) for (int i=0;i<(n);i++)
#define rep2(i,a,b) for (int i=(a);i<(b);i++)
#define rrep(i,n) for (int i=(n);i>0;i--)
#define rrep2(i,a,b) for (int i=(a);i>b;i--)
#define pb push_back
#define fi first
#define se second
#define all(a) (a).begin(),(a).end()
#define rall(a) (a).rbegin(),(a).rend()

const ll hmod1 = 999999937;
const ll hmod2 = 1000000000 + 9;
const int INF = 1<<30;
const ll INFLL = 1LL<<62;
const long double EPS = 1e-12;
const ll mod = 1000000000 + 7;
const int dx4[4] = {1, 0, -1, 0};
const int dy4[4] = {0, 1, 0, -1};
const int dx8[8] = {1, 1, 1, 0, 0, -1, -1, -1};
const int dy8[8] = {0, 1, -1, 1, -1, 0, 1, -1};
const long double pi = 3.141592653589793;

#define addm(X, Y) (X) = ((X) + ((Y) % mod) + mod) % mod
#define inside(y, x, h, w) (0 <= (y) && (y) < (h) && 0 <= (x) && (x) < (w)) ? true : false

//debug
#define DEBUG

#define DUMPOUT cout

#ifdef DEBUG
#define dump(...) DUMPOUT<<#__VA_ARGS__<<" :["<<__FUNCTION__<<":"<<__LINE__<<"]"<<endl; DUMPOUT<<"    "; dump_func(__VA_ARGS__)
#else
#define dump(...)
#endif

void dump_func() {DUMPOUT << endl;};

template <class Head, class... Tail> void dump_func(Head&& head, Tail&&... tail) {
    DUMPOUT << head;
    if (sizeof...(Tail) == 0) DUMPOUT << " ";
    else DUMPOUT << ", ";
    dump_func(std::move(tail)...);
}

//ostream
template<typename T> ostream& operator << (ostream& os, vector<T>& vec) {
    os << "["; for (int i = 0; i<vec.size(); i++) os << vec[i] << (i + 1 == vec.size() ? "" : ", "); os << "]";
    return os;
}

template<typename T, typename U> ostream& operator << (ostream& os, pair<T, U>& pair_var) {
    os << "(" << pair_var.first << ", " << pair_var.second << ")";
    return os;
}

template<typename T, typename U> ostream& operator << (ostream& os, map<T, U>& map_var) {
    os << "[";
    for (auto itr = map_var.begin(); itr != map_var.end(); itr++) {
        os << "(" << itr->first << ", " << itr->second << ")"; itr++;  if(itr != map_var.end()) os << ", "; itr--;
    }
    os << "]";
    return os;
}

template<typename T> ostream& operator << (ostream& os, set<T>& set_var) {
    os << "[";
    for (auto itr = set_var.begin(); itr != set_var.end(); itr++) {
        os << *itr; ++itr; if(itr != set_var.end()) os << ", "; itr--;
    }
    os << "]";
    return os;
}

int n, m;
int q;

struct edge{int cost, u, v;};
vector<edge> edges;
bool comp(const edge& e1, const edge& e2) {
    return e1.cost < e2.cost;
}

struct edge1{int to, cost;};
vector<vector<edge1>> g(4000);

struct UnionFind {
  vector<int> par, rank;
  UnionFind(int size) : par(size), rank(size)
  {
    rep(i, size) par[i] = i;
  }

  int find_root(int x) {
    if (par[x] != x) return par[x] = find_root(par[x]);
    else return x;
  }

  bool same(int x, int y) {
    return find_root(x) == find_root(y);
  }

  void unite(int x, int y) {
    int u = find_root(x);
    int v = find_root(y);
    if (rank[u] > rank[v]) swap(u, v);
    par[u] = v;
    if (rank[u] == rank[v]) rank[v]++;
  }
};

bool dfs(int now, int par, int t, int &ans) {
  if (now == t) return true;
  for (auto nx : g[now]) {
    if (nx.to == par) continue;
    if (dfs(nx.to, now, t, ans)) {
      ans = max(ans, nx.cost);
      return true;
    }
  }
  return false;
}


signed main(){
  cin.tie(0);
  ios::sync_with_stdio(false);
  cin >> n >> m;
  rep(i, m) {
    int a, b, c;
    cin >> a >> b >> c;
    a--; b--;
    edges.push_back({c, a, b});
  }
  sort(all(edges), comp);
  int sum_cost = 0;
  UnionFind uf(n);
  rep(i, m) {
    edge e = edges[i];
    if (uf.same(e.u, e.v)) continue;
    uf.unite(e.u, e.v);
    g[e.u].push_back({e.v, e.cost});
    g[e.v].push_back({e.u, e.cost});
    sum_cost += e.cost;
  }

  cin >> q;
  rep(i, q) {
    int s, t;
    cin >> s >> t;
    s--; t--;
    int ans = 0;
    dfs(s, -1, t, ans);
    cout << sum_cost - ans << endl;
  }
}

Submission Info

Submission Time
Task A - Graph
User roto_37
Language C++14 (GCC 5.4.1)
Score 500
Code Size 5027 Byte
Status TLE
Exec Time 3155 ms
Memory 14828 KB

Judge Result

Set Name Sample subtask1 subtask2 All
Score / Max Score 0 / 0 200 / 200 300 / 300 0 / 200
Status
AC × 2
AC × 12
AC × 21
AC × 23
TLE × 8
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, 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 1 ms 384 KB
sample_2.txt AC 1 ms 384 KB
subtask_1_1.txt AC 2 ms 384 KB
subtask_1_10.txt AC 74 ms 7152 KB
subtask_1_11.txt AC 1 ms 384 KB
subtask_1_2.txt AC 16 ms 2040 KB
subtask_1_3.txt AC 146 ms 13932 KB
subtask_1_4.txt AC 3 ms 768 KB
subtask_1_5.txt AC 4 ms 704 KB
subtask_1_6.txt AC 40 ms 3572 KB
subtask_1_7.txt AC 145 ms 13676 KB
subtask_1_8.txt AC 3 ms 768 KB
subtask_1_9.txt AC 5 ms 892 KB
subtask_2_1.txt AC 295 ms 14572 KB
subtask_2_2.txt AC 292 ms 14572 KB
subtask_2_3.txt AC 288 ms 13420 KB
subtask_2_4.txt AC 296 ms 14188 KB
subtask_2_5.txt AC 130 ms 768 KB
subtask_2_6.txt AC 155 ms 1276 KB
subtask_2_7.txt AC 180 ms 5108 KB
subtask_2_8.txt AC 287 ms 13164 KB
subtask_3_1.txt TLE 3155 ms 13164 KB
subtask_3_2.txt TLE 3155 ms 14828 KB
subtask_3_3.txt TLE 3155 ms 1664 KB
subtask_3_4.txt TLE 3155 ms 1660 KB
subtask_3_5.txt TLE 3155 ms 7664 KB
subtask_3_6.txt TLE 3155 ms 14188 KB
subtask_3_7.txt TLE 3155 ms 13420 KB
subtask_3_8.txt TLE 3155 ms 14700 KB