結果

問題 No.399 動的な領主
ユーザー koprickykopricky
提出日時 2017-10-10 23:19:33
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 158 ms / 2,000 ms
コード長 3,637 bytes
コンパイル時間 1,553 ms
コンパイル使用メモリ 166,284 KB
実行使用メモリ 24,152 KB
最終ジャッジ日時 2024-04-28 15:30:31
合計ジャッジ時間 4,999 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,812 KB
testcase_01 AC 3 ms
6,940 KB
testcase_02 AC 3 ms
6,944 KB
testcase_03 AC 3 ms
6,944 KB
testcase_04 AC 4 ms
6,940 KB
testcase_05 AC 13 ms
6,944 KB
testcase_06 AC 158 ms
16,428 KB
testcase_07 AC 155 ms
16,544 KB
testcase_08 AC 154 ms
16,476 KB
testcase_09 AC 152 ms
16,336 KB
testcase_10 AC 4 ms
6,944 KB
testcase_11 AC 13 ms
6,944 KB
testcase_12 AC 131 ms
16,892 KB
testcase_13 AC 131 ms
16,876 KB
testcase_14 AC 75 ms
24,152 KB
testcase_15 AC 97 ms
24,148 KB
testcase_16 AC 95 ms
20,292 KB
testcase_17 AC 156 ms
16,424 KB
testcase_18 AC 149 ms
16,512 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)n;++i)
#define each(a,b) for(auto (a): (b))
#define all(v) (v).begin(),(v).end()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define spair(p) cout<<#p<<": "<<p.fi<<" "<<p.se<<endl
#define svec(v) cout<<#v<<":";rep(kbrni,v.size())cout<<" "<<v[kbrni];cout<<endl
#define sset(s) cout<<#s<<":";each(kbrni,s)cout<<" "<<kbrni;cout<<endl
#define smap(m) cout<<#m<<":";each(kbrni,m)cout<<" {"<<kbrni.first<<":"<<kbrni.second<<"}";cout<<endl

using namespace std;

typedef pair<int,int>P;

const int MAX_N = 100005;

int imos[MAX_N],par[MAX_N];
int depth[MAX_N],id[MAX_N];
vector<int> ord;
vector<int> G[MAX_N];

template<typename V> class segtree {
private:
    int n,sz; vector<V> node; vector<int> node_id;
public:
    void resize(vector<V> v){
        sz = (int)v.size(); n = 1;
        while(n < sz) n *= 2;
        node.resize(2*n-1),node_id.resize(2*n-1);
        rep(i,sz){
            node[i+n-1] = v[i];
            node_id[i+n-1] = i;
        }
        for(int i=n-2; i>=0; i--){
            if(node[2*i+1] > node[2*i+2]){
                node[i] = node[2*i+2];
                node_id[i] = node_id[2*i+2];
            }else{
                node[i] = node[2*i+1];
                node_id[i] = node_id[2*i+1];
            }
        }
    }
    void update(int k,int a)
    {
    	k += n-1;
    	node[k] = a,node_id[k] = k-(n-1);
    	while(k>0){
    		k = (k-1)/2;
    		if(node[2*k+1] < node[2*k+2]){
                node[k] = node[2*k+1],node_id[k] = node_id[2*k+1];
            }else{
                node[k] = node[2*k+2],node_id[k] = node_id[2*k+2];
            }
    	}
    }
    pair<V,int> query(int a,int b,int k=0,int l=0,int r=-1)
    {
        if(r < 0) r = n;
    	if(r <= a || b <= l) return pair<V,int>(INT_MAX,-1);
    	if(a <= l && r <= b){
    		return pair<V,int>(node[k],node_id[k]);
    	}else{
    		pair<V,int> vl = query(a,b,2*k+1,l,(l+r)/2);
    		pair<V,int> vr = query(a,b,2*k+2,(l+r)/2,r);
    		return min(vl,vr);
    	}
    }
    void print()
    {
        rep(i,sz){
            pair<V,int> p;
            p = query(i,i+1);
            cout << "st[" << i << "]: " << p.fi << " " << p.se << endl;
        }
    }
};

void dfs(int u,int p,int k)
{
    id[u] = (int)ord.size();
    ord.pb(u);
    depth[u] = k;
    par[u] = p;
    rep(i,G[u].size()){
        if(G[u][i] != p){
            dfs(G[u][i],u,k+1);
            ord.pb(u);
        }
    }
}

segtree<int> st;

void lca_make(){
	dfs(0,-1,0);
    vector<int> stvec((int)ord.size());
	rep(i,ord.size()){
		stvec[i] = depth[ord[i]];
	}
    st.resize(stvec);
}

int lca(int u,int v){
	return ord[st.query(min(id[u],id[v]),max(id[u],id[v])+1).se];
}

void solve(int u,int p,ll& ans)
{
    rep(i,G[u].size()){
        if(G[u][i] != p){
            solve(G[u][i],u,ans);
            imos[u] += imos[G[u][i]];
        }
    }
    ans += (ll)imos[u]*(imos[u]+1)/2;
}

int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
    int n;
    cin >> n;
    rep(i,n-1){
        int a,b;
        cin >> a >> b;
        G[a-1].pb(b-1),G[b-1].pb(a-1);
    }
    lca_make();
    int q;
    cin >> q;
    rep(i,q){
        int a,b;
        cin >> a >> b;
        a--,b--;
        int res = lca(a,b);
        imos[a]++,imos[b]++;
        imos[res]--;
        if(par[res] >= 0){
            imos[par[res]]--;
        }
    }
    ll ans = 0;
    solve(0,-1,ans);
    cout << ans << endl;
    return 0;
}
0