#include "testlib.h" #pragma GCC optimize("Ofast") #pragma GCC optimize("unroll-loops") #pragma GCC target("avx2") #include using namespace std; struct __INIT__{__INIT__(){cin.tie(0);ios::sync_with_stdio(false);cout< vec; typedef vector> mat; typedef vector>> mat3; typedef vector svec; typedef vector> smat; templateusing V=vector; templateusing VV=V>; templateinline void output(T t){bool f=0;for(auto i:t){cout<<(f?" ":"")<inline void output2(T t){for(auto i:t)output(i);} templateinline void debug(T t){ #ifdef LOCAL bool f=0;for(auto i:t){cerr<<(f?" ":"")<inline void debug2(T t){for(auto i:t)debug(i);} #define loop(n) for(long long _=0;_<(long long)(n);++_) #define _overload4(_1,_2,_3,_4,name,...) name #define __rep(i,a) repi(i,0,a,1) #define _rep(i,a,b) repi(i,a,b,1) #define repi(i,a,b,c) for(long long i=(long long)(a);i<(long long)(b);i+=c) #define rep(...) _overload4(__VA_ARGS__,repi,_rep,__rep)(__VA_ARGS__) #define _overload3_rev(_1,_2,_3,name,...) name #define _rep_rev(i,a) repi_rev(i,0,a) #define repi_rev(i,a,b) for(long long i=(long long)(b)-1;i>=(long long)(a);--i) #define rrep(...) _overload3_rev(__VA_ARGS__,repi_rev,_rep_rev)(__VA_ARGS__) // #define rep(i,...) for(auto i:range(__VA_ARGS__)) // #define rrep(i,...) for(auto i:reversed(range(__VA_ARGS__))) // #define repi(i,a,b) for(lint i=lint(a);i<(lint)(b);++i) // #define rrepi(i,a,b) for(lint i=lint(b)-1;i>=lint(a);--i) // #define irep(i) for(lint i=0;;++i) // inline vector range(long long n){if(n<=0)return vector();vectorv(n);iota(v.begin(),v.end(),0LL);return v;} // inline vector range(long long a,long long b){if(b<=a)return vector();vectorv(b-a);iota(v.begin(),v.end(),a);return v;} // inline vector range(long long a,long long b,long long c){if((b-a+c-1)/c<=0)return vector();vectorv((b-a+c-1)/c);for(int i=0;i<(int)v.size();++i)v[i]=i?v[i-1]+c:a;return v;} // templateinline T reversed(T v){reverse(v.begin(),v.end());return v;} #define all(n) begin(n),end(n) templatebool chmin(T& s,const E& t){bool res=s>t;s=min(s,t);return res;} templatebool chmax(T& s,const E& t){bool res=s(s,t);return res;} // const array dx={1,0,-1,0,1,1,-1,-1}; // const array dy={0,1,0,-1,1,-1,1,-1}; // const string ds="RDLU"; #define SUM(v) accumulate(all(v),0LL) #if __cplusplus>=201703L templateauto make_vector(T x,int arg,Args ...args){if constexpr(sizeof...(args)==0)return vector(arg,x);else return vector(arg,make_vector(x,args...));} #endif #define extrep(v,...) for(auto v:__MAKE_MAT__({__VA_ARGS__})) #define bit(n,a) ((n>>a)&1) vector> __MAKE_MAT__(vector v){if(v.empty())return vector>(1,vector());long long n=v.back();v.pop_back();vector> ret;vector> tmp=__MAKE_MAT__(v);for(auto e:tmp)for(long long i=0;i>; templateusing graph_w=vector>>; #if __cplusplus>=201703L constexpr inline long long powll(long long a,long long b){long long res=1;while(b--)res*=a;return res;} #endif templatepair& operator+=(pair&s,const pair&t){s.first+=t.first;s.second+=t.second;return s;} templatepair& operator-=(pair&s,const pair&t){s.first-=t.first;s.second-=t.second;return s;} templatepair operator+(const pair&s,const pair&t){auto res=s;return res+=t;} templatepair operator-(const pair&s,const pair&t){auto res=s;return res-=t;} #define BEGIN_STACK_EXTEND(size) void * stack_extend_memory_ = malloc(size);void * stack_extend_origin_memory_;char * stack_extend_dummy_memory_ = (char*)alloca((1+(int)(((long long)stack_extend_memory_)&127))*16);*stack_extend_dummy_memory_ = 0;asm volatile("mov %%rsp, %%rbx\nmov %%rax, %%rsp":"=b"(stack_extend_origin_memory_):"a"((char*)stack_extend_memory_+(size)-1024)); #define END_STACK_EXTEND asm volatile("mov %%rax, %%rsp"::"a"(stack_extend_origin_memory_));free(stack_extend_memory_); #include #include #include /** * @brief グラフテンプレート */ using graph=std::vector>; template using graph_w=std::vector>>; graph load_graph(int n,int m){graph g(n);for(int i=0;i>s>>t;--s;--t;g[s].push_back(t);g[t].push_back(s);}return g;} graph load_digraph(int n,int m){graph g(n);for(int i=0;i>s>>t;--s;--t;g[s].push_back(t);}return g;} graph load_graph0(int n,int m){graph g(n);for(int i=0;i>s>>t;g[s].push_back(t);g[t].push_back(s);}return g;} graph load_digraph0(int n,int m){graph g(n);for(int i=0;i>s>>t;g[s].push_back(t);}return g;} graph load_tree(int n){graph g(n);for(int i=0;i>s>>t;--s;--t;g[s].push_back(t);g[t].push_back(s);}return g;} graph load_tree0(int n){graph g(n);for(int i=0;i>s>>t;g[s].push_back(t);g[t].push_back(s);}return g;} graph load_treep(int n){graph g(n);for(int i=0;i>t;g[i+1].push_back(t);g[t].push_back(i+1);}return g;} templategraph_w load_graph_weight(int n,int m){graph_w g(n);for(int i=0;i>s>>t>>u;--s;--t;g[s].emplace_back(t,u);g[t].emplace_back(s,u);}return g;} templategraph_w load_digraph_weight(int n,int m){graph_w g(n);for(int i=0;i>s>>t>>u;--s;--t;g[s].emplace_back(t,u);}return g;} templategraph_w load_graph0_weight(int n,int m){graph_w g(n);for(int i=0;i>s>>t>>u;g[s].emplace_back(t,u);g[t].emplace_back(s,u);}return g;} templategraph_w load_digraph0_weight(int n,int m){graph_w g(n);for(int i=0;i>s>>t>>u;g[s].emplace_back(t,u);}return g;} templategraph_w load_tree_weight(int n){graph_w g(n);for(int i=0;i>s>>t>>u;--s;--t;g[s].emplace_back(t,u);g[t].emplace_back(s,u);}return g;} templategraph_w load_tree0_weight(int n){graph_w g(n);for(int i=0;i>s>>t>>u;g[s].emplace_back(t,u);g[t].emplace_back(s,u);}return g;} templategraph_w load_treep_weight(int n){graph_w g(n);for(int i=0;i>t>>u;g[i+1].emplace_back(t,u);g[t].emplace_back(i+1,u);}return g;} #pragma once #include #include #include using namespace std; template class HLD_lazy{ int child_size(const graph& v,int n,int p){ int cnt=0; for(auto t:v[n]){ if(t!=p)cnt+=child_size(v,t,n); } return sz[n]=cnt+1; } void make(const graph& v,int root){ sz=new int[v.size()]; vertex=new int[v.size()]; par=new int[v.size()]; head=new int[v.size()]; heavy=new int[v.size()]; child_size(v,root,-1); stack>stk; stk.emplace(root,-1); int idx=0; par[root]=root; head[root]=root; while(!stk.empty()){ int n,p; tie(n,p)=stk.top(); stk.pop(); vertex[n]=idx++; heavy[n]=-1; int mx=0; for(auto t:v[n])if(t!=p&&mx* seg; public: HLD_lazy(const graph& v,int root=0){ make(v,root); seg=new atcoder::lazy_segtree(v.size()); } // HLD_lazy(const graph& v,const vector& a,int root=0,F f=F(),G g=G(),H h=H()):_f(f),_g(g),_h(h){ // vectortmp(v.size()); // make(v,root); // for(int i=0;i<(int)v.size();i++){ // tmp[vertex[i]]=a[i]; // } // seg=new lazy_segtree(tmp,f,g,h); // } // uからvに一つ進んだ点(vがuの子のときのみ) int next_u(int u,int v){ while(1){ if(par[v]==u){ if(head[v]!=par[v]){ return head[v]; }else{ return heavy[u]; } } else if(sz[u]>sz[head[v]])v=par[v]; else return heavy[u]; } } int dist(int l,int r){ int res=0; while(1){ if(head[l]==head[r])return sz[l]>sz[r]?l:r; else if(sz[head[l]]>sz[head[r]])r=par[r]; else l=par[l]; } } int lca(int l,int r){ while(1){ if(head[l]==head[r])return sz[l]>sz[r]?l:r; else if(sz[head[l]]>sz[head[r]])r=par[r]; else l=par[l]; } } inline void update_vertex(int u,E x){ update_seg(vertex[u],vertex[u],x); } inline T get_vertex(int u){ return get_seg(vertex[u],vertex[u]); } inline void update_subtree(int u,E x){ update_seg(vertex[u],vertex[u]+sz[u]-1,x); } inline T get_subtree(int u){ return get_seg(vertex[u],vertex[u]+sz[u]-1); } T get_seg(int u,int v){ if(u>v)swap(u,v); // cerr<prod(u,v+1); } void update_seg(int u,int v,E x){ if(u>v)swap(u,v); seg->apply(u,v+1,x); } void update_path(int u,int v,E x){ while(1){ if(head[u]==head[v]){ update_seg(vertex[u],vertex[v],x); break; } else if(sz[head[u]]>sz[head[v]]){ update_seg(vertex[v],vertex[head[v]],x); v=par[v]; } else{ update_seg(vertex[u],vertex[head[u]],x); u=par[u]; } } } T get_path(int u,int v){ T res; while(1){ if(head[u]==head[v]){ return F(res,get_seg(vertex[u],vertex[v])); } else if(sz[head[u]]>sz[head[v]]){ res=F(res,get_seg(vertex[v],vertex[head[v]])); v=par[v]; } else{ res=F(res,get_seg(vertex[u],vertex[head[u]])); u=par[u]; } } } }; using T=tuple; using E=tuple; T F(T s,T t){ auto [a,b,c,d]=s; auto [e,f,g,h]=t; return make_tuple(a+e,b+f,c+g,d+h); }; T TID(){ return make_tuple(0LL,0LL,0LL,0LL); }; T G(E t,T s){ auto [a,b,c,d]=s; auto [p,q,r]=t; d+=r; // cerr<>g(n); atcoder::dsu dsu(n); rep(i,n-1){ int s=inf.readInt(1,n,"s"); inf.readSpace(); int t=inf.readInt(1,n,"t"); inf.readEoln(); inf.ensure(s!=t); s--;t--; g[s].emplace_back(t); g[t].emplace_back(s); dsu.merge(s,t); } inf.ensure((int)dsu.groups().size()==1); HLD_lazyhld1(g),hld2(g); HLD_lazysz(g),sum(g); vectorpar(n); par[0]=-1; auto dfs=[&](auto dfs,lint now,lint p)->void { for(auto e:g[now]){ if(e==p)continue; par[e]=now; dfs(dfs,e,now); } }; dfs(dfs,0,-1); rep(i,n){ hld1.update_vertex(i,make_tuple(0LL,0LL,i-par[i])); hld2.update_vertex(i,make_tuple(0LL,0LL,i-par[i])); } sets; while(q--){ int u=inf.readInt(1,n,"u"); inf.readSpace(); int r=inf.readInt(1,n,"r"); inf.readSpace(); int v=inf.readInt(1,n,"v"); inf.readEoln(); u--;r--;v--; if(s.count(u)){ s.erase(u); hld1.update_path(0,u,make_tuple(-1LL,-1LL,0LL)); hld2.update_path(0,u,make_tuple(1LL,-1LL,0LL)); hld2.update_subtree(0,make_tuple(-1LL,0LL,0LL)); sz.update_vertex(u,-1); sum.update_vertex(u,-u-1); }else{ s.emplace(u); hld1.update_path(0,u,make_tuple(1LL,1LL,0LL)); hld2.update_path(0,u,make_tuple(-1LL,1LL,0LL)); hld2.update_subtree(0,make_tuple(1LL,0LL,0LL)); sz.update_vertex(u,1); sum.update_vertex(u,u+1); } // 1を根として見た時のvの部分木かつSに含まれる集合の頂点番号の総和 auto s1v=[&](auto v){ return sum.get_subtree(v); }; // 1を根として見た時のvの部分木かつSに含まれる集合の個数 auto c1v=[&](auto v){ return sz.get_subtree(v); }; // r=1の時の解 auto f1v=[&](auto v){ lint c=c1v(v); return (get<2>(hld1.get_subtree(v))+c*c*(par[v]+1)-s1v(v))/2; }; // r=vの時の解 auto fvv=[&](auto v){ return f1v(0)+get<2>(hld2.get_path(0,v)); }; if(r==v){ cout<