RoAlgo PreOJI 2024 XI-XII | Warb

This was the problem page during the contest. Access the current page here.
Time limit: 3s
Memory limit: 512MB
Input: warb.in
Output: warb.out

A fost odată ca-n povești,
A fost ca niciodată,
Din rude mari împărătești,
Un prea frumos arbore cu nn noduri.

Cerință

Gigel, un mare informatician (supranumit luceafărul informaticii românești), vrea să coloreze preafrumosul arbore folosind mm culori, codificate prin numerele naturale de la 11 la mm.

După ce a asociat fiecărei culori cc o pondere wcw_c și fiecărui nod uu o valoare requreq_u, Gigel a zis că o colorare a nodurilor arborelui este bună dacă fiecare nod uu respectă una dintre următoarele condiții:

  1. Nodul uu are culoarea 11; sau
  2. Dacă nodul uu are culoarea c>1c>1, atunci el are exact requreq_u vecini cu culoarea c1c-1.

Tot Gigel definește valoarea unei astfel de colorări ca fiind wc1+wc2++wcnw_{c_1}+w_{c_2}+\ldots+w_{c_n}, unde cuc_u reprezintă culoarea nodului uu.

Ajutați-l pe Gigel să afle cea mai mare valoare a unei colorări bune!

Date de intrare

Pe prima linie din fişierul de intrare warb.in se vor găsi două numere naturale nn și mm — numărul de noduri din arbore, respectiv numărul de culori.
Pe a doua linie se vor găsi mm numere naturale w1,w2,,wmw_1,w_2,\ldots,w_m — ponderile celor mm culori.
Pe a treia linie se vor găsi nn numere naturale req1,req2,,reqnreq_1,req_2,\ldots,req_n — valorile asociate celor nn noduri.
Pe următoarele n1n-1 linii se vor găsi câte 2 numere naturale uu si vv, semnificând faptul că există o muchie în arbore între nodurile uu și vv.

Date de iesire

Pe prima linie din fişierul de ieşire warb.out se va afişa un singur număr natural, cea mai mare valoarevaloare a unei colorari bunebune.

Restricții și precizări

  • 1n1051 \leq n \leq 10^5
  • 1m1001 \leq m \leq 100
  • 1wi1041 \leq w_i \leq 10^4
  • 0reqi0 \leq req_i \leq gradul nodului ii.
  • Se garantează că graful din fișierul de intrare este un arbore (un graf neorientat conex aciclic).
  • Pentru a obține punctele pentru un anumit subtask, cel puțin o sursă trimisă va trebui să treacă toate testele din acel subtask.
    # Punctaj Restricții
    0 0 Exemple
    1 4 n=1n=1
    2 4 m=1m=1
    3 22 n,m7n,m \le 7
    4 10 reqi=0req_i=0
    5 8 m10m \le 10, arborele este un lanț (gradele tuturor nodurilor sunt mai mici sau egale cu 22)
    6 7 arborele este un lanț
    7 10 arborele este o stea (există un nod cu gradul n1n-1)
    8 19 m,gradele tuturor nodurilor10m, \text{gradele tuturor nodurilor} \leq 10
    9 6 reqi>0req_i > 0
    10 10 Fără restricții suplimentare

Exemplul 1

warb.in

3 2
1 2
1 1 1
1 2
1 3

warb.out

5

Explicație

Pentru primul exemplu, o colorare optimă este c=[1,2,2]c=[1,2,2]:

  • Nodul 11 are culoarea 11.
  • Nodul 22 are culoarea 22 și exact un vecin cu culoarea c21=1c_2-1=1.
  • Nodul 33 are culoarea 22 și exact un vecin cu culoarea c31=1c_3-1=1.
  • Valoarea acestei colorări este egală cu wc1+wc2+wc3=w1+w2+w2=1+2+2=5w_{c_1}+w_{c_2}+w_{c_3}=w_1+w_2+w_2=1+2+2=5.

Exemplul 2

warb.in

5 4
1 2 3 4
2 3 1 1 1
1 2
1 3
2 4
2 5

warb.out

8

Explicație

Pentru al doilea exemplu, o colorare optimă este c=[2,1,1,2,2]c=[2,1,1,2,2]:

  • Nodul 11 are culoarea 22 și exact doi vecini cu culoarea c11=1c_1-1=1.
  • Nodul 22 are culoarea 11.
  • Nodul 33 are culoarea 11.
  • Nodul 44 are culoarea 22 și exact un vecin cu culoarea c41=1c_4-1=1.
  • Nodul 55 are culoarea 22 și exact un vecin cu culoarea c51=1c_5-1=1.
  • Valoarea acestei colorări este egală cu wc1+wc2+wc3+wc4+wc5=2+1+1+2+2=8w_{c_1}+w_{c_2}+w_{c_3}+w_{c_4}+w_{c_5}=2+1+1+2+2=8.

Exemplul 3

warb.in

7 4
1 9 8 7
2 2 0 0 1 1 1
1 2
1 3
2 4
2 5
3 6
3 7

warb.out

46

Explicație

Pentru al treilea exemplu, o colorare optimă este c=[2,1,1,3,2,2,2]c=[2,1,1,3,2,2,2].

Exemplul 4

warb.in

6 5
7 9 50 30 80
0 1 0 0 1 0
1 2
2 3
3 4
3 5
5 6

warb.out

380

Explicație

Pentru al patrulea exemplu, o colorare optimă este c=[4,5,5,5,5,4]c=[4,5,5,5,5,4].

Exemplul 5

warb.in

7 7
458 2960 8526 2565 6679 9621 8814
1 0 0 0 2 1 1
3 4
1 2
1 4
5 7
4 5
4 6

warb.out

49102

Explicație

Pentru al cincilea exemplu, o colorare optimă este c=[7,7,6,6,1,7,2]c=[7,7,6,6,1,7,2].

Exemplul 6

warb.in

7 7
3564 9408 8285 6312 7323 3223 4441
0 0 0 1 1 1 2
3 2
5 3
4 1
7 3
4 7
6 7

warb.out

54831

Explicație

Pentru al șaselea exemplu, o colorare optimă este c=[2,2,4,2,5,2,1]c=[2,2,4,2,5,2,1].

Contest info

Official contest

Start time: 1709532000000

Total duration: 168h0m0s

Status: Ended

Individual duration: 4h0m0s

Log in or sign up to be able to send submissions!