結果
問題 | No.1170 Never Want to Walk |
ユーザー | kyon2326 |
提出日時 | 2022-03-24 13:59:21 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 523 ms / 2,000 ms |
コード長 | 22,768 bytes |
コンパイル時間 | 5,468 ms |
コンパイル使用メモリ | 283,880 KB |
実行使用メモリ | 338,396 KB |
最終ジャッジ日時 | 2024-10-12 22:10:37 |
合計ジャッジ時間 | 11,889 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,816 KB |
testcase_02 | AC | 2 ms
6,816 KB |
testcase_03 | AC | 2 ms
6,820 KB |
testcase_04 | AC | 2 ms
6,816 KB |
testcase_05 | AC | 2 ms
6,816 KB |
testcase_06 | AC | 2 ms
6,820 KB |
testcase_07 | AC | 1 ms
6,820 KB |
testcase_08 | AC | 2 ms
6,816 KB |
testcase_09 | AC | 2 ms
6,816 KB |
testcase_10 | AC | 2 ms
6,820 KB |
testcase_11 | AC | 2 ms
6,820 KB |
testcase_12 | AC | 3 ms
6,816 KB |
testcase_13 | AC | 3 ms
6,816 KB |
testcase_14 | AC | 3 ms
6,820 KB |
testcase_15 | AC | 3 ms
6,820 KB |
testcase_16 | AC | 3 ms
6,820 KB |
testcase_17 | AC | 3 ms
6,816 KB |
testcase_18 | AC | 3 ms
6,816 KB |
testcase_19 | AC | 3 ms
6,816 KB |
testcase_20 | AC | 3 ms
6,816 KB |
testcase_21 | AC | 3 ms
6,820 KB |
testcase_22 | AC | 3 ms
6,820 KB |
testcase_23 | AC | 4 ms
6,820 KB |
testcase_24 | AC | 4 ms
6,820 KB |
testcase_25 | AC | 3 ms
6,816 KB |
testcase_26 | AC | 3 ms
6,816 KB |
testcase_27 | AC | 274 ms
125,512 KB |
testcase_28 | AC | 272 ms
126,124 KB |
testcase_29 | AC | 262 ms
119,740 KB |
testcase_30 | AC | 276 ms
126,208 KB |
testcase_31 | AC | 273 ms
123,448 KB |
testcase_32 | AC | 489 ms
314,308 KB |
testcase_33 | AC | 495 ms
304,180 KB |
testcase_34 | AC | 464 ms
291,816 KB |
testcase_35 | AC | 523 ms
338,396 KB |
testcase_36 | AC | 487 ms
312,376 KB |
testcase_37 | AC | 451 ms
277,308 KB |
testcase_38 | AC | 434 ms
254,336 KB |
コンパイルメッセージ
main.cpp:172:57: warning: 'template<class _Category, class _Tp, class _Distance, class _Pointer, class _Reference> struct std::iterator' is deprecated [-Wdeprecated-declarations] 172 | template <typename... T> class zip_iterator_base : std::iterator<std::forward_iterator_tag, std::tuple<decltype(*std::declval<T>().begin())...>> { | ^~~~~~~~ In file included from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/bits/stl_algobase.h:65, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/bits/specfun.h:45, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/cmath:1935, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/x86_64-pc-linux-gnu/bits/stdc++.h:41, from main.cpp:4: /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/bits/stl_iterator_base_types.h:127:34: note: declared here 127 | struct _GLIBCXX17_DEPRECATED iterator | ^~~~~~~~
ソースコード
#define _USE_MATH_DEFINES #define _EXT_CODECVT_SPECIALIZATIONS_H 1 #define _EXT_ENC_FILEBUF_H 1 #include <bits/stdc++.h> #include <bits/extc++.h> using namespace std; /* #include <atcoder/all> using namespace atcoder; */ /* #include <boost/multiprecision/cpp_int.hpp> #include <boost/multiprecision/cpp_dec_float.hpp> using bll = boost::multiprecision::cpp_int; using bdouble = boost::multiprecision::number<boost::multiprecision::cpp_dec_float<100>>; using namespace boost::multiprecision; */ //#define int long long using ll = long long; //constexpr ll MOD = (ll)1e9 + 7; //primitive root = 5 constexpr ll MOD = 998244353; //primitive root = 3 //INT_MAX = (1<<31)-1 = 2147483647, INT64_MAX = (1LL<<63)-1 = 9223372036854775807 constexpr ll INF = std::numeric_limits<ll>::max() == INT_MAX ? (ll)1e9 + 7 : (ll)1e18; constexpr double EPS = 1e-9; constexpr ll dx[4] = {1, 0, -1, 0}; constexpr ll dy[4] = {0, 1, 0, -1}; constexpr ll dx8[8] = {1, 0, -1, 0, 1, 1, -1, -1}; constexpr ll dy8[8] = {0, 1, 0, -1, 1, -1, 1, -1}; #if defined(LOCAL_TEST) || defined(LOCAL_DEV) #define BOOST_STACKTRACE_USE_ADDR2LINE #define BOOST_STACKTRACE_ADDR2LINE_LOCATION /usr/local/opt/binutils/bin/addr2line #define _GNU_SOURCE 1 #include <boost/stacktrace.hpp> #endif #ifdef LOCAL_TEST namespace std { template <typename T> class dvector : public std::vector<T> { public: using std::vector<T>::vector; template <typename T_ = T, typename std::enable_if_t<std::is_same_v<T_, bool>, std::nullptr_t> = nullptr> std::vector<bool>::reference operator[](std::size_t n) { if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n); } template <typename T_ = T, typename std::enable_if_t<std::is_same_v<T_, bool>, std::nullptr_t> = nullptr> const T_ operator[](std::size_t n) const { if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n); } template <typename T_ = T, typename std::enable_if_t<!std::is_same_v<T_, bool>, std::nullptr_t> = nullptr> T_& operator[](std::size_t n) { if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n); } template <typename T_ = T, typename std::enable_if_t<!std::is_same_v<T_, bool>, std::nullptr_t> = nullptr> const T_& operator[](std::size_t n) const { if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n); } }; template <typename T, typename Compare = std::less<T>, typename Allocator = std::allocator<T>> class dmultiset : public std::multiset<T,Compare,Allocator> { public: using std::multiset<T,Compare,Allocator>::multiset; const typename std::multiset<T,Compare,Allocator>::iterator erase(const typename std::multiset<T,Compare,Allocator>::iterator it) { return std::multiset<T,Compare,Allocator>::erase(it); } std::size_t erase([[maybe_unused]] const T& x) { std::cerr << boost::stacktrace::stacktrace() << '\n'; assert(false); } std::size_t erase_all_elements(const T& x) { return std::multiset<T,Compare,Allocator>::erase(x); } }; } #define vector dvector #define multiset dmultiset class SIGFPE_exception : std::exception {}; class SIGSEGV_exception : std::exception {}; void catch_SIGFPE([[maybe_unused]] int e) { std::cerr << boost::stacktrace::stacktrace() << '\n'; throw SIGFPE_exception(); } void catch_SIGSEGV([[maybe_unused]] int e) { std::cerr << boost::stacktrace::stacktrace() << '\n'; throw SIGSEGV_exception(); } signed convertedmain(); signed main() { signal(SIGFPE, catch_SIGFPE); signal(SIGSEGV, catch_SIGSEGV); return convertedmain(); } #define main() convertedmain() #else #define erase_all_elements erase #endif #ifdef LOCAL_DEV template <typename T1, typename T2> std::ostream& operator<<(std::ostream& s, const std::pair<T1, T2>& p) { return s << "(" << p.first << ", " << p.second << ")"; } template <typename T, std::size_t N> std::ostream& operator<<(std::ostream& s, const std::array<T, N>& a) { s << "{ "; for (std::size_t i = 0; i < N; ++i){ s << a[i] << "\t"; } s << "}"; return s; } template <typename T, typename Compare> std::ostream& operator<<(std::ostream& s, const std::set<T, Compare>& se) { s << "{ "; for (auto itr = se.begin(); itr != se.end(); ++itr){ s << (*itr) << "\t"; } s << "}"; return s; } template <typename T, typename Compare> std::ostream& operator<<(std::ostream& s, const std::multiset<T, Compare>& se) { s << "{ "; for (auto itr = se.begin(); itr != se.end(); ++itr){ s << (*itr) << "\t"; } s << "}"; return s; } template <typename T1, typename T2, typename Compare> std::ostream& operator<<(std::ostream& s, const std::map<T1, T2, Compare>& m) { s << "{\n"; for (auto itr = m.begin(); itr != m.end(); ++itr){ s << "\t" << (*itr).first << " : " << (*itr).second << "\n"; } s << "}"; return s; } template <typename T> std::ostream& operator<<(std::ostream& s, const std::deque<T>& v) { for (std::size_t i = 0; i < v.size(); ++i){ s << v[i]; if (i < v.size() - 1) s << "\t"; } return s; } template <typename T> std::ostream& operator<<(std::ostream& s, const std::vector<T>& v) { for (std::size_t i = 0; i < v.size(); ++i){ s << v[i]; if (i < v.size() - 1) s << "\t"; } return s; } template <typename T> std::ostream& operator<<(std::ostream& s, const std::vector<std::vector<T>>& vv) { s << "\\\n"; for (std::size_t i = 0; i < vv.size(); ++i){ s << vv[i] << "\n"; } return s; } template <typename T, std::size_t N, typename std::enable_if_t<!std::is_same_v<T, char>, std::nullptr_t> = nullptr> std::ostream& operator<<(std::ostream& s, const T (&v)[N]) { for (std::size_t i = 0; i < N; ++i){ s << v[i]; if (i < N - 1) s << "\t"; } return s; } template <typename T, std::size_t N, std::size_t M, typename std::enable_if_t<!std::is_same_v<T, char>, std::nullptr_t> = nullptr> std::ostream& operator<<(std::ostream& s, const T (&vv)[N][M]) { s << "\\\n"; for (std::size_t i = 0; i < N; ++i){ s << vv[i] << "\n"; } return s; } #if __has_include(<ext/pb_ds/assoc_container.hpp>) template <typename Key, typename Compare> std::ostream& operator<<(std::ostream& s, const __gnu_pbds::tree<Key, __gnu_pbds::null_type, Compare, __gnu_pbds::rb_tree_tag, __gnu_pbds::tree_order_statistics_node_update>& se) { s << "{ "; for (auto itr = se.begin(); itr != se.end(); ++itr){ s << (*itr) << "\t"; } s << "}"; return s; } template <typename Key, typename T, typename Hash> std::ostream& operator<<(std::ostream& s, const __gnu_pbds::gp_hash_table<Key, T, Hash>& m) { s << "{\n"; for (auto itr = m.begin(); itr != m.end(); ++itr){ s << "\t" << (*itr).first << " : " << (*itr).second << "\n"; } s << "}"; return s; } #endif void debug_impl() { std::cerr << '\n'; } template <typename Head, typename... Tail> void debug_impl(Head& head, Tail&... tail) { std::cerr << " " << head << (sizeof...(tail) ? "," : ""); debug_impl(tail...); } template <typename Head, typename... Tail> void debug_impl(const Head& head, const Tail&... tail) { std::cerr << " " << head << (sizeof...(tail) ? "," : ""); debug_impl(tail...); } #define debug(...) do { std::cerr << ":" << __LINE__ << " (" << #__VA_ARGS__ << ") ="; debug_impl(__VA_ARGS__); } while (false) constexpr inline long long prodlocal([[maybe_unused]] long long prod, [[maybe_unused]] long long local) { return local; } #else #define debug(...) do {} while (false) constexpr inline long long prodlocal([[maybe_unused]] long long prod, [[maybe_unused]] long long local) { return prod; } #endif #define repoverload3(_1, _2, _3, name, ...) name #define rep3(i, a, b) for(ll i=(a), i##_length=(b); i<i##_length; ++i) #define rep2(i, n) rep3(i, 0, n) #define rep1(n) rep3(i, 0, n) #define rep(...) repoverload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__) #define repeq3(i, a, b) rep3(i, (a)+1, (b)+1) #define repeq2(i, n) rep3(i, 1, (n)+1) #define repeq1(n) rep3(i, 1, (n)+1) #define repeq(...) repoverload3(__VA_ARGS__, repeq3, repeq2, repeq1)(__VA_ARGS__) #define rrep3(i, a, b) for(ll i=(b)-1; i>=(a); --i) #define rrep2(i, n) rrep3(i, 0, n) #define rrep1(n) rrep3(i, 0, n) #define rrep(...) repoverload3(__VA_ARGS__, rrep3, rrep2, rrep1)(__VA_ARGS__) #define rrepeq3(i, a, b) rrep3(i, (a)+1, (b)+1) #define rrepeq2(i, n) rrep3(i, 1, (n)+1) #define rrepeq1(n) rrep3(i, 1, (n)+1) #define rrepeq(...) repoverload3(__VA_ARGS__, rrepeq3, rrepeq2, rrepeq1)(__VA_ARGS__) #define all(v) std::begin(v), std::end(v) #define rall(v) std::rbegin(v), std::rend(v) void p() { std::cout << '\n'; } template <typename Head, typename... Tail> void p(Head& head, Tail&... tail) { std::cout << head << (sizeof...(tail) ? " " : ""); p(tail...); } template <typename Head, typename... Tail> void p(const Head& head, const Tail&... tail) { std::cout << head << (sizeof...(tail) ? " " : ""); p(tail...); } template <typename T> inline void pv(const T& v) { for(ll i=0, N=std::size(v); i<N; i++) std::cout << v[i] << " \n"[i==N-1]; } template <typename T> inline bool chmax(T& a, T b) { return a < b && (a = b, true); } template <typename T> inline bool chmin(T& a, T b) { return a > b && (a = b, true); } template <typename T> inline void uniq(T& v) { std::sort(v.begin(), v.end()); v.erase(std::unique(v.begin(), v.end()), v.end()); } template <typename T> inline ll sz(const T& v) { return std::size(v); } template <typename T, std::size_t N> std::vector<T> make_vector_impl(std::vector<ll>& sizes, typename std::enable_if<(N==1), const T&>::type x) { return std::vector<T>(sizes.front(),x); } template <typename T, std::size_t N> auto make_vector_impl(std::vector<ll>& sizes, typename std::enable_if<(N>1), const T&>::type x) { ll size=sizes.back(); sizes.pop_back(); return std::vector<decltype(make_vector_impl<T,N-1>(sizes,x))>(size,make_vector_impl<T,N-1>(sizes,x)); } template <typename T, std::size_t N> auto make_vector(const ll (&sizes)[N], const T& x=T()) { std::vector<ll> s(N); for(std::size_t i=0; i<N; ++i)s[i]=sizes[N-1-i]; return make_vector_impl<T,N>(s,x); } #if __has_include(<ext/pb_ds/assoc_container.hpp>) class custom_hash { public: static uint64_t splitmix64(uint64_t x) { x += 0x9e3779b97f4a7c15, x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9, x = (x ^ (x >> 27)) * 0x94d049bb133111eb; return x ^ (x >> 31); } size_t operator()(uint64_t x) const { static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count(); return splitmix64(x + FIXED_RANDOM); } }; template <typename Key, typename Mapped, typename Hash = custom_hash, typename std::enable_if_t<std::is_integral_v<Key>, std::nullptr_t> = nullptr> class fmap : public __gnu_pbds::gp_hash_table<Key, Mapped, Hash> { public: using __gnu_pbds::gp_hash_table<Key, Mapped, Hash>::gp_hash_table; template <typename T> fmap(std::initializer_list<std::initializer_list<T>> il) : __gnu_pbds::gp_hash_table<Key, Mapped, Hash>() { for (auto&& x : il) __gnu_pbds::gp_hash_table<Key, Mapped, Hash>::insert(std::pair<Key, Mapped>(*x.begin(), *(x.begin() + 1))); } template <typename T> ll count(const T& x) const { return __gnu_pbds::gp_hash_table<Key, Mapped, Hash>::find(x) != __gnu_pbds::gp_hash_table<Key, Mapped, Hash>::end(); } }; #else template <typename Key, typename Mapped> using fmap = std::map<Key, Mapped>; #endif template <typename... T> class zip_iterator_base : std::iterator<std::forward_iterator_tag, std::tuple<decltype(*std::declval<T>().begin())...>> { public: ll idx_; std::tuple<decltype(std::declval<T>().begin())...> iters_; template <std::size_t... I> auto deref(std::index_sequence<I...>) const { return typename zip_iterator_base::value_type{*std::get<I>(iters_)...}; } template <std::size_t... I> void increment(std::index_sequence<I...>) { [[maybe_unused]] auto l = {(++std::get<I>(iters_), 0)...}; } explicit zip_iterator_base(decltype(iters_) iters) : idx_(0), iters_{std::move(iters)} {} zip_iterator_base& operator++() { ++idx_; increment(std::index_sequence_for<T...>{}); return *this; } zip_iterator_base operator++(int) { auto saved{*this}; ++idx_; increment(std::index_sequence_for<T...>{}); return saved; } bool operator!=(const zip_iterator_base& other) const { return iters_ != other.iters_; } }; template <typename... T> class zip_iterator : public zip_iterator_base<T...> { public: using zip_iterator_base<T...>::zip_iterator_base; auto operator*() const { return this->deref(std::index_sequence_for<T...>{}); } }; template <typename... T> class zipindex_iterator : public zip_iterator_base<T...> { public: using zip_iterator_base<T...>::zip_iterator_base; auto operator*() const { return std::tuple_cat(std::make_tuple(this->idx_), this->deref(std::index_sequence_for<T...>{})); } }; template <typename IteratorType, typename... T> class zip_helper { public: zip_helper(T&... seqs) : begin_{std::make_tuple(seqs.begin()...)}, end_{std::make_tuple(seqs.end()...)} {} IteratorType begin() const { return begin_; } IteratorType end() const { return end_; } IteratorType begin_, end_; }; template <typename... T> auto zip(T&&... seqs) { return zip_helper<zip_iterator<T...>, T...>{seqs...}; } template <typename... T> auto zipindex(T&&... seqs) { return zip_helper<zipindex_iterator<T...>, T...>{seqs...}; } /*-----8<-----template-----8<-----*/ //[lib]unionfind.cpp class UnionFind { public: vector<ll> v; UnionFind() = default; UnionFind(size_t size) : v(size, -1) {} ll root(ll x) { return (v[x] < 0 ? x : v[x] = root(v[x])); } bool is_root(ll x) { return x == root(x); } bool is_same(ll x, ll y) { return root(x) == root(y); } bool unite(ll x, ll y) { x = root(x); y = root(y); if (x == y) return false; if (v[x] > v[y]) swap(x, y); v[x] += v[y]; v[y] = x; return true; } ll size(ll x) { return -v[root(x)]; } }; std::ostream& operator<<(std::ostream& s, const UnionFind& uf) { for (size_t i = 0; i < uf.v.size(); ++i){ s << uf.v[i]; if (i < uf.v.size() - 1) s << "\t"; } return s; } //[lib](internal)edge.cpp using EdgeCostType = ll; using usize = ll; template <class T> class edge_type { public: usize from, to; T cost; edge_type() {} edge_type(usize from, usize to, T cost) : from(from), to(to), cost(cost) {} bool operator<(const edge_type& r) const { return r.cost < cost; } }; using Edge = edge_type<EdgeCostType>; ostream& operator<<(ostream& s, const Edge& e) { s << "{ " << e.from << " -> " << e.to << ", " << e.cost << " }"; return s; } inline void addedge(vector<vector<Edge>>& g, usize from, usize to, EdgeCostType cost) { g[from].emplace_back(from, to, cost); g[to].emplace_back(to, from, cost); } //最短路木の親頂点を元にstart->goalの経路を作成 vector<ll> buildPath(const vector<ll> &prev, ll goal) { vector<ll> path; for (ll u = goal; u >= 0; u = prev[u]) path.push_back(u); reverse(path.begin(), path.end()); return path; } //[lib]dijkstra.cpp //[depends on](internal)edge.cpp template <class T> class fibonacci_heap { class node_type; using node_ptr = node_type *; class node_type { public: node_ptr parent; node_ptr child; node_ptr left; node_ptr right; usize rank; bool mark; T key; usize prev; node_type() : parent(nullptr), child(nullptr), left(nullptr), right(nullptr), rank(0), mark(false), key(std::numeric_limits<T>::max()), prev(-1) {} }; vector<node_type> nodes; node_ptr root; vector<node_ptr> table; public: fibonacci_heap(const usize n) : nodes(n), root(nullptr), table(std::ceil(std::log(n + 1) * 2.08), nullptr) {} bool empty() const { return root == nullptr; } edge_type<T> pop() { edge_type<T> ret = {root->prev, static_cast<usize>(root - nodes.data()), root->key}; usize max = 0; const auto push = [&](node_ptr v) -> void { while (true) { node_ptr u = table[v->rank]; if (u == nullptr) { table[v->rank] = v; break; } table[v->rank] = nullptr; if (u->key < v->key) { std::swap(u, v); } const node_ptr c = v->child; if (c == nullptr) { u->left = u; u->right = u; v->child = u; } else { u->left = c->left; u->right = c; c->left->right = u; c->left = u; } u->parent = v; v->rank += 1; } max = std::max(max, v->rank + 1); }; { node_ptr v = root->right; while (v != root) { const node_ptr next = v->right; push(v); v = next; } } if (root->child != nullptr) { node_ptr v = root->child; do { const node_ptr next = v->right; v->mark = false; push(v); v = next; } while (v != root->child); } root = nullptr; for (usize i = 0; i != max; i += 1) { const node_ptr v = table[i]; if (v == nullptr) { continue; } table[i] = nullptr; v->parent = nullptr; if (root == nullptr) { root = v; v->left = v; v->right = v; } else { v->left = root->left; v->right = root; root->left->right = v; root->left = v; if (root->key > v->key) { root = v; } } } return ret; } void update_key(const usize v_, const T key, const usize prev) { node_ptr v = &nodes[v_]; if (v->key <= key) { return; } v->key = key; v->prev = prev; if (v->left == nullptr) { if (root == nullptr) { v->left = v; v->right = v; root = v; } else { v->left = root->left; v->right = root; root->left->right = v; root->left = v; if (key < root->key) { root = v; } } return; } if (v->parent == nullptr) { if (key < root->key) { root = v; } return; } else { if (v->parent->key <= key) { return; } } while (true) { const node_ptr p = v->parent; v->left->right = v->right; v->right->left = v->left; v->parent = nullptr; p->rank -= 1; if (p->child == v) { if (p->rank == 0) { p->child = nullptr; } else { p->child = v->right; } } v->left = root->left; v->right = root; root->left->right = v; root->left = v; v->mark = false; v = p; if (v->parent == nullptr) { break; } if (!v->mark) { v->mark = true; break; } } if (root->key > key) { root = &nodes[v_]; } } }; /* 計算量:O(E+VlogV) 引数 g:探索するグラフ start:探索するスタートノード番号 戻り値 dist:スタートノードから各頂点までの距離 prev:最短路木の親頂点 */ void dijkstra(const vector<vector<Edge>> &g, ll start, vector<EdgeCostType> &dist, vector<ll> &prev) { dist.assign(g.size(), INF); dist[start] = 0; prev.assign(g.size(), -1); fibonacci_heap<EdgeCostType> heap(g.size()); heap.update_key(start, 0, -1); while (!heap.empty()) { const auto top = heap.pop(); dist[top.to] = top.cost; prev[top.to] = top.from; for (const auto &edge : g[top.to]) { heap.update_key(edge.to, top.cost + edge.cost, edge.from); } } } template <class Func> void dijkstra(const ll gsize, const Func &gfunc, ll start, vector<EdgeCostType> &dist, vector<ll> &prev) { dist.assign(gsize, INF); dist[start] = 0; prev.assign(gsize, -1); fibonacci_heap<EdgeCostType> heap(gsize); heap.update_key(start, 0, -1); while (!heap.empty()) { const auto top = heap.pop(); dist[top.to] = top.cost; prev[top.to] = top.from; gfunc(top.to, [&](const usize nextv, const EdgeCostType nextcost) { heap.update_key(nextv, top.cost + nextcost, top.to); }); } } //[lib]dijkstra区間.cpp //[depends on]dijkstra.cpp //セグ木の形にして区間に辺を張る class Graph { public: ll N, sz; //この g にグラフを作るので、これを使ってdijkstraしてください vector<vector<Edge>> g; Graph (ll N) : N(N) { sz = 1; while (sz < N) sz <<= 1; g.resize(sz*3-2); ll diff=sz; for(ll i=0; i<sz/2-1; i++){ g[i*2+1+diff].emplace_back(i*2+1+diff, i+diff, 0); g[i*2+2+diff].emplace_back(i*2+2+diff, i+diff, 0); } for(ll i=sz*3/2-1; i<sz*2-1; i++){ ll t=(i-(sz*3/2-1))*2; g[t].emplace_back(t, i, 0); g[t+1].emplace_back(t+1, i, 0); } diff=sz*2-1; for(ll i=0; i<sz/2-1; i++){ g[i+diff].emplace_back(i+diff, i*2+1+diff, 0); g[i+diff].emplace_back(i+diff, i*2+2+diff, 0); } for(ll i=sz*5/2-2; i<sz*3-2; i++){ ll t=(i-(sz*5/2-2))*2; g[i].emplace_back(i, t, 0); g[i].emplace_back(i, t+1, 0); } }; //[froml, fromr) -> [tol, tor) に 距離costの辺を張る void add_edge(ll froml, ll fromr, ll tol, ll tor, EdgeCostType cost) { if(froml>=fromr || tol>=tor) return; ll newv = g.size(); g.emplace_back(); connect(froml, fromr, -1, newv, sz-1, cost); connect(tol, tor, newv, -1, sz*2-2, 0); } private: void connect(ll l, ll r, ll from, ll to, ll diff, EdgeCostType cost) { bool isfirst = true; for (l += sz, r += sz; l < r; l >>= 1, r >>= 1, isfirst = false) { if (l & 1) { ll v = isfirst ? l-sz : l+diff; connect_edge(from, to, v, cost); l++; } if (r & 1) { --r; ll v = isfirst ? r-sz : r+diff; connect_edge(from, to, v, cost); } } } void connect_edge(ll from, ll to, ll v, EdgeCostType cost) { if (from != -1) { g[from].emplace_back(from, v, cost); } else { g[v].emplace_back(v, to, cost); } } }; /*-----8<-----library-----8<-----*/ void solve() { ll N, A, B; cin >> N >> A >> B; vector<ll> a(N); rep(i, N) cin >> a[i]; Graph graph(N); rep(i,N){ { ll l = lower_bound(all(a), a[i] - B) - a.begin(); ll r = upper_bound(all(a), a[i] - A) - a.begin(); graph.add_edge(i, i + 1, l, r, 1); } { ll l = lower_bound(all(a), a[i] + A) - a.begin(); ll r = upper_bound(all(a), a[i] + B) - a.begin(); graph.add_edge(i, i + 1, l, r, 1); } } auto& g = graph.g; UnionFind uf(g.size()); vector<ll> seen(g.size(), INF); rep(k,N){ if (seen[k] != INF) continue; vector<ll> v; queue<ll> que; ll startpos = k; que.push(startpos); seen[k] = 0; while(!que.empty()) { auto x = que.front(); if (x < N) v.push_back(x); que.pop(); for(auto&& e:g[x]){ // seen[nx][ny]の書き換えはここでやること if(chmin(seen[e.to], seen[x] + 1)) { que.push(e.to); } } } rep(i,v.size()-1){ uf.unite(v[i], v[i + 1]); } } rep(i,N){ p(uf.size(i)); } std::cout << std::flush; _Exit(0); } signed main() { #ifndef LOCAL_DEV std::cin.tie(nullptr); std::ios::sync_with_stdio(false); #endif //ll Q; cin >> Q; while(Q--)solve(); solve(); return 0; }