結果
問題 | No.1718 Random Squirrel |
ユーザー |
![]() |
提出日時 | 2021-10-23 03:19:19 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 102 ms / 2,000 ms |
コード長 | 6,262 bytes |
コンパイル時間 | 2,350 ms |
コンパイル使用メモリ | 212,232 KB |
最終ジャッジ日時 | 2025-01-25 04:52:32 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h>#define REP_(i, a_, b_, a, b, ...) \for (int i = (a), END_##i = (b); i < END_##i; ++i)#define REP(i, ...) REP_(i, __VA_ARGS__, __VA_ARGS__, 0, __VA_ARGS__)#define ALL(x) std::begin(x), std::end(x)using Int = long long;using Real = long double;template<typename T, typename U>inline bool chmax(T &a, U b) {return a < b and ((a = std::move(b)), true);}template<typename T, typename U>inline bool chmin(T &a, U b) {return a > b and ((a = std::move(b)), true);}template<typename T>inline int ssize(const T &a) {return (int) a.size();}inline void check(bool cond, const char *message = "!ERROR!") {if (not cond) {std::cout.flush(), std::cerr.flush();throw std::runtime_error(message);}}template<class T>inline std::ostream &print_one(const T &x, char endc) {if constexpr (std::is_same_v<T, bool>) {return std::cout << (x ? "Yes" : "No") << endc;} else {return std::cout << x << endc;}}template<class T>inline std::ostream &print(const T &x) { return print_one(x, '\n'); }template<typename T, typename... Ts>std::ostream &print(const T &head, Ts... tail) {return print_one(head, ' '), print(tail...);}inline std::ostream &print() { return std::cout << '\n'; }template<typename Container>std::ostream &print_seq(const Container &a, const char *sep = " ",const char *ends = "\n",std::ostream &os = std::cout) {auto b = std::begin(a), e = std::end(a);for (auto it = std::begin(a); it != e; ++it) {if (it != b) os << sep;os << *it;}return os << ends;}template<typename T, typename = void>struct is_iterable : std::false_type {};template<typename T>struct is_iterable<T, std::void_t<decltype(std::begin(std::declval<T>())),decltype(std::end(std::declval<T>()))>>: std::true_type {};template<typename T, typename = std::enable_if_t<is_iterable<T>::value && !std::is_same<T, std::string>::value>>std::ostream &operator<<(std::ostream &os, const T &a) {return print_seq(a, ", ", "", (os << "{")) << "}";}struct CastInput {template<typename T>operator T() const {T x;std::cin >> x;return x;}struct Sized {std::size_t n;template<typename T>operator T() const {T x(n);for (auto &e: x) std::cin >> e;return x;}};Sized operator()(std::size_t n) const { return {n}; }} const in;#ifdef MY_DEBUG#include "debug_dump.hpp"#include "backward.hpp"backward::SignalHandling kSignalHandling;#else#define DUMP(...)#define cerr if(false)std::cerr#endifusing namespace std;struct Edge {int to;};using Graph = vector<vector<Edge>>;struct Task {using NV = tuple<bool, Int, Int>; // Node Valueusing EV = NV; // Edge Valuevector<char> is_d;explicit Task(vector<char> is_d) : is_d(move(is_d)) {}EV id() const { return {false, 0, 0}; }NV add_node(const EV &ev, int v) const {auto[has_d, roundtrip, oneway] = ev;if (not has_d) {return {is_d[v], 0, 0};}return ev;}EV add_edge(const NV &nv, const Edge &e) const {auto[has_d, roundtrip, oneway] = nv;if (not has_d) return nv;return {true, roundtrip + 2, oneway + 1};}EV merge(const EV &ev1, const EV &ev2) const {auto[has_d1, roundtrip1, oneway1] = ev1;auto[has_d2, roundtrip2, oneway2] = ev2;if (not has_d1) return ev2;if (not has_d2) return ev1;return {true, roundtrip1 + roundtrip2, min(roundtrip1 + oneway2, roundtrip2 + oneway1)};}};template<typename Rerootable = Task>class Rerooter {private:using NV = typename Rerootable::NV;using EV = typename Rerootable::EV;Rerootable task;int n; // number of nodesstd::vector<std::vector<Edge>> g; // graph (tree)std::vector<NV> sub; // values for each subtree rooted at istd::vector<NV> full; // values for each entire tree rooted at iint base_root; // base root node where we start DFSpublic:explicit Rerooter(Rerootable task, std::vector<std::vector<Edge>> g,int r = 0): task(move(task)),n((int) g.size()),g(move(g)),sub(n),full(n),base_root(r) {}const std::vector<NV> &run() {pull_up(base_root, -1);push_down(base_root, -1, std::nullopt);return full;}private:NV pull_up(int v, int par) {EV res = task.id();for (auto &e: g[v]) {int u = e.to;if (u == par) continue;auto sub = pull_up(u, v);res = task.merge(res, task.add_edge(std::move(sub), e));}return (sub[v] = task.add_node(res, v));}void push_down(int v, int par, std::optional<NV> upper_sub) {int m = g[v].size();std::vector<EV> cuml(m + 1, task.id()), cumr(m + 1, task.id());for (int i = 0; i < m; ++i) {auto &e = g[v][i];int u = e.to;if (u == par) {assert(upper_sub.has_value());cuml[i + 1] = task.merge(cuml[i], task.add_edge(*upper_sub, e));} else {cuml[i + 1] = task.merge(cuml[i], task.add_edge(sub[u], e));}}for (int i = m - 1; i >= 0; --i) {auto &e = g[v][i];int u = e.to;if (u == par) {cumr[i] = task.merge(task.add_edge(*upper_sub, e), cumr[i + 1]);} else {cumr[i] = task.merge(task.add_edge(sub[u], e), cumr[i + 1]);}}full[v] = task.add_node(cuml[m], v);for (int i = 0; i < m; ++i) {auto &e = g[v][i];int u = e.to;if (u == par) continue;std::optional<NV> next_upper_sub{task.add_node(task.merge(cuml[i], cumr[i + 1]), v)};push_down(u, v, std::move(next_upper_sub));}}};auto solve() {int n = in, K = in;Graph g(n);REP(i, n - 1) {int u = in, v = in;--u, --v;g[u].push_back({v});g[v].push_back({u});}vector<char> D(n);REP(i, K) {int d = in;--d;D[d] = true;}Task task(D);Rerooter<> rerooter(task, g);auto res = rerooter.run();REP(v, n) {print(get<2>(res[v]));}}int main() {ios_base::sync_with_stdio(false), cin.tie(nullptr);solve();}