Laskettelukeskus

var readline = require("readline");

// Super Epic Input Tool For Testing :))))
// Fun fact to the moderators: I'm a professional Node.js developer (I have done it as a job and a freelancer, since I was 13!).
// Kinda sad you use such old Node.js version as 12, but suit yourself.

if (process.env.TEST !== "1") {
  let lines = [];
  const r = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
    terminal: false,
  });
  r.on("line", (line) => {
    lines.push(line);
    if (lines[0] && lines.length === parseInt(lines[0]) + 1) {
      run(...lines);
      r.close();
    }
  });
} else {
  // This part of the program is used for input management while testing the code :)
  // CSES doesn't run this, so it can be pretty much ignored. :)
  const lines = require("fs")
    .readFileSync(`${__filename.replace(".js", ".txt")}`)
    .toString()
    .split("\n");
  run(...lines);
}

// SOLUTION STARTS
function run(...lines) {
  let slopes = []; // Offset by one

  const slopeCount = parseInt(lines.shift());
  for (let i = 0; i < slopeCount; i++) {
    slopes[i] = {
      id: i,
      needed: 0,
      passed: 0,
      route: [],
    };
  }

  // Listen for input
  for (let i = 0; i < slopeCount - 1; i++) {
    const [from, to] = lines
      .shift()
      .split(" ")
      .map((v) => parseInt(v));
    slopes[to - 1] = {
      ...slopes[to - 1],
      route: [...(slopes[from - 1].route || []), from - 1],
    };
  }

  lines
    .shift()
    .split(" ")
    .forEach((v, index) => (slopes[index].needed = parseInt(v)));

  slopes = slopes.sort((a, b) => b.route.length - a.route.length);

  const pass = (id, times = 1) => {
    const ind = slopes.findIndex((v) => v.id === id);
    slopes[ind] = {
      ...slopes[ind],
      passed: slopes[ind].passed + times,
    };
  };

  let amount = 0;
  slopes.forEach((slope, index) => {
    if (slope.needed > 0 && slope.passed < slope.needed) {
      [slope.id, ...slope.route].forEach((id) => pass(id, slope.needed));
      amount += slope.needed;
    }
  });

  console.log(amount);
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED