Yandex.Algorithm 2014 Round 3 A問題 - Distance

Source

Yandex.Algorithm 2014
Yandex.Algorithm 2014 Round 3
問題文

問題概要

ノード数 $N$ の木が与えられる.
ノード間の距離が $2$ であるようなノードのペアがいくつ存在するかを求める問題.

解法

距離 $2$ なのでちょうど $1$ つのノードを中間点として経由する.
その中間点に着目すると,次数 $k$ のノードを中間点とする距離 $2$ のノードのペアの数は $C(k, 2) = k(k-1)/2$ になる.
各ノードの次数を数えて,そのノードを中間点とするものの数を足し上げれば良い.
答えは $32$ ビット整数で収まらないことに注意.

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

#define ll long long

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
void writer(ll x, char c){int i,sz=0,m=0;char buf[20];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}

int N, A[110000], B[110000], deg[110000];

int main(){
  int i, j, k;
  ll res = 0;

  reader(&N);
  rep(i,N-1) reader(A+i,B+i), A[i]--, B[i]--;

  rep(i,N) deg[i] = 0;
  rep(i,N-1) deg[A[i]]++, deg[B[i]]++;

  rep(i,N) res += (ll)deg[i] * (deg[i]-1) / 2;
  writer(res, '\n');

  return 0;
}

Current time: 2017年09月21日05時06分51秒
Last modified: 2014年08月06日01時15分57秒 (by laycrs)
Tags: Competitive_Programming Yandex_Algorithm Yandex_Algorithm_2014 Yandex_Algorithm_2014_Round3
トップページに戻る

Logged in as: unknown user (not login)

ログイン: