Submission #1801000
Source Code Expand
#include <iostream> #include <iomanip> // << fixed << setprecision(xxx) #include <algorithm> // do { } while ( next_permutation(A, A+xxx) ) ; #include <vector> #include <string> // to_string(nnn) // substr(m, n) // stoi(nnn) #include <complex> #include <tuple> // get<n>(xxx) #include <queue> #include <stack> #include <map> // if (M.find(key) != M.end()) { } #include <set> // S.insert(M); // if (S.find(key) != S.end()) { } // for (auto it=S.begin(); it != S.end(); it++) { } // auto it = S.lower_bound(M); #include <random> // random_device rd; mt19937 mt(rd()); #include <cctype> #include <cassert> #include <cmath> #include <cstdio> #include <cstdlib> // atoi(xxx) using namespace std; #define DEBUG 0 // change 0 -> 1 if we need debug. // insert #if<tab> by my emacs. #if DEBUG == 1 ... #end typedef long long ll; // const int dx[4] = {1, 0, -1, 0}; // const int dy[4] = {0, 1, 0, -1}; // const int C = 1e6+10; const ll infty = 1000000000007; typedef tuple<ll, int, int> edge; int N, Q; int A[200010]; int B[200010]; ll C[200010]; ll X[200010]; vector<edge> V; const int UFSIZE = 200010; int union_find[UFSIZE]; void init() { for (auto i=0; i<UFSIZE; i++) { union_find[i] = i; } } int root(int a) { if (a == union_find[a]) return a; return (union_find[a] = root(union_find[a])); } bool issame(int a, int b) { return root(a) == root(b); } void unite(int a, int b) { union_find[root(a)] = root(b); } bool isroot(int a) { return root(a) == a; } int main () { cin >> N >> Q; for (auto i = 0; i < Q; ++i) { cin >> A[i] >> B[i] >> C[i]; } fill(X, X+200010, infty); for (auto i = 0; i < Q; ++i) { V.push_back(edge(C[i], A[i], B[i])); X[A[i]] = min(X[A[i]], C[i]+1); X[B[i]] = min(X[B[i]], C[i]+2); } for (auto i = 0; i < 3*N; ++i) { X[(i+1)%N] = min(X[i%N]+2, X[(i+1)%N]); } for (auto i = 0; i < N; ++i) { V.push_back(edge(X[i], i, i+1)); } sort(V.begin(), V.end()); init(); int used = 0; ll ans = 0; auto it = V.begin(); while (used < N-1) { ll c = get<0>(*it); ll a = get<1>(*it); ll b = get<2>(*it); if (!issame(a, b)) { used++; ans += c; unite(a, b); } it++; } cout << ans << endl; }
Submission Info
Submission Time | |
---|---|
Task | G - Zigzag MST |
User | kazunetakahashi |
Language | C++14 (GCC 5.4.1) |
Score | 0 |
Code Size | 2323 Byte |
Status | WA |
Exec Time | 286 ms |
Memory | 15340 KB |
Judge Result
Set Name | sample | all | ||||||
---|---|---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 0 / 1300 | ||||||
Status |
|
|
Set Name | Test Cases |
---|---|
sample | sample-01.txt, sample-02.txt, sample-03.txt |
all | sample-01.txt, sample-02.txt, sample-03.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt, 01-15.txt, 01-16.txt, 01-17.txt, 01-18.txt, 01-19.txt, 01-20.txt, 01-21.txt, 01-22.txt, 01-23.txt, 01-24.txt, 01-25.txt, 01-26.txt, 01-27.txt, 01-28.txt, 01-29.txt, 01-30.txt, sample-01.txt, sample-02.txt, sample-03.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
01-01.txt | AC | 3 ms | 4352 KB |
01-02.txt | AC | 171 ms | 10608 KB |
01-03.txt | WA | 269 ms | 13292 KB |
01-04.txt | WA | 26 ms | 9068 KB |
01-05.txt | AC | 18 ms | 8684 KB |
01-06.txt | WA | 23 ms | 8940 KB |
01-07.txt | WA | 22 ms | 9712 KB |
01-08.txt | WA | 25 ms | 9072 KB |
01-09.txt | WA | 40 ms | 8048 KB |
01-10.txt | WA | 154 ms | 12780 KB |
01-11.txt | WA | 257 ms | 14444 KB |
01-12.txt | WA | 276 ms | 13676 KB |
01-13.txt | WA | 276 ms | 13548 KB |
01-14.txt | WA | 280 ms | 13548 KB |
01-15.txt | WA | 279 ms | 14700 KB |
01-16.txt | WA | 280 ms | 14188 KB |
01-17.txt | WA | 277 ms | 14700 KB |
01-18.txt | AC | 255 ms | 13420 KB |
01-19.txt | WA | 21 ms | 9712 KB |
01-20.txt | WA | 36 ms | 7920 KB |
01-21.txt | WA | 98 ms | 9968 KB |
01-22.txt | WA | 242 ms | 13420 KB |
01-23.txt | WA | 245 ms | 14572 KB |
01-24.txt | AC | 30 ms | 8432 KB |
01-25.txt | AC | 276 ms | 15340 KB |
01-26.txt | AC | 26 ms | 9200 KB |
01-27.txt | AC | 42 ms | 8816 KB |
01-28.txt | WA | 189 ms | 12652 KB |
01-29.txt | WA | 268 ms | 13676 KB |
01-30.txt | WA | 286 ms | 14700 KB |
sample-01.txt | AC | 3 ms | 4352 KB |
sample-02.txt | AC | 3 ms | 4352 KB |
sample-03.txt | AC | 3 ms | 4352 KB |