結果
問題 | No.2598 Kadomatsu on Tree |
ユーザー | 👑 bo9chan |
提出日時 | 2024-01-02 18:32:49 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 489 ms / 2,000 ms |
コード長 | 27,429 bytes |
コンパイル時間 | 2,406 ms |
コンパイル使用メモリ | 183,288 KB |
実行使用メモリ | 54,368 KB |
最終ジャッジ日時 | 2024-09-27 17:51:35 |
合計ジャッジ時間 | 20,791 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,816 KB |
testcase_02 | AC | 2 ms
6,812 KB |
testcase_03 | AC | 2 ms
6,816 KB |
testcase_04 | AC | 2 ms
6,944 KB |
testcase_05 | AC | 2 ms
6,944 KB |
testcase_06 | AC | 2 ms
6,944 KB |
testcase_07 | AC | 2 ms
6,940 KB |
testcase_08 | AC | 2 ms
6,944 KB |
testcase_09 | AC | 2 ms
6,948 KB |
testcase_10 | AC | 2 ms
6,944 KB |
testcase_11 | AC | 2 ms
6,944 KB |
testcase_12 | AC | 2 ms
6,944 KB |
testcase_13 | AC | 9 ms
6,944 KB |
testcase_14 | AC | 9 ms
6,944 KB |
testcase_15 | AC | 8 ms
6,940 KB |
testcase_16 | AC | 10 ms
6,944 KB |
testcase_17 | AC | 5 ms
6,940 KB |
testcase_18 | AC | 8 ms
6,944 KB |
testcase_19 | AC | 10 ms
6,940 KB |
testcase_20 | AC | 6 ms
6,944 KB |
testcase_21 | AC | 7 ms
6,944 KB |
testcase_22 | AC | 9 ms
6,944 KB |
testcase_23 | AC | 456 ms
47,924 KB |
testcase_24 | AC | 485 ms
48,968 KB |
testcase_25 | AC | 469 ms
48,928 KB |
testcase_26 | AC | 468 ms
48,172 KB |
testcase_27 | AC | 466 ms
48,168 KB |
testcase_28 | AC | 470 ms
48,832 KB |
testcase_29 | AC | 479 ms
49,056 KB |
testcase_30 | AC | 465 ms
47,932 KB |
testcase_31 | AC | 482 ms
49,080 KB |
testcase_32 | AC | 463 ms
48,476 KB |
testcase_33 | AC | 478 ms
49,388 KB |
testcase_34 | AC | 457 ms
47,592 KB |
testcase_35 | AC | 485 ms
49,832 KB |
testcase_36 | AC | 482 ms
49,508 KB |
testcase_37 | AC | 481 ms
49,344 KB |
testcase_38 | AC | 483 ms
49,848 KB |
testcase_39 | AC | 489 ms
49,724 KB |
testcase_40 | AC | 484 ms
49,720 KB |
testcase_41 | AC | 484 ms
49,724 KB |
testcase_42 | AC | 482 ms
49,720 KB |
testcase_43 | AC | 484 ms
49,716 KB |
testcase_44 | AC | 475 ms
49,712 KB |
testcase_45 | AC | 480 ms
49,848 KB |
testcase_46 | AC | 473 ms
49,720 KB |
testcase_47 | AC | 482 ms
49,716 KB |
testcase_48 | AC | 469 ms
51,960 KB |
testcase_49 | AC | 36 ms
7,552 KB |
testcase_50 | AC | 303 ms
34,168 KB |
testcase_51 | AC | 77 ms
12,392 KB |
testcase_52 | AC | 128 ms
17,280 KB |
testcase_53 | AC | 152 ms
24,460 KB |
testcase_54 | AC | 127 ms
21,152 KB |
testcase_55 | AC | 49 ms
9,940 KB |
testcase_56 | AC | 232 ms
34,704 KB |
testcase_57 | AC | 86 ms
15,096 KB |
testcase_58 | AC | 427 ms
48,272 KB |
testcase_59 | AC | 438 ms
48,952 KB |
testcase_60 | AC | 432 ms
52,088 KB |
testcase_61 | AC | 449 ms
54,368 KB |
testcase_62 | AC | 445 ms
52,140 KB |
ソースコード
#ifdef __LOCAL #include <mytemplate.hpp> // プリコンパイル済みヘッダ ~/local/include/mytemplate.hpp.gch #else #include <algorithm> #include <array> #include <bitset> #include <cassert> #include <climits> #include <cmath> #include <deque> #include <functional> #include <iomanip> #include <iostream> #include <iterator> #include <map> #include <memory> #include <numeric> #include <optional> #include <queue> #include <random> #include <set> #include <stack> #include <string> #include <utility> #include <vector> #endif #pragma GCC target "no-avx" // gcc12.2のバグ回避用。少し実行速度が落ちる using namespace std; // #include <boost/multiprecision/cpp_int.hpp> // using bint = boost::multiprecision::cpp_int; using int128 = __int128_t; using ll = long long; using vl = vector<ll>; using vvl = vector<vl>; using vb = vector<bool>; using vvb = vector<vb>; using vd = vector<double>; using vvd = vector<vd>; using vc = vector<char>; using vvc = vector<vc>; using pll = pair<ll, ll>; #define REP1(i, n) REP3(i, 0, n, 1) #define REP2(i, s, n) REP3(i, s, n, 1) #define REP3(i, s, n, d) for(ll i = (ll)(s); i < (ll)(n); i += (d)) #define REP_OVERLOAD(e1, e2, e3, e4, NAME,...) NAME #define rep(...) REP_OVERLOAD(__VA_ARGS__, REP3, REP2, REP1)(__VA_ARGS__) #define DEP1(i, n) DEP3(i, n, -1, 1) #define DEP2(i, n, s) DEP3(i, n, s, 1) #define DEP3(i, n, s, d) for(ll i = (ll)(n); (ll)(s) < i; i -= (d)) #define DEP_OVERLOAD(e1, e2, e3, e4, NAME,...) NAME #define dep(...) DEP_OVERLOAD(__VA_ARGS__, DEP3, DEP2, DEP1)(__VA_ARGS__) #define fore(e, a) for (auto&& e: (a)) #define len(a) (ll)(a).size() #define all(a) (a).begin(), (a).end() #define rall(a) (a).rbegin(), (a).rend() #define pb push_back #define eb emplace_back #define fi first #define se second #define th third /********* C O N S T A N T **********/ constexpr ll INF = LONG_LONG_MAX / 2 - 10000LL; // 4,611,686,018,427,377,903 ~= 4.6e+18, 19桁 const double PI = acos(-1); /********* P A I R **********/ template<class T, class U> inline pair<T, U>& operator+=(pair<T, U>& a, const pair<T, U> b); template<class T, class U> inline pair<T, U>& operator-=(pair<T, U>& a, const pair<T, U> b); template<class T, class U> inline pair<T, U>& operator*=(pair<T, U>& a, const pair<T, U> b); template<class T, class U> inline pair<T, U>& operator/=(pair<T, U>& a, const pair<T, U> b); template<class T, class U> inline pair<T, U>& operator%=(pair<T, U>& a, const pair<T, U> b); template<class T, class U, class V> inline pair<T, U>& operator+=(pair<T, U>& a, const V b); template<class T, class U, class V> inline pair<T, U>& operator-=(pair<T, U>& a, const V b); template<class T, class U, class V> inline pair<T, U>& operator*=(pair<T, U>& a, const V b); template<class T, class U, class V> inline pair<T, U>& operator/=(pair<T, U>& a, const V b); template<class T, class U, class V> inline pair<T, U>& operator%=(pair<T, U>& a, const V b); template<class T, class U, class V> inline pair<T, U> operator+(pair<T, U> a, const V& b); template<class T, class U, class V> inline pair<T, U> operator-(pair<T, U> a, const V& b); template<class T, class U, class V> inline pair<T, U> operator*(pair<T, U> a, const V& b); template<class T, class U, class V> inline pair<T, U> operator/(pair<T, U> a, const V& b); template<class T, class U, class V> inline pair<T, U> operator%(pair<T, U> a, const V& b); template<class T, class U> inline pair<T, U> operator+(pair<T, U> a); template<class T, class U> inline pair<T, U> operator-(pair<T, U> a); template<class T, class U> istream& operator >>(istream& stream, pair<T, U>& a); template<class T, class U> ostream& operator <<(ostream &stream, const pair<T, U>& a); /********* V E C T O R **********/ template<class T> inline vector<T>& operator+=(vector<T>& a, const vector<T> b); template<class T> inline vector<T>& operator-=(vector<T>& a, const vector<T> b); template<class T> inline vector<T>& operator*=(vector<T>& a, const vector<T> b); template<class T> inline vector<T>& operator/=(vector<T>& a, const vector<T> b); template<class T> inline vector<T>& operator%=(vector<T>& a, const vector<T> b); template<class T, class U> inline vector<T>& operator+=(vector<T>& a, const U b); template<class T, class U> inline vector<T>& operator-=(vector<T>& a, const U b); template<class T, class U> inline vector<T>& operator*=(vector<T>& a, const U b); template<class T, class U> inline vector<T>& operator/=(vector<T>& a, const U b); template<class T, class U> inline vector<T>& operator%=(vector<T>& a, const U b); template<class T, class U> inline vector<T> operator+(vector<T> a, const U& b); template<class T, class U> inline vector<T> operator-(vector<T> a, const U& b); template<class T, class U> inline vector<T> operator*(vector<T> a, const U& b); template<class T, class U> inline vector<T> operator/(vector<T> a, const U& b); template<class T, class U> inline vector<T> operator%(vector<T> a, const U& b); template<class T> inline vector<T> operator+(vector<T> a); template<class T> inline vector<T> operator-(vector<T> a); template<class T> istream& operator >>(istream& stream, vector<T>& a); template<class T> ostream& operator <<(ostream& stream, const vector<T>& v); template<class T> ostream& operator <<(ostream& stream, const vector<vector<T>>& vv); template<class T> inline T Sum(const vector<T>& v) {return reduce(all(v));} // v.size() == 0 のとき T() を返す template<class T> inline T Max(const vector<T>& v) {assert(v.size()); return *max_element(all(v));} template<class T> inline T Min(const vector<T>& v) {assert(v.size()); return *min_element(all(v));} template<class T> inline ll Argmax(const vector<T>& v) {assert(v.size()); return max_element(all(v)) - v.begin();} template<class T> inline ll Argmin(const vector<T>& v) {assert(v.size()); return min_element(all(v)) - v.begin();} template<class T, class U> inline bool Contains(const vector<T>& v, const U& a) {return find(all(v), a) != v.end();} template<class T> inline void Unique(vector<T>& v) {sort(all(v)); v.erase(unique(all(v)), v.end());} // ソートされたユニーク値のvectorに書き換える, O(NlogN). template<class T, class U> vector<T> make_vector(int n, U v) { return vector<T>(n, v); } template <class T, class... Args> auto make_vector(int n, Args... args) {auto val = make_vector<T>(args...); return make_vector<decltype(val)>(n, move(val));} /********* S E T **********/ template<class T> istream& operator >>(istream& stream, set<T>& st); template<class T> ostream& operator <<(ostream& stream, const set<T>& st); template<class T> inline T Max(const set<T>& st) {assert(st.size()); return *prev(st.end());} template<class T> inline T Min(const set<T>& st) {assert(st.size()); return *st.begin();} /********* M U L T I S E T **********/ template<class T> istream& operator >>(istream& stream, multiset<T>& st); template<class T> ostream& operator <<(ostream& stream, const multiset<T>& st); template<class T> inline T Max(const multiset<T>& st) {assert(st.size()); return *prev(st.end());} template<class T> inline T Min(const multiset<T>& st) {assert(st.size()); return *st.begin();} template<class T> inline bool EraseOne(multiset<T>& st, T x) {auto it=st.find(x); if (it!=st.end()) {st.erase(it); return true;} else return false;} // 要素xを1つ削除. 消せたらtrueを返す. /********* S T R I N G **********/ inline string ToUpper(const string& s) {string t; t.resize(s.size()); std::transform(all(s), t.begin(), ::toupper); return t; } inline string ToLower(const string& s) {string t; t.resize(s.size()); std::transform(all(s), t.begin(), ::tolower); return t; } inline char ToUpper(const char& c) {string s{c}, t; t.resize(s.size()); std::transform(all(s), t.begin(), ::toupper); return t[0]; } inline char ToLower(const char& c) {string s{c}, t; t.resize(s.size()); std::transform(all(s), t.begin(), ::tolower); return t[0]; } /********* T R I O **********/ template<class T1, class T2, class T3> struct trio { T1 first; T2 second; T3 third; // コンストラクタ・代入演算子 trio() {first=T1(); second=T2(); third=T3();} trio(const T1& x) : first(x), second(x), third(x) {} trio(const T1& x, const T2& y, const T3& z) : first(x), second(y), third(z) {} trio(const trio& t) {first=t.first; second=t.second; third=t.third;} // コピーコンストラクタ trio& operator =(const trio& t) {first=t.first; second=t.second; third=t.third; return *this;} // 比較演算子 auto operator <=>(const trio&) const = default; // 単項演算子(+ -) trio operator +() const { return *this; } trio operator -() const { return (-1) * (*this); } // 複合代入演算子 trio& operator +=(const trio& t) {first += t.first; second += t.second; third += t.third; return *this;} trio& operator -=(const trio& t) {first -= t.first; second -= t.second; third -= t.third; return *this;} trio& operator *=(const trio& t) {first *= t.first; second *= t.second; third *= t.third; return *this;} trio& operator /=(const trio& t) {first /= t.first; second /= t.second; third /= t.third; return *this;} trio& operator %=(const trio& t) {first %= t.first; second %= t.second; third %= t.third; return *this;} // 算術演算子 friend trio operator +(const trio& lhs, const trio& rhs) {return trio(lhs) += rhs;} friend trio operator -(const trio& lhs, const trio& rhs) {return trio(lhs) -= rhs;} friend trio operator *(const trio& lhs, const trio& rhs) {return trio(lhs) *= rhs;} friend trio operator /(const trio& lhs, const trio& rhs) {return trio(lhs) /= rhs;} friend trio operator %(const trio& lhs, const trio& rhs) {return trio(lhs) %= rhs;} // 入出力 template<class U1, class U2, class U3> friend istream& operator >>(istream&, trio<U1, U2, U3>&); template<class U1, class U2, class U3> friend ostream& operator <<(ostream&, const trio<U1, U2, U3>&); operator tuple<T1&, T2&, T3&>() {return tie(first, second, third);} }; using tll = trio<ll, ll, ll>; /********* R A N D O M **********/ struct Random { mt19937_64 rnd; Random() { random_device seed_gen; rnd.seed(seed_gen()); } ll randint(ll a, ll b) { // [a, b] uniform_int_distribution<ll> dist(a, b); return dist(rnd); } double randreal(double a, double b) { uniform_real_distribution<double> dist(a, b); return dist(rnd); } char randchar(char a, char b) { // [a, b] uniform_int_distribution<ll> dist(a, b); return dist(rnd); } }; /********* P R I N T **********/ template<class T> inline void print(const T& e); template<class H, class... T> inline void print(const H& h, const T&... t); template<class... T> inline void End(const T&... t); /********* D E B U G **********/ #ifdef __LOCAL #define debug(...) if(DEBUG) do{cout << '[' << #__VA_ARGS__ << "] ";debug_(__VA_ARGS__);}while(0) #else #define debug(...) #endif void dbg_(const long long& e); template<class T> void dbg_(const T& e); template<class T, class U> void dbg_(const pair<T, U>& p); template<class T1, class T2, class T3> void dbg_(const trio<T1, T2, T3>& t); template<class T> void debug_(const T& e); template<class T> void debug_(const vector<T>& v); template<class T> void debug_(const vector<vector<T>>& vv); template<class T, class U> void debug_(const map<T, U>& mp); template<class T, class U> void debug_(const vector<map<T, U>>& vm); template<class T> void debug_(const set<T>& st); template<class T> void debug_(const multiset<T>& st); template<class T> void debug_(const vector<set<T>>& vs); template<class T> void debug_(const vector<multiset<T>>& vs); template<class H, class... T> void debug_(const H& h, const T&... t); /********* O T H E R S **********/ template<class T, class U> inline bool chmin(T& a, U b) {if (a > b) {a = b; return true;} return false;} // bは値渡し! template<class T, class U> inline bool chmax(T& a, U b) {if (a < b) {a = b; return true;} return false;} template<class T, class U> inline auto Mod(const T& a, const U& m) {return (a % m + m) % m;} // 負もOK template<class T, class U> inline auto Ceil(const T& x, const U& y) {return x < 0 ? x/y : (x + y - 1) / y;} // 負もOK template<class T, class U> inline auto Floor(const T& x, const U& y) {return -(Ceil(-x, y)); } // 負もOK inline ll Isqrt(ll n) {assert(n >= 0); ll x = round(sqrt(n)); while(x * x > n) --x; return x;} inline ll Comb(ll n, ll r) {if (r < 0 || n < r) return 0; r = min(r, n - r); ll ret = 1; rep(i, r) {ret *= n - i; ret /= i + 1;} return ret;} // n=60, r=30までOK template<class T> inline T Pow(T x, ll n) {assert(n >= 0); T ret = 1; while(1) {if (n % 2) ret *= x; n /= 2; if(!n) return ret; x = x * x;}} // べき乗 mintもOK template<class T> inline T Aseries(T a, T d, ll n) {assert(n >= 0); return a * n + n * (n - 1) / 2 * d;} // 等差級数 mintもOK template<class T> inline T Gseries(T a, T r, ll n) {assert(n >= 0); if (r == 1) return a * n; else return a * (1 - Pow(r, n)) / (1 - r);} // 等比級数 mintもOK template<class T> using min_priority_queue = priority_queue<T, vector<T>, greater<T>>; inline bool Bit(ll b, int i) {assert(0 <= i && i < 64); return (b >> i) & 1;} inline ll Popcount(ll b) {return __builtin_popcountll(b);} inline ll Mask(ll n) {assert(0 <= n && n < 63); return (1LL << n) - 1LL;} // [0, n)のbitが立った64bit整数を返す。 inline ll Mask(ll n, ll m) {assert(n >= m); return Mask(n) ^ Mask(m);} // [m, n)のbitが立った64bit整数を返す。 inline void PrintYesNo(bool b) {b ? print("Yes") : print("No");} /********* M O D **********/ #include <atcoder/modint> // https://atcoder.github.io/ac-library/production/document_ja/ using namespace atcoder; using mint = modint998244353; // modint1000000007; istream& operator >>(istream& stream, mint& e) {ll n; stream >> n; e = n; return stream; } ostream& operator <<(ostream& stream, const mint& e) { stream << e.val(); return stream; } using vm = vector<mint>; using vvm = vector<vm>; /***************************************/ constexpr bool DEBUG = true; // LCAを計算(蟻本p.292). // verify済 https://onlinejudge.u-aizu.ac.jp/problems/GRL_5_C struct LCA { private: ll root; // 根ノード番号 ll N; // 頂点数 ll logN; // log2(頂点数)の小数点切り上げ vl depth_; // (v) = 根ノードから頂点vまでの最短距離 vvl parent_; // (k, 頂点v) = 頂点vの2^k個上の親の頂点番号(いない場合-1) vl siz; // 頂点pを根とする部分木のサイズ // depth_, parent_[0] & sizを計算. 返り値: pを根とする部分木サイズ ll dfs(const vvl& G, ll p, ll d) { depth_[p] = d; ll cnt = 1; for (auto& v : G[p]) { if (depth_[v] != -1) continue; parent_[0][v] = p; cnt += dfs(G, v, d + 1); } return siz[p] = cnt; } public: // G: 木フラフ, 隣接リスト表現(0-origin), root: 根ノード番号 LCA(const vvl& G, ll root) : root(root) { N = len(G); assert(0 <= root && root < N); logN = (ll)ceil(log2(N)); depth_.assign(N, -1); parent_.assign(logN + 1, vl(N, -1)); siz.resize(N); // depth_ & parent_[0]を計算 dfs(G, root, 0); // parent_[1] ... parent_[logN]を計算 rep(k, logN) { rep(v, N) { ll p = parent_[k][v]; if (p == -1) parent_[k + 1][v] = -1; else parent_[k + 1][v] = parent_[k][p]; } } } // LCAを計算。計算量O(logN) ll lca(ll u, ll v) const { assert(0 <= u && u < N); assert(0 <= v && v < N); if (depth_[u] > depth_[v]) swap(u, v); // uとvの高さをそろえる。 rep(k, logN + 1) { ll d = depth_[v] - depth_[u]; if ((d >> k) & 1) v = parent_[k][v]; } // LCAである場合、処理終了 if (u == v) return u; // uとvをLCAの1つ下まで移動する。 dep(k, logN) { if (parent_[k][u] != parent_[k][v]) { u = parent_[k][u]; v = parent_[k][v]; } } return parent_[0][u]; } // lca.lcaに同じ ll operator()(ll u, ll v) const { return lca(u, v); } // 頂点vのd個上の親の番号を取得, O(logN). 根を越えるとき-1を返す ll parent(ll v, ll d) { assert(0 <= v && v < N); if (d > depth_[v]) return -1; rep(k, logN + 1) { if ((d >> k) & 1) v = parent_[k][v]; } return v; } // 頂点vの1個上の親の番号を取得, O(1). vが根のとき-1を返す ll parent(ll v) { return parent_[0][v]; } // 頂点uとvの距離を計算 ll dist(ll u, ll v) { ll p = lca(u, v); return depth_[u] + depth_[v] - 2 * depth_[p]; } // 頂点uのrootからの深さを返す ll depth(ll u) { assert(0 <= u && u < N); return depth_[u]; } // 部分木のサイズを返す ll subtree_size(ll u) { assert(0 <= u && u < N); return siz[u]; } }; // LCA lca(G, root); // lca.lca(u, v); // 頂点uとvのLCAを取得, O(logN) // lca(u, v); // lca.lca(u, v)に同じ // lca.parent(v, d); // 頂点vのd個上の親の番号を取得, O(logN). 根を越えるとき-1を返す // lca.parent(v); // 頂点vの1個上の親の番号を取得, O(1). vが根のとき-1を返す // lca.dist(u, v); // 頂点uとvとの距離を取得, O(logN) // lca.depth(u); // 頂点uのrootからの深さを取得(rootがゼロ), O(1) // lca.subtree_size(u); // 頂点uを根とする部分木のサイズを返す int main() { cin.tie(nullptr); cout << fixed << setprecision(10); ll N; cin >> N; vvl G(N); rep(i, N - 1) { ll u, v; cin >> u >> v; u--, v--; G[u].pb(v); G[v].pb(u); } auto lca = LCA(G, 0); vl A(N); cin >> A; mint ans = 0; rep(p, N) { vl vec1, vec2; fore(v, G[p]) { if (lca.parent(v) == p) { ll n = lca.subtree_size(v); if (A[p] < A[v]) vec1.pb(n); else if (A[p] > A[v]) vec2.pb(n); } else { ll n = N - lca.subtree_size(p); if (A[p] < A[v]) vec1.pb(n); else if (A[p] > A[v]) vec2.pb(n); } } vvl vvec = {vec1, vec2}; fore(vec, vvec) { mint val = Sum(vec); val *= val; fore(e, vec) { val -= e * e; } ans += val / 2; } } print(ans); } /********* P A I R **********/ template<class T, class U> inline pair<T, U>& operator+=(pair<T, U>& a, const pair<T, U> b) {a.fi += b.fi; a.se += b.se; return a;} template<class T, class U> inline pair<T, U>& operator-=(pair<T, U>& a, const pair<T, U> b) {a.fi -= b.fi; a.se -= b.se; return a;} template<class T, class U> inline pair<T, U>& operator*=(pair<T, U>& a, const pair<T, U> b) {a.fi *= b.fi; a.se *= b.se; return a;} template<class T, class U> inline pair<T, U>& operator/=(pair<T, U>& a, const pair<T, U> b) {a.fi /= b.fi; a.se /= b.se; return a;} template<class T, class U> inline pair<T, U>& operator%=(pair<T, U>& a, const pair<T, U> b) {a.fi %= b.fi; a.se %= b.se; return a;} template<class T, class U, class V> inline pair<T, U>& operator+=(pair<T, U>& a, const V b) {a.fi += b; a.se += b; return a;} template<class T, class U, class V> inline pair<T, U>& operator-=(pair<T, U>& a, const V b) {a.fi -= b; a.se -= b; return a;} template<class T, class U, class V> inline pair<T, U>& operator*=(pair<T, U>& a, const V b) {a.fi *= b; a.se *= b; return a;} template<class T, class U, class V> inline pair<T, U>& operator/=(pair<T, U>& a, const V b) {a.fi /= b; a.se /= b; return a;} template<class T, class U, class V> inline pair<T, U>& operator%=(pair<T, U>& a, const V b) {a.fi %= b; a.se %= b; return a;} template<class T, class U, class V> inline pair<T, U> operator+(pair<T, U> a, const V& b) {a += b; return a;} template<class T, class U, class V> inline pair<T, U> operator-(pair<T, U> a, const V& b) {a -= b; return a;} template<class T, class U, class V> inline pair<T, U> operator*(pair<T, U> a, const V& b) {a *= b; return a;} template<class T, class U, class V> inline pair<T, U> operator/(pair<T, U> a, const V& b) {a /= b; return a;} template<class T, class U, class V> inline pair<T, U> operator%(pair<T, U> a, const V& b) {a %= b; return a;} template<class T, class U> inline pair<T, U> operator+(pair<T, U> a) {return a;} template<class T, class U> inline pair<T, U> operator-(pair<T, U> a) {return a * (-1);} template<class T, class U> istream& operator >>(istream& stream, pair<T, U>& a) {stream >> a.fi >> a.se; return stream;} template<class T, class U> ostream& operator <<(ostream &stream, const pair<T, U>& a) { stream << a.fi << " " << a.se; return stream; } /********* V E C T O R **********/ template<class T> inline vector<T>& operator+=(vector<T>& a, const vector<T> b) {assert(a.size() == b.size()); rep(i, 0, a.size()) a[i] += b[i]; return a;} template<class T> inline vector<T>& operator-=(vector<T>& a, const vector<T> b) {assert(a.size() == b.size()); rep(i, 0, a.size()) a[i] -= b[i]; return a;} template<class T> inline vector<T>& operator*=(vector<T>& a, const vector<T> b) {assert(a.size() == b.size()); rep(i, 0, a.size()) a[i] *= b[i]; return a;} template<class T> inline vector<T>& operator/=(vector<T>& a, const vector<T> b) {assert(a.size() == b.size()); rep(i, 0, a.size()) a[i] /= b[i]; return a;} template<class T> inline vector<T>& operator%=(vector<T>& a, const vector<T> b) {assert(a.size() == b.size()); rep(i, 0, a.size()) a[i] %= b[i]; return a;} template<class T, class U> inline vector<T>& operator+=(vector<T>& a, const U b) {fore(e, a) e += b; return a;} template<class T, class U> inline vector<T>& operator-=(vector<T>& a, const U b) {fore(e, a) e -= b; return a;} template<class T, class U> inline vector<T>& operator*=(vector<T>& a, const U b) {fore(e, a) e *= b; return a;} template<class T, class U> inline vector<T>& operator/=(vector<T>& a, const U b) {fore(e, a) e /= b; return a;} template<class T, class U> inline vector<T>& operator%=(vector<T>& a, const U b) {fore(e, a) e %= b; return a;} template<class T, class U> inline vector<T> operator+(vector<T> a, const U& b) {a += b; return a;} template<class T, class U> inline vector<T> operator-(vector<T> a, const U& b) {a -= b; return a;} template<class T, class U> inline vector<T> operator*(vector<T> a, const U& b) {a *= b; return a;} template<class T, class U> inline vector<T> operator/(vector<T> a, const U& b) {a /= b; return a;} template<class T, class U> inline vector<T> operator%(vector<T> a, const U& b) {a %= b; return a;} template<class T> inline vector<T> operator+(vector<T> a) {return a;} template<class T> inline vector<T> operator-(vector<T> a) {return a * (-1);} template<class T> istream& operator >>(istream& stream, vector<T>& a) {fore(e, a) stream >> e; return stream;} template<class T> ostream& operator <<(ostream& stream, const vector<T>& v) {if(v.size()){stream << v[0]; rep(i, 1, v.size()) cout << " " << v[i];} return stream;} template<class T> ostream& operator <<(ostream& stream, const vector<vector<T>>& vv) {if(vv.size()){stream << vv[0]; rep(i, 1, vv.size()) cout << '\n' << vv[i];} return stream;} /********* S E T **********/ template<class T> istream& operator >>(istream& stream, set<T>& st) {T e; stream >> e; st.insert(e); return stream;} template<class T> ostream& operator <<(ostream& stream, const set<T>& st) {if(st.size()){auto it=st.begin(); stream << *it++; for(; it!=st.end(); it++) cout << " " << *it;} return stream;} /********* M U L T I S E T **********/ template<class T> istream& operator >>(istream& stream, multiset<T>& st) {T e; stream >> e; st.insert(e); return stream;} template<class T> ostream& operator <<(ostream& stream, const multiset<T>& st) {if(st.size()){auto it=st.begin(); stream << *it++; for(; it!=st.end(); it++) cout << " " << *it;} return stream;} /********* P R I N T **********/ template<class T> inline void print(const T& e) {cout << e << '\n';} template<class H, class... T> inline void print(const H& h, const T&... t) {cout << h << ' '; print(t...);} template<class... T> inline void End(const T&... t) {print(t...); exit(0);} /********* D E B U G **********/ void dbg_(const long long& e) {if (e == INF) cout << "INF"; else if (e == -INF) cout << "-INF"; else cout << e;} template<class T> void dbg_(const T& e) {cout << e;} template<class T, class U> void dbg_(const pair<T, U>& p) {cout << '('; dbg_(p.first); cout << ' '; dbg_(p.second); cout << ')';} template<class T1, class T2, class T3> void dbg_(const trio<T1, T2, T3>& t) {cout << '('; dbg_(t.first); cout << ' '; dbg_(t.second); cout << ' '; dbg_(t.third); cout << ')';} template<class T> void debug_(const T& e) {dbg_(e); cout << '\n';} template<class T> void debug_(const vector<T>& v) {if (v.size()){auto it=v.begin(); dbg_(*it++); for(; it!=v.end(); ++it){cout << ' '; dbg_(*it);}} cout << '\n';} template<class T> void debug_(const vector<vector<T>>& vv) {cout << '\n'; ll cnt=0; for(auto&& v : vv){cout << cnt++ << ": "; debug_(v);}} template<class T, class U> void debug_(const map<T, U>& mp) {if (mp.size()) {auto it = mp.begin(); dbg_(*it++); for(; it != mp.end(); ++it) {cout << ' '; dbg_(*it);}} cout << '\n';} template<class T, class U> void debug_(const vector<map<T, U>>& vm){cout << '\n'; ll cnt=0; for(auto&& mp : vm){cout << cnt++ << ": "; debug_(mp);}} template<class T> void debug_(const set<T>& st) {if(st.size()){auto it=st.begin(); dbg_(*it++); for(; it!=st.end(); ++it) {cout << ' '; dbg_(*it);}}cout << '\n';} template<class T> void debug_(const multiset<T>& st) {if(st.size()) {auto it=st.begin(); dbg_(*it++); for(; it != st.end(); ++it) {cout << ' '; dbg_(*it);}} cout << '\n';} template<class T> void debug_(const vector<set<T>>& vs) {cout << '\n'; ll cnt=0; for(auto&& st : vs){cout << cnt++ << ": "; debug_(st);}} template<class T> void debug_(const vector<multiset<T>>& vs) {cout << '\n'; ll cnt=0;for(auto&& st : vs){cout << cnt++ << ": "; debug_(st);}} template<class H, class... T> void debug_(const H& h, const T&... t) {dbg_(h); cout << ", "; debug_(t...);} /********* T R I O **********/ template<class T1, class T2, class T3> istream& operator >>(istream &stream, trio<T1, T2, T3>& t) {return stream >> t.first >> t.second >> t.third;} template<class T1, class T2, class T3> ostream& operator <<(ostream &stream, const trio<T1, T2, T3>& t) {return stream << t.first << " " << t.second << " " << t.third;}