R01277
지름 쌍 세기
레이팅
의견: 0
설명
가중치가 있는 트리가 주어진다. 트리의 지름은 두 정점 사이 경로 길이의 최댓값이다.
거리(경로의 길이)가 정확히 지름과 같은 정점 쌍 \(\{u, v\}\)의 개수를 구하여라.
제약
\(2 \le n \le 200\,000\), \(1 \le w \le 10^9\). 답은 64비트 정수 범위 안에 들어간다.
입력 형식
첫째 줄에 정점의 개수 \(n\)이 주어진다. 다음 \(n-1\)개의 줄에 간선의 정보 \(u\ v\ w\)가 주어진다 — 정점 \(u\)와 \(v\)를 잇는 가중치 \(w\)의 간선이다.
출력 형식
지름을 이루는 정점 쌍의 개수를 출력한다.
예제 1
입력
5
1 2 3
2 3 4
2 4 2
1 5 6
출력
1
설명
지름은 \(13\)이고 이를 만족하는 쌍은 \((3, 5)\) 하나뿐이다.
예제 2
입력
4
1 2 1
1 3 1
1 4 1
출력
3
설명
지름은 \(2\)이고 \((2,3)\), \((2,4)\), \((3,4)\) 세 쌍이 지름을 이룬다.
힌트
막혔나요? 코인으로 단계별 힌트를 잠금 해제하세요 — 첫 힌트는 가벼운 방향 제시, 뒤로 갈수록 더 많이 알려 줍니다. 문제를 풀면 모든 힌트가 무료로 공개됩니다.
문제 정보
riseoj 작성
출처 Original
태그
평가 및 의견
지름 쌍 세기
개요
Log in to rate problems.
개별 의견
아직 의견이 없습니다. 자격이 된다면 위 양식에서 가장 먼저 평가해 보세요.
풀이 제출
지름 쌍 세기
게스트로 둘러보고 있습니다. 로그인하면 풀이를 제출하고 진행 상황을 확인할 수 있습니다.
로그인하고 제출하기