結果

問題 No.399 動的な領主
ユーザー kzyKTkzyKT
提出日時 2016-07-15 23:00:51
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 110 ms / 2,000 ms
コード長 2,166 bytes
コンパイル時間 1,984 ms
コンパイル使用メモリ 146,828 KB
実行使用メモリ 25,684 KB
最終ジャッジ日時 2023-08-07 14:58:30
合計ジャッジ時間 3,629 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
13,780 KB
testcase_01 AC 5 ms
13,580 KB
testcase_02 AC 5 ms
13,656 KB
testcase_03 AC 5 ms
13,640 KB
testcase_04 AC 5 ms
13,612 KB
testcase_05 AC 12 ms
13,968 KB
testcase_06 AC 109 ms
17,872 KB
testcase_07 AC 110 ms
17,968 KB
testcase_08 AC 105 ms
17,940 KB
testcase_09 AC 106 ms
17,928 KB
testcase_10 AC 5 ms
13,624 KB
testcase_11 AC 11 ms
14,020 KB
testcase_12 AC 79 ms
18,356 KB
testcase_13 AC 76 ms
18,328 KB
testcase_14 AC 64 ms
25,684 KB
testcase_15 AC 75 ms
25,680 KB
testcase_16 AC 81 ms
21,316 KB
testcase_17 AC 100 ms
17,920 KB
testcase_18 AC 103 ms
17,912 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define all(c) (c).begin(),(c).end()
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;i--)
#define REP(i,m,n) for(int i=(int)(m);i<(int)(n);i++)
#define rep(i,n) REP(i,0,n)
#define iter(c) __typeof((c).begin())
#define tr(it,c) for(iter(c) it=(c).begin();it!=(c).end();it++)
#define mem(a) memset(a,0,sizeof(a))
#define pd(a) printf("%.10f\n",a)
#define pb(a) push_back(a)
#define in(a) insert(a)
#define pi M_PI
#define R cin>>
#define F first
#define S second
#define C class
#define ll long long
#define ln cout<<'\n'
template<C T>void pr(T a){cout<<a;ln;}
template<C T,C T2>void pr(T a,T2 b){cout<<a<<' '<<b;ln;}
template<C T,C T2,C T3>void pr(T a,T2 b,T3 c){cout<<a<<' '<<b<<' '<<c;ln;}
template<C T>void PR(T a,int n){rep(i,n){if(i)cout<<' ';cout<<a[i];}ln;}
bool check(int n,int m,int x,int y){return x>=0&&x<n&&y>=0&&y<m;}
const ll MAX=1000000007,MAXL=1LL<<60,dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
typedef pair<int,int> P;

ll d[100001],a[100001];
int dep[100010],p[100001];
int par[100010][19];
vector<int> g[100010];
void dfs(int v,int u,int d){
  dep[v]=d;
  p[v]=u;
  for(int i=0; i<g[v].size(); i++){
    int w=g[v][i];if(w==u) continue;
    par[w][0]=v;dfs(w,v,d+1);
  }
}
int lcp(int a,int b){
  if(dep[a]<dep[b]) swap(a,b);
  for(int i=0; i<19; i++){
    if(((dep[a]-dep[b])&(1<<i))) a=par[a][i];
  }
  if(a==b) return a;
  for(int i=18;i>=0;i--){
    if(par[a][i]!=par[b][i]) a=par[a][i],b=par[b][i];
  }
  return par[a][0];
}

ll dfs2(int x,int z) {
  ll sum=0;
  rep(i,g[x].size()) {
    int y=g[x][i];
    if(y==z) continue;
    sum+=dfs2(y,x);
    d[x]+=d[y];
  }
  return sum+(d[x]+1)*d[x]/2;
}
void Main() {
  ll n;
  cin >> n;
  rep(i,n-1) {
    int x,y;
    cin >> x >> y;
    x--,y--;
    g[x].pb(y);
    g[y].pb(x);
  }
  memset(par,0,sizeof(par));
  dfs(0,-1,0);
  for(int i=0; i<18; i++) for(int j=0; j<n; j++) par[j][i+1]=par[par[j][i]][i];
  int m;
  cin >> m;
  while(m--) {
    ll x,y;
    cin >> x >> y;
    x--,y--;
    d[x]++;
    d[y]++;
    int l=lcp(x,y);
    d[l]--;
    if(l>=0) d[p[l]]--;
  }
  pr(dfs2(0,-1));
}

int main() {
  ios::sync_with_stdio(0);cin.tie(0);
  Main();return 0;
}
0