Sadonkorjuu

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Scanner;

public class Cities2 {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int cityCount = Integer.parseInt(input.nextLine());
        City[] cities = new City[cityCount];
        ArrayList<Route> cursors = new ArrayList<>();

        // Cities
        int i = 0;
        for (String s : input.nextLine().split(" ")) {
            boolean isField = s.equals("1");
            City city = new City(isField);
            cities[i++] = city;
        }

        // Routes
        for (int linei = 1; linei < cityCount; linei++) {
            String[] line = input.nextLine().split(" ");

            City cityA = cities[Integer.parseInt(line[0]) - 1];
            City cityB = cities[Integer.parseInt(line[1]) - 1];
            int length = Integer.parseInt(line[2]);

            Route route1 = new Route(cityA, cityB, length);
            if (!cityA.isField) cursors.add(route1);
            else cityA.routes.add(route1);

            Route route2 = new Route(cityB, cityA, length);
            if (!cityB.isField) cursors.add(route2);
            else cityB.routes.add(route2);
        }

        // Actual stuff starts here.
        long totalLength = 0;
        while (!cursors.isEmpty()) {
            ArrayList<Route> nextCursors = new ArrayList<>();
            for (Route route : cursors) {
                City destination = route.to;
                destination.routes.remove(route);
                int newLength = route.from.shortestRoute + route.length;
                if (newLength < destination.shortestRoute) {
                    totalLength = totalLength - (destination.shortestRoute == Integer.MAX_VALUE ? 0 : destination.shortestRoute) + newLength;
                    destination.shortestRoute = newLength;
                    nextCursors.addAll(destination.routes);
                }
            }
            cursors = nextCursors;
        }

        System.out.println(totalLength);
    }

    public static class City {
        public boolean isField;
        public LinkedList<Route> routes = new LinkedList<>();
        public int shortestRoute = Integer.MAX_VALUE;

        public City(boolean isField) {
            this.isField = isField;
            if (!isField) shortestRoute = 0;
        }
    }

    public static class Route {
        public City from;
        public City to;
        public int length;

        public Route(City from, City to, int length) {
            this.from = from;
            this.to = to;
            this.length = length;
        }
    }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

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: ACCEPTED

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 1, 2

Verdict: ACCEPTED

Test 16

Group: 1, 2

Verdict: ACCEPTED

Test 17

Group: 1, 2

Verdict: ACCEPTED

Test 18

Group: 1, 2

Verdict: ACCEPTED

Test 19

Group: 1, 2

Verdict: ACCEPTED

Test 20

Group: 1, 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 1, 2

Verdict: ACCEPTED

Test 25

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 26

Group: 1, 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 28

Group: 1, 2

Verdict: ACCEPTED

Test 29

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 30

Group: 1, 2

Verdict: ACCEPTED

Test 31

Group: 2

Verdict: TIME LIMIT EXCEEDED