트리 위의 표식

시간 제한10초메모리 제한1024 MB

요약
트리의 정점 K개를 독립적으로 균등하게 뽑을 때, 모든 표식이 거리 L 안에서 만날 확률을 998244353으로 나눈 나머지로 구한다.
난이도

어려움10점 중 8점

유형
트리, 동적 계획법, 조합론, 수학
정답자
아직 제출이 없습니다

문제

정점이 NN개인 트리가 주어진다. 간선은 N−1N-1개이며, ii번째 간선은 정점 A_iA\_i와 B_iB\_i를 길이 C_i(>0)C\_i(>0)인 도로로 연결한다.

트리 위의 지점은 정점뿐만 아니라 간선 위의 임의의 위치까지 포함한다. 트리가 사이클이 없는 구조이므로, 임의의 두 지점 x,yx, y 사이에는 항상 유일한 단순 경로가 존재한다. 두 지점 사이의 거리 d(x,y)d(x, y)를 다음과 같이 정의한다.

  • d(x,y)d(x, y) = xx에서 yy로 가는 단순 경로를 따라 이동할 때 거쳐야 하는 도로의 총 길이
  • 특히, x=yx = y이면 d(x,y)=0d(x, y) = 0

트리 위에 서로 구분되는 KK개의 표식이 놓인다. 각 표식은 NN개의 정점 중 하나를 균등하고 독립적으로 선택하여 위치한다. 즉, 하나의 표식이 임의의 정점 vv에 있을 확률은 1/N1/N이다.

트리 위의 표식들이 가깝다는 것은, 모든 표식이 현재 위치에서 최대 LL만큼 이동해서 트리 위의 어떠한 지점에서 만날 수 있음을 뜻한다. 이 때 지점은 정점 혹은, 간선 위의 어떤 위치일 수 있다.

표식들이 가까울 확률을 구하여라. 확률을 기약분수 PQ\tfrac{P}{Q}로 나타냈을 때, 다음 조건을 만족하는 정수 RR을 출력해야 한다. \[0 \le R < 998244353, \qquad R \cdot Q \equiv P \pmod{998244353}.\] 제한 조건을 만족하는 모든 입력에 대해 RR이 유일하게 존재함을 보일 수 있다.

입력

첫째 줄에 세 정수 NN, KK, LL이 공백을 사이에 두고 주어진다.

둘째 줄부터 N−1N-1개의 줄에 걸쳐 간선의 정보가 주어진다. 각 줄에는 세 정수 A_iA\_i, B_iB\_i, C_iC\_i가 주어진다. 이는 A_iA\_i번 정점과 B_iB\_i번 정점이 길이 C_iC\_i​의 간선으로 연결됨을 의미한다.

출력

문제의 조건을 만족하는 정수 \(R\)을 한 줄에 출력한다.

제한

  • 1≤N≤200,0001 \le N \le 200\\,000
  • 1≤K≤10181 \le K \le 10^{18}
  • 0≤L≤10180 \le L \le 10^{18}
  • 1≤A_i,B_i≤N1 \le A\_i, B\_i \le N (1≤i≤N−11 \le i \le N-1)
  • 1≤C_i≤1091 \le C\_i \le 10^9 (1≤i≤N−11 \le i \le N-1)
  • 입력으로 주어진 그래프는 트리다.
  • 입력으로 주어진 모든 수는 정수다.

예제2

  1. 예제 1

    입력
    3 4 0
    1 2 5
    2 3 5
    
    예상 출력
    480636170
    
  2. 예제 2

    입력
    3 2 3
    1 2 5
    2 3 5
    
    예상 출력
    110916040