만드는 중입니다

아직 만드는 중이라 바뀔 수 있습니다

아티팩트

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

요약
가중치가 있는 트리에서 k종류의 유물을 모두 지나는 최소 이동 거리를 구합니다. 시작점과 끝점은 이동 거리에서 제외됩니다.
난이도

어려움10점 중 8점

유형
트리, 동적 계획법, 비트 연산, DFS
정답자
아직 제출이 없습니다

문제

리크와 모티는 다른 차원에서 행성의 지도를 발견했다. 지도에는 길로 연결된 nn개의 발굴 지점이 표시되어 있다. ii번째 길은 지점 uiu_i와 viv_i를 잇고, 길이는 cic_i이다. 길은 정확히 n−1n-1개이며, 어느 지점에서든 다른 모든 지점으로 갈 수 있다. 즉, 이 구조는 트리이다. 외계인 정보원에 따르면 아티팩트는 총 kk종류이고, 지점 ii에는 종류 aia_i의 아티팩트가 있다. 리크는 대회에서 kk종류를 모두 모아야 한다. 이 차원에서는 고급 기술이 작동하지 않는다. 리크는 경로의 시작 지점과 끝 지점에 포털을 미리 만들 수 있지만, 나머지 경로는 걸어야 한다. 리크는 걷는 거리가 최소가 되도록 시작 지점, 끝 지점, 경로(단순 경로일 필요는 없다)를 고르되, 경로 위에서 모든 종류의 아티팩트를 모아야 한다.

입력

첫 줄에 공백으로 구분된 두 정수 nn과 kk (2⩽n⩽1052 \leqslant n \leqslant 10^5; 1⩽k⩽61 \leqslant k \leqslant 6)가 주어진다. 각각 지점의 수와 필요한 아티팩트 종류의 수이다. 다음 줄에는 nn개의 정수 aia_i (0⩽ai⩽k0 \leqslant a_i \leqslant k)가 주어지며, 각 지점에 있는 아티팩트의 종류를 나타낸다. ai=0a_i = 0이면 지점 ii에는 아티팩트가 없다. 이어지는 n−1n-1개의 줄에는 세 정수 uiu_i, viv_i, cic_i (1⩽ui,vi⩽n1 \leqslant u_i, v_i \leqslant n; 1⩽ci⩽1091 \leqslant c_i \leqslant 10^9)가 주어진다. 이는 지점 uiu_i와 viv_i 사이에 길이 cic_i인 길이 있음을 뜻한다. 입력 그래프는 트리임이 보장된다.

출력

kk종류의 아티팩트를 모두 모을 수 없으면 −1-1을 출력한다. 그렇지 않으면 모든 종류의 아티팩트를 모으기 위해 리크가 걸어야 하는 최소 거리를 출력한다.

힌트

첫 번째 예제에서 최적 경로 중 하나는 1→3→2→3→41 \to 3 \to 2 \to 3 \to 4이다. 두 번째 예제에는 종류 5의 아티팩트가 있는 지점이 없으므로 다섯 종류를 모두 모을 수 없다. 세 번째 예제에서 최적 경로 중 하나는 2→3→42 \to 3 \to 4이다.

예제3

  1. 예제 1

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

    입력
    5 5
    1 3 2 4 4
    1 3 1000
    2 3 5123
    3 4 3341
    4 5 7197
    
    예상 출력
    -1
    
  3. 예제 3

    입력
    4 3
    0 1 2 3
    1 2 10
    2 3 1
    3 4 50
    
    예상 출력
    51