結果

問題 No.399 動的な領主
ユーザー define
提出日時 2020-02-21 17:47:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 348 ms / 2,000 ms
コード長 3,646 bytes
コンパイル時間 2,028 ms
コンパイル使用メモリ 187,104 KB
実行使用メモリ 38,708 KB
最終ジャッジ日時 2024-10-08 19:53:58
合計ジャッジ時間 6,241 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In member function 'void HeavyLightDecomposition::make_path(long long int, long long int, long long int)':
main.cpp:94:42: warning: 'mxindex' may be used uninitialized [-Wmaybe-uninitialized]
   94 |                                 make_path(i,v,id);
      |                                 ~~~~~~~~~^~~~~~~~
main.cpp:85:21: note: 'mxindex' was declared here
   85 |                 int mxindex,mx=0;
      |                     ^~~~~~~

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define rep(i,n) for(int i=0;i<n;i++)
#define REP(i,n) for(int i=1;i<n;i++)
#define all(v) v.begin(),v.end()
#define inf (int)(3e18)
#define P pair<int,int>
#define mod (int)(1e9+7)

#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")

template<class T> inline void chmin(T &a, T b) {
	a = min(a, b);
}
template<class T> inline void chmax(T &a, T b) {
	a = max(a, b);
}

//RAQ-RSQ
struct Segtree{
	vector<int>dat,lazy;
	int size=1;
	Segtree(int x){
		while(size<x)size*=2;
		dat.resize(size*2-1);
		lazy.resize(size*2-1);
	}
	void eval(int k,int l,int r){
		dat[k]+=lazy[k]*(r-l);
		if(r-l>1){
			lazy[k*2+1]+=lazy[k];
			lazy[k*2+2]+=lazy[k];
		}
		lazy[k]=0;
	}
	void update(int a,int b,int x,int k=0,int l=0,int r=-1){
		if(r==-1)r=size;
		eval(k,l,r);
		if(r<=a||b<=l)return;
		if(a<=l&&r<=b){
			lazy[k]+=x;
			eval(k,l,r);
			return;
		}
		update(a,b,x,k*2+1,l,(l+r)/2);
		update(a,b,x,k*2+2,(l+r)/2,r);
		dat[k]=dat[k*2+1]+dat[k*2+2];
	}
	int query(int a,int b,int k=0,int l=0,int r=-1){
		if(r==-1)r=size;
		eval(k,l,r);
		if(r<=a||b<=l)return 0;
		if(a<=l&&r<=b){
			return dat[k];
		}
		int lv=query(a,b,k*2+1,l,(l+r)/2);
		int rv=query(a,b,k*2+2,(l+r)/2,r);
		return lv+rv;
	}
};

struct HeavyLightDecomposition{
	struct heavy_set{
		vector<int>element;
		int depth,parent;
		heavy_set(int v,int d,int par):element(1,v),depth(d),parent(par){}
	};
	vector<vector<int>>G;
	vector<heavy_set>S;
	vector<int>subtree_size,set_index,ele_index;

	int subtree(int v,int par){
		int &sz=subtree_size[v];
		if(sz>0)return sz;
		sz=1;
		for(int i:G[v])if(i!=par)sz+=subtree(i,v);
		return sz;
	}
	void make_path(int v,int par,int id){
		set_index[v]=id;
		ele_index[v]=S[id].element.size()-1;
		int mxindex,mx=0;
		for(int i:G[v])if(i!=par){
			if(mx<subtree(i,v)){
				mx=subtree(i,v);mxindex=i;
			}
		}
		for(int i:G[v])if(i!=par){
			if(mxindex==i){
				S[id].element.push_back(i);
				make_path(i,v,id);
			}else {
				S.emplace_back(i,S[id].depth+1,v);
				make_path(i,v,S.size()-1);
			}
		}
	}
	void init(int root){
		subtree_size.resize(G.size());
		set_index.resize(G.size());
		ele_index.resize(G.size());
		S.emplace_back(root,0,root);
		make_path(root,root,0);
		subtree_size.clear();
	}
	HeavyLightDecomposition(vector<vector<int>>&G,int root=0):G(G){
		init(root);
	}
	P get(int v){
		return {set_index[v],ele_index[v]};
	}
};

signed main() {
	int N;
	cin>>N;
	vector<vector<int>>G(N);
	rep(i,N-1){
		int u,v;cin>>u>>v;u--;v--;
		G[u].push_back(v);G[v].push_back(u);
	}
	HeavyLightDecomposition hld(G);
	vector<Segtree>segtree;
	rep(i,hld.S.size())segtree.emplace_back(hld.S[i].element.size());

	int ans=0;
	int Q;cin>>Q;
	while(Q--){
		int u,v;cin>>u>>v;u--;v--;

		if(hld.S[hld.get(v).first].depth<hld.S[hld.get(u).first].depth)swap(u,v);
		while(hld.S[hld.get(v).first].depth>hld.S[hld.get(u).first].depth){
			P tmp=hld.get(v);
			segtree[tmp.first].update(0,tmp.second+1,1);
			ans+=segtree[tmp.first].query(0,tmp.second+1);
			v=hld.S[tmp.first].parent;
		}

		while(hld.get(v).first!=hld.get(u).first){
			P tmp=hld.get(v),tmp2=hld.get(u);
			segtree[tmp.first].update(0,tmp.second+1,1);
			ans+=segtree[tmp.first].query(0,tmp.second+1);
			v=hld.S[tmp.first].parent;
			segtree[tmp2.first].update(0,tmp2.second+1,1);
			ans+=segtree[tmp2.first].query(0,tmp2.second+1);
			u=hld.S[tmp2.first].parent;
		}

		P tu=hld.get(u),tv=hld.get(v);
		int mn=min(tu.second,tv.second),mx=max(tu.second,tv.second);
		segtree[tu.first].update(mn,mx+1,1);
		ans+=segtree[tu.first].query(mn,mx+1);
	}
	cout<<ans<<endl;
}
0