結果
問題 | No.2987 Colorful University of Tree |
ユーザー | 👑 rin204 |
提出日時 | 2024-12-12 21:50:43 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 840 ms / 5,000 ms |
コード長 | 22,810 bytes |
コンパイル時間 | 5,087 ms |
コンパイル使用メモリ | 284,600 KB |
実行使用メモリ | 57,708 KB |
最終ジャッジ日時 | 2024-12-12 21:51:20 |
合計ジャッジ時間 | 28,729 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 133 ms
5,248 KB |
testcase_03 | AC | 151 ms
5,248 KB |
testcase_04 | AC | 143 ms
5,248 KB |
testcase_05 | AC | 498 ms
36,204 KB |
testcase_06 | AC | 77 ms
12,672 KB |
testcase_07 | AC | 101 ms
12,032 KB |
testcase_08 | AC | 536 ms
37,120 KB |
testcase_09 | AC | 95 ms
13,056 KB |
testcase_10 | AC | 804 ms
50,656 KB |
testcase_11 | AC | 821 ms
48,040 KB |
testcase_12 | AC | 753 ms
57,708 KB |
testcase_13 | AC | 823 ms
48,000 KB |
testcase_14 | AC | 709 ms
47,872 KB |
testcase_15 | AC | 707 ms
54,460 KB |
testcase_16 | AC | 788 ms
52,404 KB |
testcase_17 | AC | 840 ms
51,328 KB |
testcase_18 | AC | 803 ms
48,036 KB |
testcase_19 | AC | 806 ms
47,992 KB |
testcase_20 | AC | 806 ms
48,128 KB |
testcase_21 | AC | 765 ms
48,080 KB |
testcase_22 | AC | 653 ms
48,000 KB |
testcase_23 | AC | 581 ms
57,672 KB |
testcase_24 | AC | 211 ms
7,756 KB |
testcase_25 | AC | 212 ms
7,944 KB |
testcase_26 | AC | 210 ms
7,784 KB |
testcase_27 | AC | 207 ms
7,708 KB |
testcase_28 | AC | 211 ms
7,784 KB |
testcase_29 | AC | 9 ms
5,248 KB |
testcase_30 | AC | 51 ms
5,248 KB |
testcase_31 | AC | 50 ms
5,248 KB |
testcase_32 | AC | 117 ms
5,248 KB |
testcase_33 | AC | 145 ms
9,216 KB |
testcase_34 | AC | 182 ms
13,952 KB |
testcase_35 | AC | 209 ms
25,856 KB |
testcase_36 | AC | 726 ms
45,308 KB |
testcase_37 | AC | 60 ms
9,344 KB |
testcase_38 | AC | 276 ms
25,728 KB |
testcase_39 | AC | 61 ms
8,948 KB |
testcase_40 | AC | 528 ms
38,656 KB |
99_evil_hack_000.txt | AC | 2 ms
5,248 KB |
ソースコード
// #pragma GCC target("avx2") // #pragma GCC optimize("O3") // #pragma GCC optimize("unroll-loops") // #define INTERACTIVE #include <bits/stdc++.h> using namespace std; namespace templates { // type using ll = long long; using ull = unsigned long long; using Pii = pair<int, int>; using Pil = pair<int, ll>; using Pli = pair<ll, int>; using Pll = pair<ll, ll>; template <class T> using pq = priority_queue<T>; template <class T> using qp = priority_queue<T, vector<T>, greater<T>>; // clang-format off #define vec(T, A, ...) vector<T> A(__VA_ARGS__); #define vvec(T, A, h, ...) vector<vector<T>> A(h, vector<T>(__VA_ARGS__)); #define vvvec(T, A, h1, h2, ...) vector<vector<vector<T>>> A(h1, vector<vector<T>>(h2, vector<T>(__VA_ARGS__))); // clang-format on // for loop #define fori1(a) for (ll _ = 0; _ < (a); _++) #define fori2(i, a) for (ll i = 0; i < (a); i++) #define fori3(i, a, b) for (ll i = (a); i < (b); i++) #define fori4(i, a, b, c) for (ll i = (a); ((c) > 0 || i > (b)) && ((c) < 0 || i < (b)); i += (c)) #define overload4(a, b, c, d, e, ...) e #define fori(...) overload4(__VA_ARGS__, fori4, fori3, fori2, fori1)(__VA_ARGS__) // declare and input // clang-format off #define INT(...) int __VA_ARGS__; inp(__VA_ARGS__); #define LL(...) ll __VA_ARGS__; inp(__VA_ARGS__); #define STRING(...) string __VA_ARGS__; inp(__VA_ARGS__); #define CHAR(...) char __VA_ARGS__; inp(__VA_ARGS__); #define DOUBLE(...) double __VA_ARGS__; STRING(str___); __VA_ARGS__ = stod(str___); #define VEC(T, A, n) vector<T> A(n); inp(A); #define VVEC(T, A, n, m) vector<vector<T>> A(n, vector<T>(m)); inp(A); // clang-format on // const value const ll MOD1 = 1000000007; const ll MOD9 = 998244353; const double PI = acos(-1); // other macro #if !defined(RIN__LOCAL) && !defined(INTERACTIVE) #define endl "\n" #endif #define spa ' ' #define len(A) ll(A.size()) #define all(A) begin(A), end(A) // function vector<char> stoc(string &S) { int n = S.size(); vector<char> ret(n); for (int i = 0; i < n; i++) ret[i] = S[i]; return ret; } string ctos(vector<char> &S) { int n = S.size(); string ret = ""; for (int i = 0; i < n; i++) ret += S[i]; return ret; } template <class T> auto min(const T &a) { return *min_element(all(a)); } template <class T> auto max(const T &a) { return *max_element(all(a)); } template <class T, class S> auto clamp(T &a, const S &l, const S &r) { return (a > r ? r : a < l ? l : a); } template <class T, class S> inline bool chmax(T &a, const S &b) { return (a < b ? a = b, 1 : 0); } template <class T, class S> inline bool chmin(T &a, const S &b) { return (a > b ? a = b, 1 : 0); } template <class T, class S> inline bool chclamp(T &a, const S &l, const S &r) { auto b = clamp(a, l, r); return (a != b ? a = b, 1 : 0); } template <typename T> T sum(vector<T> &A) { T tot = 0; for (auto a : A) tot += a; return tot; } template <typename T> vector<T> compression(vector<T> X) { sort(all(X)); X.erase(unique(all(X)), X.end()); return X; } // input and output namespace io { // __int128_t std::ostream &operator<<(std::ostream &dest, __int128_t value) { std::ostream::sentry s(dest); if (s) { __uint128_t tmp = value < 0 ? -value : value; char buffer[128]; char *d = std::end(buffer); do { --d; *d = "0123456789"[tmp % 10]; tmp /= 10; } while (tmp != 0); if (value < 0) { --d; *d = '-'; } int len = std::end(buffer) - d; if (dest.rdbuf()->sputn(d, len) != len) { dest.setstate(std::ios_base::badbit); } } return dest; } // vector<T> template <typename T> istream &operator>>(istream &is, vector<T> &A) { for (auto &a : A) is >> a; return is; } template <typename T> ostream &operator<<(ostream &os, vector<T> &A) { for (size_t i = 0; i < A.size(); i++) { os << A[i]; if (i != A.size() - 1) os << ' '; } return os; } // vector<vector<T>> template <typename T> istream &operator>>(istream &is, vector<vector<T>> &A) { for (auto &a : A) is >> a; return is; } template <typename T> ostream &operator<<(ostream &os, vector<vector<T>> &A) { for (size_t i = 0; i < A.size(); i++) { os << A[i]; if (i != A.size() - 1) os << endl; } return os; } // pair<S, T> template <typename S, typename T> istream &operator>>(istream &is, pair<S, T> &A) { is >> A.first >> A.second; return is; } template <typename S, typename T> ostream &operator<<(ostream &os, pair<S, T> &A) { os << A.first << ' ' << A.second; return os; } // vector<pair<S, T>> template <typename S, typename T> istream &operator>>(istream &is, vector<pair<S, T>> &A) { for (size_t i = 0; i < A.size(); i++) { is >> A[i]; } return is; } template <typename S, typename T> ostream &operator<<(ostream &os, vector<pair<S, T>> &A) { for (size_t i = 0; i < A.size(); i++) { os << A[i]; if (i != A.size() - 1) os << endl; } return os; } // tuple template <typename T, size_t N> struct TuplePrint { static ostream &print(ostream &os, const T &t) { TuplePrint<T, N - 1>::print(os, t); os << ' ' << get<N - 1>(t); return os; } }; template <typename T> struct TuplePrint<T, 1> { static ostream &print(ostream &os, const T &t) { os << get<0>(t); return os; } }; template <typename... Args> ostream &operator<<(ostream &os, const tuple<Args...> &t) { TuplePrint<decltype(t), sizeof...(Args)>::print(os, t); return os; } // io functions void FLUSH() { cout << flush; } void print() { cout << endl; } template <class Head, class... Tail> void print(Head &&head, Tail &&...tail) { cout << head; if (sizeof...(Tail)) cout << spa; print(std::forward<Tail>(tail)...); } template <typename T, typename S> void prisep(vector<T> &A, S sep) { int n = A.size(); for (int i = 0; i < n; i++) { cout << A[i]; if (i != n - 1) cout << sep; } cout << endl; } template <typename T, typename S> void priend(T A, S end) { cout << A << end; } template <typename T> void prispa(T A) { priend(A, spa); } template <typename T, typename S> bool printif(bool f, T A, S B) { if (f) print(A); else print(B); return f; } template <class... T> void inp(T &...a) { (cin >> ... >> a); } } // namespace io using namespace io; // read graph vector<vector<int>> read_edges(int n, int m, bool direct = false, int indexed = 1) { vector<vector<int>> edges(n, vector<int>()); for (int i = 0; i < m; i++) { INT(u, v); u -= indexed; v -= indexed; edges[u].push_back(v); if (!direct) edges[v].push_back(u); } return edges; } vector<vector<int>> read_tree(int n, int indexed = 1) { return read_edges(n, n - 1, false, indexed); } template <typename T = long long> vector<vector<pair<int, T>>> read_wedges(int n, int m, bool direct = false, int indexed = 1) { vector<vector<pair<int, T>>> edges(n, vector<pair<int, T>>()); for (int i = 0; i < m; i++) { INT(u, v); T w; inp(w); u -= indexed; v -= indexed; edges[u].push_back({v, w}); if (!direct) edges[v].push_back({u, w}); } return edges; } template <typename T = long long> vector<vector<pair<int, T>>> read_wtree(int n, int indexed = 1) { return read_wedges<T>(n, n - 1, false, indexed); } // yes / no namespace yesno { // yes inline bool yes(bool f = true) { cout << (f ? "yes" : "no") << endl; return f; } inline bool Yes(bool f = true) { cout << (f ? "Yes" : "No") << endl; return f; } inline bool YES(bool f = true) { cout << (f ? "YES" : "NO") << endl; return f; } // no inline bool no(bool f = true) { cout << (!f ? "yes" : "no") << endl; return f; } inline bool No(bool f = true) { cout << (!f ? "Yes" : "No") << endl; return f; } inline bool NO(bool f = true) { cout << (!f ? "YES" : "NO") << endl; return f; } // possible inline bool possible(bool f = true) { cout << (f ? "possible" : "impossible") << endl; return f; } inline bool Possible(bool f = true) { cout << (f ? "Possible" : "Impossible") << endl; return f; } inline bool POSSIBLE(bool f = true) { cout << (f ? "POSSIBLE" : "IMPOSSIBLE") << endl; return f; } // impossible inline bool impossible(bool f = true) { cout << (!f ? "possible" : "impossible") << endl; return f; } inline bool Impossible(bool f = true) { cout << (!f ? "Possible" : "Impossible") << endl; return f; } inline bool IMPOSSIBLE(bool f = true) { cout << (!f ? "POSSIBLE" : "IMPOSSIBLE") << endl; return f; } // Alice Bob inline bool Alice(bool f = true) { cout << (f ? "Alice" : "Bob") << endl; return f; } inline bool Bob(bool f = true) { cout << (f ? "Bob" : "Alice") << endl; return f; } // Takahashi Aoki inline bool Takahashi(bool f = true) { cout << (f ? "Takahashi" : "Aoki") << endl; return f; } inline bool Aoki(bool f = true) { cout << (f ? "Aoki" : "Takahashi") << endl; return f; } } // namespace yesno using namespace yesno; } // namespace templates using namespace templates; void solve() { INT(n); vvec(int, edges, n); vec(int, deg, n); vec(Pii, E, n - 1, {-1, -1}); fori(i, n) { cin >> deg[i]; fori(deg[i]) { INT(x); x--; if (E[x].first == -1) { E[x].first = i; } else { E[x].second = i; } edges[i].push_back(x); } } vec(int, idx, n); iota(all(idx), 0); sort(all(idx), [&](int i, int j) { return deg[i] > deg[j]; }); ll x = 0; while (x * x < 2 * n - 2) x++; ll y = x + 2; for (auto d : deg) chmax(x, d); auto vertex_color = [&](ll x) -> vector<ll> { vec(ll, color, n, -1); ll y_ = min(x, y); vec(ll, cnt, y_, x); for (auto i : idx) { ll ma = -1; int ii = -1; fori(j, y_) { if (chmax(ma, cnt[j])) { ii = j; } } if (ma < deg[i]) { return {}; } cnt[ii] -= deg[i]; color[i] = ii; } return color; }; vec(ll, V_color, 0); while (V_color.empty()) { V_color = vertex_color(x); x++; } x--; for (auto &[u, v] : E) { if (V_color[u] > V_color[v]) swap(u, v); } vvec(int, Es, x); vvec(int, same, x); fori(i, n - 1) { auto [u, v] = E[i]; if (V_color[u] == V_color[v]) { same[V_color[u]].push_back(i); } else { Es[V_color[u]].push_back(i); Es[V_color[v]].push_back(i); } } vec(Pii, E_color, n - 1, {-1, -1}); fori(t, x) { int a = 0; int b = 1; sort(all(Es[t]), [&](int i, int j) { auto [u1, v1] = E[i]; auto [u2, v2] = E[j]; return V_color[u1] + V_color[v1] < V_color[u2] + V_color[v2]; }); bool op2 = false; fori(i, len(Es[t]) - 1) { int j1 = Es[t][i]; int j2 = Es[t][i + 1]; int u1 = E[j1].first; int u2 = E[j2].first; if (V_color[u1] == V_color[u2]) { op2 = true; break; } } if (!same[t].empty()) { op2 = false; } for (auto j : Es[t]) { auto [u, v] = E[j]; if (V_color[u] == t) { op2 = false; break; } } if (!op2) { for (auto j : Es[t]) { auto [u, v] = E[j]; if (a > b) swap(a, b); int c = b + 1; if (t == V_color[u]) { E_color[j].first = a; a = c; } else { if (a != E_color[j].first) { E_color[j].second = a; a = c; } else { E_color[j].second = b; b = c; } } } for (auto j : same[t]) { auto [u, v] = E[j]; E_color[j] = {a, b}; int c = max(a, b) + 1; int d = c + 1; a = c; b = d; } } if (op2) { int idx = -1; fori(i, len(Es[t]) - 1) { int j1 = Es[t][i]; int j2 = Es[t][i + 1]; int u1 = E[j1].first; int u2 = E[j2].first; if (V_color[u1] == V_color[u2]) { idx = i; break; } } int a = 0; int b = 1; fori(i, len(Es[t])) { if (i == idx or i == idx + 1) continue; int j = Es[t][i]; auto [u, v] = E[j]; if (a > b) swap(a, b); int c = b + 1; if (t == V_color[u]) { E_color[j].first = a; a = c; } else { if (a != E_color[j].first) { E_color[j].second = a; a = c; } else { E_color[j].second = b; b = c; } } } { if (E_color[Es[t][idx]].first != a and E_color[Es[t][idx + 1]].first != b) { E_color[Es[t][idx]].second = a; E_color[Es[t][idx + 1]].second = b; } else { E_color[Es[t][idx]].second = b; E_color[Es[t][idx + 1]].second = a; } } } } print(x); fori(i, n) { vec(int, out, deg[i] + 1, -1); out[0] = V_color[i] + 1; fori(j, deg[i]) { int ii = edges[i][j]; if (i == E[ii].first) { out[j + 1] = E_color[ii].first + 1; } else { out[j + 1] = E_color[ii].second + 1; } } print(out); } { fori(i, n - 1) { assert(E_color[i].first != E_color[i].second); } map<Pii, Pii> mp; fori(i, n) { vec(int, out, deg[i] + 1, -1); out[0] = V_color[i] + 1; fori(j, deg[i]) { int ii = edges[i][j]; assert(0 <= E_color[ii].first and E_color[ii].first < x); assert(0 <= E_color[ii].second and E_color[ii].second < x); if (i == E[ii].first) { out[j + 1] = E_color[ii].first + 1; } else { out[j + 1] = E_color[ii].second + 1; } assert(!mp.count({out[0], out[j + 1]})); mp[{out[0], out[j + 1]}] = {i, j}; } } } } int main() { #ifndef INTERACTIVE std::cin.tie(0)->sync_with_stdio(0); #endif // std::cout << std::fixed << std::setprecision(12); int t; t = 1; std::cin >> t; while (t--) solve(); return 0; } // // #pragma GCC target("avx2") // // #pragma GCC optimize("O3") // // #pragma GCC optimize("unroll-loops") // // #define INTERACTIVE // // #include "kyopro-cpp/template.hpp" // // void solve() { // INT(n); // vvec(int, edges, n); // vec(int, deg, n); // vec(Pii, E, n - 1, {-1, -1}); // fori(i, n) { // cin >> deg[i]; // fori(deg[i]) { // INT(x); // x--; // if (E[x].first == -1) { // E[x].first = i; // } else { // E[x].second = i; // } // edges[i].push_back(x); // } // } // // vec(int, idx, n); // iota(all(idx), 0); // sort(all(idx), [&](int i, int j) { return deg[i] > deg[j]; }); // ll x = 0; // while (x * x < 2 * n - 2) x++; // ll y = x + 2; // for (auto d : deg) chmax(x, d); // // auto vertex_color = [&](ll x) -> vector<ll> { // vec(ll, color, n, -1); // ll y_ = min(x, y); // vec(ll, cnt, y_, x); // for (auto i : idx) { // ll ma = -1; // int ii = -1; // // fori(j, y_) { // if (chmax(ma, cnt[j])) { // ii = j; // } // } // if (ma < deg[i]) { // return {}; // } // // cnt[ii] -= deg[i]; // color[i] = ii; // } // return color; // }; // // vec(ll, V_color, 0); // while (V_color.empty()) { // V_color = vertex_color(x); // x++; // } // x--; // // for (auto &[u, v] : E) { // if (V_color[u] > V_color[v]) swap(u, v); // } // // vvec(int, Es, x); // vvec(int, same, x); // // fori(i, n - 1) { // auto [u, v] = E[i]; // if (V_color[u] == V_color[v]) { // same[V_color[u]].push_back(i); // } else { // Es[V_color[u]].push_back(i); // Es[V_color[v]].push_back(i); // } // } // // vec(Pii, E_color, n - 1, {-1, -1}); // // fori(t, x) { // int a = 0; // int b = 1; // // sort(all(Es[t]), [&](int i, int j) { // auto [u1, v1] = E[i]; // auto [u2, v2] = E[j]; // return V_color[u1] + V_color[v1] < V_color[u2] + V_color[v2]; // }); // // bool op2 = false; // fori(i, len(Es[t]) - 1) { // int j1 = Es[t][i]; // int j2 = Es[t][i + 1]; // int u1 = E[j1].first; // int u2 = E[j2].first; // // if (V_color[u1] == V_color[u2]) { // op2 = true; // break; // } // } // // if (!same[t].empty()) { // op2 = false; // } // for (auto j : Es[t]) { // auto [u, v] = E[j]; // if (V_color[u] == t) { // op2 = false; // break; // } // } // // if (!op2) { // for (auto j : Es[t]) { // auto [u, v] = E[j]; // // if (a > b) swap(a, b); // int c = b + 1; // if (t == V_color[u]) { // E_color[j].first = a; // a = c; // } else { // if (a != E_color[j].first) { // E_color[j].second = a; // a = c; // } else { // E_color[j].second = b; // b = c; // } // } // } // for (auto j : same[t]) { // auto [u, v] = E[j]; // E_color[j] = {a, b}; // int c = max(a, b) + 1; // int d = c + 1; // a = c; // b = d; // } // } // // if (op2) { // int idx = -1; // fori(i, len(Es[t]) - 1) { // int j1 = Es[t][i]; // int j2 = Es[t][i + 1]; // int u1 = E[j1].first; // int u2 = E[j2].first; // // if (V_color[u1] == V_color[u2]) { // idx = i; // break; // } // } // // int a = 0; // int b = 1; // fori(i, len(Es[t])) { // if (i == idx or i == idx + 1) continue; // int j = Es[t][i]; // auto [u, v] = E[j]; // if (a > b) swap(a, b); // // int c = b + 1; // if (t == V_color[u]) { // E_color[j].first = a; // a = c; // } else { // if (a != E_color[j].first) { // E_color[j].second = a; // a = c; // } else { // E_color[j].second = b; // b = c; // } // } // } // { // if (E_color[Es[t][idx]].first != a and E_color[Es[t][idx + 1]].first != b) { // E_color[Es[t][idx]].second = a; // E_color[Es[t][idx + 1]].second = b; // } else { // E_color[Es[t][idx]].second = b; // E_color[Es[t][idx + 1]].second = a; // } // } // } // } // // print(x); // // fori(i, n) { // vec(int, out, deg[i] + 1, -1); // out[0] = V_color[i] + 1; // fori(j, deg[i]) { // int ii = edges[i][j]; // if (i == E[ii].first) { // out[j + 1] = E_color[ii].first + 1; // } else { // out[j + 1] = E_color[ii].second + 1; // } // } // print(out); // } // // { // fori(i, n - 1) { // assert(E_color[i].first != E_color[i].second); // } // map<Pii, Pii> mp; // fori(i, n) { // vec(int, out, deg[i] + 1, -1); // out[0] = V_color[i] + 1; // fori(j, deg[i]) { // int ii = edges[i][j]; // assert(0 <= E_color[ii].first and E_color[ii].first < x); // assert(0 <= E_color[ii].second and E_color[ii].second < x); // if (i == E[ii].first) { // out[j + 1] = E_color[ii].first + 1; // } else { // out[j + 1] = E_color[ii].second + 1; // } // assert(!mp.count({out[0], out[j + 1]})); // mp[{out[0], out[j + 1]}] = {i, j}; // } // } // } // } // // int main() { // #ifndef INTERACTIVE // std::cin.tie(0)->sync_with_stdio(0); // #endif // // std::cout << std::fixed << std::setprecision(12); // int t; // t = 1; // std::cin >> t; // while (t--) solve(); // return 0; // }