#include using namespace std; using ll=long long; using ull=unsigned long long; using P=pair; templateusing minque=priority_queue,greater>; templatebool chmax(T &a,const T &b){return (abool chmin(T &a,const T &b){return (a>b?(a=b,true):false);} templateistream &operator>>(istream &is,pair&p){is>>p.first>>p.second;return is;} templateistream &operator>>(istream &is,tuple&a){is>>std::get<0>(a)>>std::get<1>(a)>>std::get<2>(a);return is;} templateistream &operator>>(istream &is,array&a){for(auto&i:a)is>>i;return is;} templateistream &operator>>(istream &is,vector &a){for(auto &i:a)is>>i;return is;} templatevoid operator++(pair&a,int n){a.first++,a.second++;} templatevoid operator--(pair&a,int n){a.first--,a.second--;} templatevoid operator++(vector&a,int n){for(auto &i:a)i++;} templatevoid operator--(vector&a,int n){for(auto &i:a)i--;} #define overload3(_1,_2,_3,name,...) name #define rep1(i,n) for(int i=0;i<(int)(n);i++) #define rep2(i,l,r) for(int i=(int)(l);i<(int)(r);i++) #define rep(...) overload3(__VA_ARGS__,rep2,rep1)(__VA_ARGS__) #define reps(i,l,r) rep2(i,l,r) #define all(x) x.begin(),x.end() #define pcnt(x) __builtin_popcountll(x) #define fin(x) return cout<<(x)<<'\n',static_cast(0) #define yn(x) cout<<((x)?"Yes\n":"No\n") #define uniq(x) sort(all(x)),x.erase(unique(all(x)),x.end()) template inline int fkey(vector&z,T key){return lower_bound(z.begin(),z.end(),key)-z.begin();} ll myceil(ll a,ll b){return (a+b-1)/b;} template auto vec(const int (&d)[n],const T &init=T()){ if constexpr (id(d,init)); else return init; } #ifdef LOCAL #include #define SWITCH(a,b) (a) #else #define debug(...) static_cast(0) #define debugg(...) static_cast(0) #define SWITCH(a,b) (b) templateostream &operator<<(ostream &os,const pair&p){os<>testcase; for(int i=0;i struct Edge{ int from,to; T weight; int index; Edge(int from_,int to_,T weight_=T(),int index_=-1):from(from_),to(to_),weight(weight_),index(index_){} Edge():from(-1),to(-1),weight(),index(-1){} friend std::ostream &operator<<(std::ostream &os,const Edge&e){ os<<'['; os<<"from:"< struct Tree{ protected: int n,r; std::vector>edge; std::vector>g; std::vectorptr; struct tree_range{ using iterator=typename std::vector>::iterator; iterator l,r; iterator begin()const{return l;} iterator end()const{return r;} int size()const{return r-l;} bool empty()const{return !size();} Edge &operator[](int i)const{return l[i];} }; struct const_tree_range{ using iterator=typename std::vector>::const_iterator; iterator l,r; iterator begin()const{return l;} iterator end()const{return r;} int size()const{return r-l;} bool empty()const{return !size();} const Edge &operator[](int i)const{return l[i];} }; public: explicit Tree(int n_):n(n_){ edge.reserve(n-1); } Tree():n(0){} Tree(int n_,const std::vector>&e,bool dir=false):n(n_),r(-1),edge(e){ if(!dir)build(); else{ std::vectorseen(n,false); ptr.resize(n+1); for(const Edge&i:edge)ptr[i.from]++,ptr[i.to]++,seen[e.to]=true; for(int i=1;i<=n;i++)ptr[i]+=ptr[i-1]; r=std::find(seen.begin(),seen.end(),false)-seen.begin(); assert(ptr[n]==n*2-2); g.resize(ptr[n]); for(const Edge&i:edge)g[--ptr[i.to]]=Edge(i.to,i.from,i.weight,i.index); for(const Edge&i:edge)g[--ptr[i.from]]=i; } } template void read(){ for(int i=0;i>u>>v; if constexpr(index)u--,v--; if constexpr(weighted)std::cin>>w; else w=1; edge.emplace_back(u,v,w,i); } build(); } template void readp(){ ptr.resize(n+1); for(int i=1;i>p; if constexpr(index)p--; edge.emplace_back(p,i,1,i-1); ptr[p]++; ptr[i]++; } for(int i=1;i<=n;i++)ptr[i]+=ptr[i-1]; g.resize(n*2-2); for(auto&&[u,v,w,i]:edge)g[--ptr[v]]=Edge(v,u,w,i); for(int i=0;ipar(n,-1); par[root]=-1; std::queueque; que.push(root); while(!que.empty()){ int x=que.front(); que.pop(); for(int i=ptr[x];i&e=g[i]; if(e.to!=par[x]){ par[e.to]=x; assert(e.indexbfs_order()const{ assert(is_directed()); std::vectorbfs(n); int p=0,q=0; bfs[q++]=root(); while(p&e:(*this)[x])bfs[q++]=e.to; } return bfs; } std::vectordfs_order()const{ assert(is_directed()); std::vectorres; res.reserve(n); std::vectorst(n); int p=0; st[p++]=root(); while(p){ int x=st[--p]; res.push_back(x); p+=(*this)[x].size(); for(const Edge&e:(*this)[x])st[--p]=e.to; p+=(*this)[x].size(); } return res; } std::vectorrbfs_order()const{ std::vectorbfs=bfs_order(); std::reverse(bfs.begin(),bfs.end()); return bfs; } void hld(){ assert(is_directed()); std::vectorsub(n); for(int x:rbfs_order()){ sub[x]=1; int mx=-1; for(Edge&e:(*this)[x]){ sub[x]+=sub[e.to]; if(mx,std::vector>in_out_order(){ assert(is_directed()); std::vectorin(n),out(n); int p=0; auto dfs=[&](auto self,int x)->void { in[x]=p++; for(const Edge&e:(*this)[x]){ self(self,e.to); } out[x]=p; }; dfs(dfs,root()); return std::make_pair(in,out); } std::pair>diameter()const{ assert(!is_directed()); static constexpr T inf=std::numeric_limits::max(); std::vectordst(n,inf); dst[0]=0; std::vectorque(n); int p=0,q=1; que[0]=0; while(p&e:(*this)[x])if(dst[e.to]==inf){ dst[e.to]=dst[x]+e.weight; que[q++]=e.to; } } int u=std::max_element(dst.begin(),dst.end())-dst.begin(); std::fill(dst.begin(),dst.end(),inf); dst[u]=0; p=0,q=1; que[0]=u; while(p&e:(*this)[x])if(dst[e.to]==inf){ dst[e.to]=dst[x]+e.weight; que[q++]=e.to; } } int v=std::max_element(dst.begin(),dst.end())-dst.begin(); T weight=dst[v]; std::vectorres; while(u!=v){ res.push_back(v); for(const Edge&e:(*this)[v])if(dst[e.to]& get_edge(int i)const{return edge[i];} inline int parent(int i)const{return i==r?-1:g[ptr[i+1]-1].to;} inline int root()const{return r;} typename std::vector>::iterator begin(){return edge.begin();} typename std::vector>::iterator end(){return edge.end();} typename std::vector>::const_iterator begin()const{return edge.begin();} typename std::vector>::const_iterator end()const{return edge.end();} }; struct StaticTopTree{ private: template void build(Treet){ int n=t.size(); left.reserve(n*2-1),right.reserve(n*2-1),par.reserve(n*2-1),A.reserve(n*2-1),B.reserve(n*2-1); left.resize(n,-1),right.resize(n,-1),par.resize(n,-1),A.resize(n),B.resize(n); for(int i=0;istd::pair { std::vector>vs{{0,x}}; while(t[x].size()>=1){ std::priority_queue,std::vector>,std::greater>>que; int heavy=t[x][0].to; que.emplace(0,heavy); for(int i=1;i=2){ auto [d1,v1]=que.top();que.pop(); auto [d2,v2]=que.top();que.pop(); if(B[v2]==heavy)std::swap(d1,d2),std::swap(v1,v2); int nv=left.size(); left.push_back(v1),right.push_back(v2),par.push_back(-1),A.push_back(x),B.push_back(B[v1]); par[v1]=par[v2]=nv; que.emplace(std::max(d1,d2)+1,nv); } vs.push_back(que.top()); while(true){ int sz=vs.size(); if(sz>=3&&(vs[sz-3].first==vs[sz-2].first||vs[sz-3].first<=vs[sz-1].first)){ int nv=left.size(); left.push_back(vs[sz-3].second),right.push_back(vs[sz-2].second),par.push_back(-1),A.push_back(A[vs[sz-3].second]),B.push_back(B[vs[sz-2].second]); par[vs[sz-3].second]=par[vs[sz-2].second]=nv; vs[sz-3].first=std::max(vs[sz-3].first,vs[sz-2].first)+1; vs[sz-3].second=nv; vs[sz-2]=vs[sz-1]; vs.pop_back(); } else if(sz>=2&&vs[sz-2].first<=vs[sz-1].first){ int nv=left.size(); left.push_back(vs[sz-2].second),right.push_back(vs[sz-1].second),par.push_back(-1),A.push_back(A[vs[sz-2].second]),B.push_back(B[vs[sz-1].second]); par[vs[sz-2].second]=par[vs[sz-1].second]=nv; vs[sz-2].first=std::max(vs[sz-2].first,vs[sz-1].first)+1; vs[sz-2].second=nv; vs.pop_back(); } else break; } x=heavy; } while((int)vs.size()>=2){ int sz=vs.size(); int nv=left.size(); left.push_back(vs[sz-2].second),right.push_back(vs[sz-1].second),par.push_back(-1),A.push_back(A[vs[sz-2].second]),B.push_back(B[vs[sz-1].second]); par[vs[sz-2].second]=par[vs[sz-1].second]=nv; vs[sz-2].first=std::max(vs[sz-2].first,vs[sz-1].first)+1; vs[sz-2].second=nv; vs.pop_back(); } return vs[0]; }; dfs(dfs,t.root()); } public: std::vectorleft,right,par,A,B; StaticTopTree(){} template explicit StaticTopTree(Treet,int){ build(std::move(t)); } template explicit StaticTopTree(Treet){ assert(t.is_directed()); t.hld(); build(std::move(t)); } }; struct ContourQuery{ private: std::vector>ptr; std::vector>dst; std::vectorstt_par,dep,lr; int vsc; public: ContourQuery(){} template ContourQuery(Treet){ assert(!t.is_directed()); int n=t.size(); t.remove_parent(); StaticTopTree stt(t); dep.resize(n*2-1); for(int i=n*2-1;i-->n;){ dep[stt.left[i]]=dep[stt.right[i]]=dep[i]+1; } dst.resize(*std::max_element(dep.begin(),dep.begin()+n),std::vector(n,-1)); ptr.resize(n*2-1); std::vectorque(n+dst.size()); int p=0,q=0; vsc=n; auto dfs=[&](auto self,int v)->std::vector { if(v{v}; } const int d=dep[v]; int lv=stt.left[v],rv=stt.right[v]; std::vectorlch=self(self,lv); p=q=0; if(stt.A[lv]==stt.A[rv]){ for(int x:lch){ dst[d][x]=1; que[q++]=x; } while(p&e:t[x]){ dst[d][e.to]=dst[d][x]+1; que[q++]=e.to; } } int len=dst[d][que[q-1]]+1; ptr[lv]=std::make_pair(vsc,len); vsc+=len; } else{ dst[d][stt.B[lv]]=0; bool boundaryA=false; if(t.parent(stt.B[lv])==stt.A[lv])que[q++]=~stt.B[lv],boundaryA=true; else{ que[q++]=t.parent(stt.B[lv]); dst[d][stt.B[lv]]=0,dst[d][que[0]]=1; } while(p&e:t[x])if(dst[d][e.to]==-1){ dst[d][e.to]=dst[d][x]+1; que[q++]=e.to; } int par=t.parent(x); if(par==stt.A[lv]){ if(!boundaryA){ que[q++]=~x; boundaryA=true; } } else if(dst[d][par]==-1){ dst[d][par]=dst[d][x]+1; que[q++]=par; } } } int len; if(que[q-1]<0)len=q>=2?dst[d][que[q-2]]+1:1; else len=dst[d][que[q-1]]+1; ptr[lv]=std::make_pair(vsc,len); vsc+=len; } std::vectorrch=self(self,rv); p=q=0; for(int x:rch){ dst[d][x]=1; que[q++]=x; } while(p&e:t[x]){ dst[d][e.to]=dst[d][x]+1; que[q++]=e.to; } } ptr[rv]=std::make_pair(vsc,dst[d][que[q-1]]+1); vsc+=dst[d][que[q-1]]+1; if(stt.A[lv]==stt.A[rv]){ if(std::ssize(lch)get_vs(int v)const{ std::vectorres; res.reserve(dep[v]+1); res.push_back(v); int d=dep[v]-1; int x=v; while(d>=0){ res.push_back(ptr[v].first+dst[d][x]); v=stt_par[v]; d--; } return res; } std::vector>get_range(int v,int l,int r)const{ std::vector>res; if(l>=r)return res; if(l<=0&&1<=r){ res.reserve(dep[v]+1); res.emplace_back(v,v+1); } else res.reserve(dep[v]); int x=v; while(true){ int par=stt_par[v]; if(par==-1)break; int another=lr[par-std::ssize(lr)-1]^v; int d=dep[par]; int rtov=dst[d][x]; int nl=std::clamp(l-rtov,0,ptr[another].second); int nr=std::clamp(r-rtov,0,ptr[another].second); if(nl!=nr)res.emplace_back(ptr[another].first+nl,ptr[another].first+nr); v=par; } return res; } inline int size()const{return vsc;} }; #include #include template constexpr std::enable_if_t::digits<=32,int>msb(T n){return n==0?-1:31-__builtin_clz(n);} template constexpr std::enable_if_t<(std::numeric_limits::digits>32),int>msb(T n){return n==0?-1:63-__builtin_clzll(n);} template constexpr std::enable_if_t::digits<=32,int>lsb(T n){return n==0?-1:__builtin_ctz(n);} template constexpr std::enable_if_t<(std::numeric_limits::digits>32),int>lsb(T n){return n==0?-1:__builtin_ctzll(n);} template constexpr std::enable_if_t,T>floor_pow2(T n){return n==0?0:T(1)< constexpr std::enable_if_t,T>ceil_pow2(T n){return n<=1?1:T(1)<<(msb(n-1)+1);} template constexpr T safe_div(T a,T b){return a/b-(a%b&&(a^b)<0);} template constexpr T safe_ceil(T a,T b){return a/b+(a%b&&(a^b)>0);} template struct SparseTable{ using S=typename M::S; private: std::vectordat,prefix,suffix; std::vector>sp; public: SparseTable(){} SparseTable(std::vectora):dat(a){ int n=a.size(); n=(n+(1<d2(n>>L,M::e()); for(int i=0;i<(int)d2.size();i++){ for(int j=0;j<(1<>L);i++){ for(int j=1;j<(1<>L)-1;i>=0;i--){ for(int j=(1<=1;j--)suffix[(i<=j-w;k--)sp[i][k]=M::op(d2[k],sp[i][k+1]); int r=std::min(d2.size(),j+w); for(int k=j+1;k>L,rid=r>>L; if(lid==rid){ S ret=M::e(); for(int i=l;i<=r;i++)ret=M::op(ret,dat[i]); return ret; } else{ lid++; rid--; S mid=M::e(); if(lid==rid)mid=sp[0][lid]; else if(lid; static S op(const S&x,const S&y){return x.first::max(),-1);} }; SparseTablesp; std::vectoridx; public: template LowestCommonAncestor(const Tree&t):idx(t.size()){ assert(t.is_directed()); int r=t.root(); int ord=0; std::vectordep(t.size()); dep[r]=0; std::vector>init(t.size()*2-1); auto dfs=[&](auto&&self,int x)->void { idx[x]=ord; init[ord++]=std::make_pair(dep[x],x); for(const auto&e:t[x]){ dep[e.to]=dep[e.from]+1; self(self,e.to); init[ord++]=std::make_pair(dep[x],x); } }; dfs(dfs,r); sp=SparseTable(init); } LowestCommonAncestor(){} int query(int u,int v)const{ if(u==v)return u; if(idx[u]>idx[v])std::swap(u,v); return sp.prod(idx[u],idx[v]+1).second; } }; template struct DualSegmentTree{ using S=typename M::S; using F=typename M::F; private: int n,z,log2n; std::vectordat; std::vectorlazy; inline void push(int i){ lazy[i*2]=M::composition(lazy[i],lazy[i*2]); lazy[i*2+1]=M::composition(lazy[i],lazy[i*2+1]); lazy[i]=M::id(); } void path_push(int i){ int l=lsb(i); for(int j=log2n;j>l;j--)push(i>>j); } public: DualSegmentTree():n(0),log2n(0),z(0){} DualSegmentTree(int n_):n(n_),z(ceil_pow2(n_)){ log2n=msb(z); dat.resize(n,M::e()); lazy.resize(z*2,M::id()); } DualSegmentTree(const std::vector&init):n(init.size()),z(ceil_pow2((int)init.size())),dat(init){ log2n=msb(z); lazy.resize(z*2,M::id()); } void apply(int l,int r,const F&f){ l+=z,r+=z; path_push(l),path_push(r); while(l>=1,r>>=1; } } S get(int i){ i+=z; for(int j=log2n;j>=1;j--)push(i>>j); return M::mapping(lazy[i],dat[i-z],1); } void set(int i,const S&x){ i+=z; for(int j=log2n;j>=1;j--)push(i>>j); lazy[i]=M::id(); dat[i-z]=x; } std::vectorget_all(){ for(int i=1;ires(n); for(int i=0;i struct RangeLinearAddRangeSum{ using S=std::pair; using F=std::pair; static inline S op(const S&x,const S&y){return x.first==-1?y:std::make_pair(x.first,x.second+y.second);} static inline S e(){return std::make_pair(-1,T());} static inline S mapping(const F&f,const S&x,long long k){return std::make_pair(x.first,x.second+f.first*(x.first*k+(((k-1)|1)*((k&-2)>>1)))+f.second*k);} static inline F composition(const F&f,const F&g){return std::make_pair(f.first+g.first,f.second+g.second);} static inline F id(){return std::make_pair(T(),T());} }; /* 偶数:k-1 奇数:k */ void SOLVE(){ int n; cin>>n; Tree t(n); t.read(); ContourQuery cq(t); t.remove_parent(); LowestCommonAncestor lca(t); vectordst(n); for(int x:t.bfs_order()){ for(auto e:t[x])dst[e.to]=dst[e.from]+1; } auto dist=[&](int u,int v)->int { int l=lca.query(u,v); return dst[u]+dst[v]-dst[l]*2; }; ll base=0; rep(i,n-1)base+=dist(i,i+1); vectorans(n,base); using M=RangeLinearAddRangeSum; vectorinit(cq.size()); rep(i,cq.size())init[i]={i,0}; DualSegmentTree>seg(init); vectorleader(cq.size(),-1); rep(i,n){ for(int v:cq.get_vs(i))leader[v]=i; } debug(leader); for(int i=n-2;i>=0;i--){ for(int v:cq.get_vs(i)){ ans[i]+=seg.get(v).second; } ll d=dist(i,i+1); for(auto [l,r]:cq.get_range(i+1,0,d)){ int v=leader[l]; if(v==-1){ l++; v=leader[l]; } assert(v!=-1); int blank=dist(v,i+1); seg.apply(l,r,{0,-d}); seg.apply(l,r,{1,-l+1+blank}); } // rep(k,i)if(dist(k,i+1)=0;i--){ // rep(k,i){ // ll d=1+dist(k,i+1); // if(dist(i,i+1)>d)ans[k]+=d-dist(i,i+1); // } // } rep(i,n)cout<