結果

問題 No.399 動的な領主
ユーザー char134217728
提出日時 2017-12-20 11:34:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 158 ms / 2,000 ms
コード長 2,352 bytes
コンパイル時間 2,334 ms
コンパイル使用メモリ 198,140 KB
最終ジャッジ日時 2025-01-05 06:03:19
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:65:1: warning: ISO C++ forbids declaration of ‘main’ with no type [-Wreturn-type]
   65 | main(){
      | ^~~~

ソースコード

diff # 

#include <bits/stdc++.h>
#define FOR(i,a,b) for (int i=(a);i<(b);i++)
#define FORR(i,a,b) for (int i=(a);i>=(b);i--)
#define pb push_back
#define pcnt __builtin_popcount
#define show(x) cout<<#x<<" = "<<x<<endl;
#define maxs(x,y) x = max(x,y)
#define mins(x,y) x = min(x,y)
#define fi first
#define se second
#define rng(a) a.begin(),a.end()
#define each(it,c) for(__typeof((c).begin()) it=(c).begin();it!=(c).end();it++)
#define sz(x) (int)(x).size()
#define mp make_pair

using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<pii> vpii;
typedef set<int> si;
typedef pair<ll,ll> pll;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<pll> vpll;
typedef set<ll> sl;
template<typename T>string join(vector<T>&v)
{stringstream s;FOR(i,0,sz(v))s<<' '<<v[i];return s.str().substr(1);}
ll gcd(ll a,ll b){if(a>b)swap(a,b);for(;a>0;b%=a,swap(a,b));return b;}
int modpow(ll a,ll n,int m){if(a==0)return a;ll p=1;for(;n>0;n/=2,a=a*a%m)if(n&1)p=p*a%m;return(int)p;}
void dout(double d){printf("%.12f\n",d);}

const int iinf = 1e9;
const ll linf = 1e18;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-10;
ll n, dp[100001][2], Q, u, v, a, b, c, dd, r, ans;
int db[100001][18], dpth[100001];
vi m[100001], m2[100001];
bool y[100001];
void setp(int i, int d){
  dpth[i] = d;
  each(itr, m[i]){
    if(y[*itr]) continue;
    y[*itr] = true;
    db[*itr][0] = i;
    m2[i].pb(*itr);
    setp(*itr, d+1);
  }
}
ll calc(int i){
  ll s = dp[i][0], ss;
  each(itr, m2[i]) s += calc(*itr);
  ss = s + dp[i][1];
  ans += ss*(ss+1)/2;
  return s;
}
int root(int i, int j){
  FORR(k, 17, 0)if(db[i][k] != db[j][k]) return root(db[i][k], db[j][k]);
  return db[i][0];
}

main(){
  cin.tie(0);
  ios::sync_with_stdio(false);
  cin >> n;
  FOR(i, 0, n-1){
    cin >> u >> v;
    m[u].pb(v);
    m[v].pb(u);
  }
  m[0].pb(1);
  db[0][0] = 0;
  y[0] = true;
  setp(0, 0);
  FOR(i, 1, 18)FOR(j, 0, n+1) db[j][i] = db[db[j][i-1]][i-1];
  cin >> Q;
  FOR(i, 0, Q){
    cin >> a >> b;
    if(dpth[a] > dpth[b]) swap(a, b);
    c = b;
    dd = dpth[b] - dpth[a];
    FOR(j, 0, 18)if((dd & (1<<j)) > 0) b = db[b][j];
    r = (a == b ? a : root(a, b));
    dp[a][0]++;
    dp[c][0]++;
    dp[r][0] -= 2;
    dp[r][1]++;
  }
  calc(0);
  cout << ans << endl;
  return 0;
}
0