//#define _GLIBCXX_DEBUG //#pragma GCC target("avx2") //#pragma GCC optimize("O3") //#pragma GCC optimize("unroll-loops") #include using namespace std; #ifdef LOCAL #include #define OUT(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else #define OUT(...) (static_cast(0)) #endif #define endl '\n' #define lfs cout<= (ll)(n); i--) namespace template_tute{ using ll = long long; using ld = long double; const ll MOD1 = 1e9+7; const ll MOD9 = 998244353; const ll INF = 4.1e18; using P = pair; template using PQ = priority_queue; template using QP = priority_queue,greater>; templatebool chmin(T1 &a,T2 b){if(a>b){a=b;return true;}else return false;} templatebool chmax(T1 &a,T2 b){if(a({a,b,c})-min({a,b,c});} void ans1(bool x){if(x) cout<<"Yes"<void ans(bool x,T1 y,T2 z){if(x)cout<void anss(T1 x,T2 y,T3 z){ans(x!=y,x,z);}; templatevoid debug(const T &v,ll h,ll w,string sv=" "){for(ll i=0;ivoid debug(const T &v,ll n,string sv=" "){if(n!=0)cout<void debug(const vector&v){debug(v,v.size());} templatevoid debug(const vector>&v){for(auto &vv:v)debug(vv,vv.size());} templatevoid debug(stack st){while(!st.empty()){cout<void debug(queue st){while(!st.empty()){cout<void debug(deque st){while(!st.empty()){cout<void debug(PQ st){while(!st.empty()){cout<void debug(QP st){while(!st.empty()){cout<void debug(const set&v){for(auto z:v)cout<void debug(const multiset&v){for(auto z:v)cout<void debug(const array &a){for(auto z:a)cout<void debug(const map&v){for(auto z:v)cout<<"["<vector>vec(ll x, ll y, T w){vector>v(x,vector(y,w));return v;} vectordx={1,-1,0,0,1,1,-1,-1};vectordy={0,0,1,-1,1,-1,1,-1}; templatevector make_v(size_t a,T b){return vector(a,b);} templateauto make_v(size_t a,Ts... ts){return vector(a,make_v(ts...));} templateostream &operator<<(ostream &os, const pair&p){return os << "(" << p.first << "," << p.second << ")";} templateostream &operator<<(ostream &os, const vector &v){os<<"[";for(auto &z:v)os << z << ",";os<<"]"; return os;} templatevoid rearrange(vector&ord, vector&v){ auto tmp = v; for(int i=0;ivoid rearrange(vector&ord,Head&& head, Tail&&... tail){ rearrange(ord, head); rearrange(ord, tail...); } template vector ascend(const vector&v){ vectorord(v.size());iota(ord.begin(),ord.end(),0); sort(ord.begin(),ord.end(),[&](int i,int j){return make_pair(v[i],i) vector descend(const vector&v){ vectorord(v.size());iota(ord.begin(),ord.end(),0); sort(ord.begin(),ord.end(),[&](int i,int j){return make_pair(v[i],-i)>make_pair(v[j],-j);}); return ord; } template vector inv_perm(const vector&ord){ vectorinv(ord.size()); for(int i=0;i0);return n>=0?n/div:(n-div+1)/div;} ll CEIL(ll n,ll div){assert(div>0);return n>=0?(n+div-1)/div:n/div;} ll digitsum(ll n){ll ret=0;while(n){ret+=n%10;n/=10;}return ret;} ll modulo(ll n,ll d){return (n%d+d)%d;}; templateT min(const vector&v){return *min_element(v.begin(),v.end());} templateT max(const vector&v){return *max_element(v.begin(),v.end());} templateT acc(const vector&v){return accumulate(v.begin(),v.end(),T(0));}; templateT reverse(const T &v){return T(v.rbegin(),v.rend());}; //mt19937 mt(chrono::steady_clock::now().time_since_epoch().count()); int popcount(ll x){return __builtin_popcountll(x);}; int poplow(ll x){return __builtin_ctzll(x);}; int pophigh(ll x){return 63 - __builtin_clzll(x);}; templateT poll(queue &q){auto ret=q.front();q.pop();return ret;}; templateT poll(priority_queue &q){auto ret=q.top();q.pop();return ret;}; templateT poll(QP &q){auto ret=q.top();q.pop();return ret;}; templateT poll(stack &s){auto ret=s.top();s.pop();return ret;}; ll MULT(ll x,ll y){if(LLONG_MAX/x<=y)return LLONG_MAX;return x*y;} ll POW2(ll x, ll k){ll ret=1,mul=x;while(k){if(mul==LLONG_MAX)return LLONG_MAX;if(k&1)ret=MULT(ret,mul);mul=MULT(mul,mul);k>>=1;}return ret;} ll POW(ll x, ll k){ll ret=1;for(int i=0;isputn(d, len) != len) { dest.setstate(std::ios_base::badbit); } } return dest; } namespace converter{ int dict[500]; const string lower="abcdefghijklmnopqrstuvwxyz"; const string upper="ABCDEFGHIJKLMNOPQRSTUVWXYZ"; const string digit="0123456789"; const string digit1="123456789"; void regi_str(const string &t){ for(int i=0;ito_int(const string &s,const string &t){ regi_str(t); vectorret(s.size()); for(int i=0;ito_int(const string &s){ auto t=s; sort(t.begin(),t.end()); t.erase(unique(t.begin(),t.end()),t.end()); return to_int(s,t); } vector>to_int(const vector&s,const string &t){ regi_str(t); vector>ret(s.size(),vector(s[0].size())); for(int i=0;i>to_int(const vector&s){ string t; for(int i=0;i&s,const string &t){ regi_int(t); string ret; for(auto z:s)ret+=dict[z]; return ret; } vector to_str(const vector>&s,const string &t){ regi_int(t); vectorret(s.size()); for(int i=0;i struct edge { int to; T cost; int id; edge():to(-1),id(-1){}; edge(int to, T cost = 1, int id = -1):to(to), cost(cost), id(id){} operator int() const { return to; } }; template using Graph = vector>>; template Graphrevgraph(const Graph &g){ Graphret(g.size()); for(int i=0;i Graph readGraph(int n,int m,int indexed=1,bool directed=false,bool weighted=false){ Graph ret(n); for(int es = 0; es < m; es++){ int u,v; T w=1; cin>>u>>v;u-=indexed,v-=indexed; if(weighted)cin>>w; ret[u].emplace_back(v,w,es); if(!directed)ret[v].emplace_back(u,w,es); } return ret; } template Graph readParent(int n,int indexed=1,bool directed=true){ Graphret(n); for(int i=1;i>p; p-=indexed; ret[p].emplace_back(i); if(!directed)ret[i].emplace_back(p); } return ret; } } using namespace template_tute; template struct TreeDistanceQuery{ int n; const Graph &g; vector used;//既に重心として選ばれたか vector sub;//部分木のサイズ bool weighted; TreeDistanceQuery(const Graph &g, bool weighted = false):n(g.size()),g(g),weighted(weighted){ sub.assign(g.size(), 0); used.assign(g.size(), false); build(); } int search_centroid(int root){ return dfs_search(root, -1, dfs_sz(root, -1) / 2 ); } int dfs_sz(int k, int par){ sub[k] = 1; for(auto e:g[k]){ if(e.to == par || used[e.to])continue; sub[k] += dfs_sz(e.to, k); } return sub[k]; } int dfs_search(int k,int par, int mid){ for(auto e:g[k]){ if(e.to == par || used[e.to])continue; if(sub[e.to] > mid)return dfs_search(e.to, k, mid); } return k; } vector>>belong; //所属position, index 区間系の場合最初だけ無視 vector>dist; //dist[i][j]:=position i, index jの重心からの距離 vector>border; int ass_position = 0; void build(){ belong.resize(n); dist.resize(4*n); run(0); if(!weighted){ border.resize(4*n); for(int i=0;i<4*n;i++){ if(dist[i].empty())continue; border[i].resize(max(dist[i])+2); border[i].back() = dist[i].size(); for(int j=dist[i].size()-1;j>=0;j--){ border[i][dist[i][j]]=j; } } } } void run(int v){ int cent = search_centroid(v); used[cent] = true; using pi = pair; priority_queue, greater>pq; vector>>childs; for(auto e:g[cent]){ if(used[e.to])continue; run(e.to); queue>que; childs.emplace_back(); que.emplace(e.cost, e.to, cent); while(!que.empty()){ auto [d, k, par] = que.front(); childs.back().emplace_back(d, k); que.pop(); for(auto e:g[k]){ if(e.to == par || used[e.to])continue; que.emplace(d + e.cost, e.to, k); } } if(weighted)sort(childs.back().begin(), childs.back().end()); pq.emplace(childs.back().size(), childs.size() - 1); } while(pq.size() > 1){ auto [sz1, idx1] = pq.top(); pq.pop(); auto [sz2, idx2] = pq.top(); pq.pop(); dist[ass_position].reserve(childs[idx1].size()); for(auto [d, vi]: childs[idx1]){ belong[vi].emplace_back(ass_position, dist[ass_position].size()); dist[ass_position].push_back(d); } ass_position++; dist[ass_position].reserve(childs[idx2].size()); for(auto [d, vi]: childs[idx2]){ belong[vi].emplace_back(ass_position, dist[ass_position].size()); dist[ass_position].push_back(d); } ass_position++; vector>tmp; merge(childs[idx1].begin(),childs[idx1].end(),childs[idx2].begin(),childs[idx2].end(),back_inserter(tmp)); childs[idx2] = tmp; pq.emplace(tmp.size(), idx2); } if(pq.empty()){ pq.emplace(0, 0); childs.emplace_back(); } int idx = pq.top().second; dist[ass_position].reserve(childs[idx].size() + 1); belong[cent].insert(belong[cent].begin(), make_pair(ass_position, dist[ass_position].size())); dist[ass_position].push_back(0); for(auto [d, vi]: childs[idx]){ belong[vi].emplace_back(ass_position, dist[ass_position].size()); dist[ass_position].push_back(d); } ass_position++; belong[cent].emplace_back(ass_position,0); dist[ass_position].push_back(0); ass_position++; used[cent] = false; } //[l,r) vector>query(int v,T l,T r){ vector>ret; for(int i = 1; i < belong[v].size(); i++){ auto [pos, idx] = belong[v][i]; T vd = dist[pos][idx]; int li, ri; if(weighted){ li = lower_bound(dist[pos^1].begin(), dist[pos^1].end(), l - vd) - dist[pos^1].begin(); ri = lower_bound(dist[pos^1].begin(), dist[pos^1].end(), r - vd) - dist[pos^1].begin(); } else{ li = border[pos^1][clamp(l - vd, 0, T(border[pos^1].size()) - 1)]; ri = border[pos^1][clamp(r - vd, 0, T(border[pos^1].size()) - 1)]; } ret.emplace_back(pos^1, li, ri); } return ret; } }; template< typename sum_t, typename key_t > struct ReRooting { struct Edge { int to; key_t data; sum_t dp, ndp; }; using F = function< sum_t(sum_t, sum_t) >; using G = function< sum_t(sum_t, key_t) >; vector< vector< Edge > > g; vector< sum_t > subdp, dp; vector< sum_t > init; const sum_t ident; const F f; const G gg; ReRooting(int V, const F f, const G g, const sum_t &ident) : g(V), f(f), gg(g), ident(ident), subdp(V, ident), dp(V, ident), init(V, ident) {} void add_edge(int u, int v, const key_t &d) { g[u].emplace_back((Edge) {v, d, ident, ident}); g[v].emplace_back((Edge) {u, d, ident, ident}); } void add_edge_bi(int u, int v, const key_t &d, const key_t &e) { g[u].emplace_back((Edge) {v, d, ident, ident}); g[v].emplace_back((Edge) {u, e, ident, ident}); } void set(int idx, sum_t s){ subdp[idx] = s; init[idx] = s; } void dfs_sub(int idx, int par) { for(auto &e : g[idx]) { if(e.to == par) continue; dfs_sub(e.to, idx); subdp[idx] = f(subdp[idx], gg(subdp[e.to], e.data)); } } void dfs_all(int idx, int par, const sum_t &top) { sum_t buff{init[idx]}; for(int i = 0; i < (int) g[idx].size(); i++) { auto &e = g[idx][i]; e.ndp = buff; e.dp = gg(par == e.to ? top : subdp[e.to], e.data); buff = f(buff, e.dp); } dp[idx] = buff; buff = ident; for(int i = (int) g[idx].size() - 1; i >= 0; i--) { auto &e = g[idx][i]; if(e.to != par) dfs_all(e.to, idx, f(e.ndp, buff)); e.ndp = f(e.ndp, buff); buff = f(buff, e.dp); } } vector< sum_t > build() { dfs_sub(0, -1); dfs_all(0, -1, ident); return dp; } }; template struct HLD{ using D=long long; int n; vectorsz;//部分木サイズ vectordep; vectorpar; vectorhead; Graph &g;//隣接リスト vector>edges;//データ構造に乗せるedge列 vectorin,out;//[in,out)で部分木、[ina,inb]でa~bのパス(aが上) vectorcomp;//連結成分の根 //inは頂点のindexを表す。また、edge列の下側の頂点である HLD(Graph &g,int r=-1):g(g),n(g.size()){ hld_build(r); } void hld_build(int root = -1){ in.assign(n,-1);out.assign(n,-1);dep.assign(n,0); par.assign(n,-1);head.assign(n,-1);sz.assign(n,-1);comp.assign(n,-1); edges.assign(n,edge()); if(root == -1){//根がどこでも良い場合(森でも可) for(int i=0;i sz[g[k][0].to])swap(e, g[k][0]); } } int time = 0; void dfs_hld(int k){ in[k] = time++; for(auto e:g[k]){ if(e.to == par[k])continue; head[e.to] = (e.to == g[k][0].to ? head[k]: e.to); edges[time] = e; dfs_hld(e.to); } out[k] = time; } int lca(int p,int q){ while(1){ if(in[p] < in[q])swap(p,q); if(head[p] == head[q])return q; p = par[head[p]]; } } vector>query_path(int p,int q,bool isEdge){ int r=lca(p,q); vector>ret; for(int i=0;i<2;i++){ if(i == 1)swap(p,q); while(1){ if(isEdge&&p==r)break; if(head[p]==head[r]){ ret.emplace_back(in[r]+(isEdge?1:i),in[p]+1); break; } ret.emplace_back(in[head[p]],in[p]+1); p = par[head[p]]; } } return ret; } vector>>query_order_path(int p,int q,bool isEdge){ //配列0:反転 配列1:通常 vector>>ret(2); int r=lca(p,q); for(int i=0;i<2;i++){ if(i == 1)swap(p,q); while(1){ if(isEdge&&p==r)break; if(head[p]==head[r]){ ret[i].emplace_back(in[r]+(isEdge?1:i),in[p]+1); break; } ret[i].emplace_back(in[head[p]],in[p]+1); p = par[head[p]]; } } reverse(ret[1].begin(), ret[1].end()); return ret; } pairquery_subtree(int p,bool isEdge){ return make_pair(in[p]+isEdge,out[p]); } //uのv方向の子 子孫関係は前もって確認すること(in,outを見る) int child(int u,int v)const{ while(1){ if(head[u]==head[v]){ v=g[u][0].to; break; } v=head[v]; if(par[v]==u)break; v=par[v]; } return v; } //uをv方向に一つ進めた頂点 int move(int u,int v)const{ assert(u!=v); if(in[u]rev_in; int climb(int u,int k){ if(rev_in.empty()){ rev_in.resize(n); for(int i=0;i(dep[u]-k, 0); while(dep[u]>nd){ if(dep[head[u]]>nd){ u=par[head[u]]; } else{ u=rev_in[in[head[u]]+nd-dep[head[u]]]; } } return u; } int jump(int from,int to, int k){ int r = lca(from, to); int d1 = dep[from] - dep[r]; int d2 = dep[to] - dep[r]; if(d1 + d2 < k)return -1; else if(k <= d1)return climb(from, k); else return climb(to, d1 + d2 - k); } //from,to,c from->to辺をc進んだところに到着 重み0は進めるだけ進む //根に到達する場合は(根,根,0)を返す vectordep_in; tupleclimb_weighted(int u,T k){ if(rev_in.empty()){ rev_in.resize(n); for(int i=0;i(dep[u]-k, 0); while(1){ if(dep[head[u]]>=nd){ u=head[u]; if(dep[par[u]] Graphlca_tree(vector&v){ auto compare=[&](int x,int y){return in[x]ret(sz2); stackst; for(int i=0;i>n; auto g1=readGraph(n,n-1); auto g2=readGraph(n,n-1); TreeDistanceQuerytdq(g1); HLD hld(g2); vector>tdv(tdq.dist.size()); rep(i,0,tdq.dist.size())tdv[i].resize(tdq.dist[i].size(),-1); rep(i,0,n){ for(auto [pos,idx]:tdq.belong[i]){ tdv[pos][idx]=i; } } using ull=unsigned long long; ull ret=0; vector>td(n,MP(-1,-1)); for(int i=0;iver; rep(j,i,i+2){ rep(o,0,tdq.dist[j].size()){ if(tdv[j][o]!=-1){ ver.push_back(tdv[j][o]); td[tdv[j][o]]=MP(j-i,tdq.dist[j][o]); } } } auto tree=hld.lca_tree(ver); using pi=pair; auto f1=[&](pi x,pi y){return MP(x.fi+y.fi,x.se+y.se);}; auto f2=[&](pi x,ull y){return MP(x.fi+x.se*y,x.se);}; OUT("-----"); OUT(td); rep(tt,0,2){ ReRootingrr(tree.size(),f1,f2,MP(0,0)); rep(j,0,tree.size()){ for(auto e:tree[j]){ if(e.to>T; while(T--){ solve(); } return 0; }