Task: | Sadonkorjuu |
Sender: | lompakkomies |
Submission time: | 2022-11-04 10:45:16 +0200 |
Language: | Java |
Status: | READY |
Result: | 33 |
group | verdict | score |
---|---|---|
#1 | ACCEPTED | 33 |
#2 | TIME LIMIT EXCEEDED | 0 |
test | verdict | time | group | |
---|---|---|---|---|
#1 | ACCEPTED | 0.13 s | 1, 2 | details |
#2 | ACCEPTED | 0.13 s | 1, 2 | details |
#3 | ACCEPTED | 0.13 s | 1, 2 | details |
#4 | ACCEPTED | 0.13 s | 1, 2 | details |
#5 | ACCEPTED | 0.13 s | 1, 2 | details |
#6 | ACCEPTED | 0.28 s | 1, 2 | details |
#7 | TIME LIMIT EXCEEDED | -- | 2 | details |
#8 | ACCEPTED | 0.23 s | 1, 2 | details |
#9 | TIME LIMIT EXCEEDED | -- | 2 | details |
#10 | ACCEPTED | 0.22 s | 1, 2 | details |
#11 | TIME LIMIT EXCEEDED | -- | 2 | details |
#12 | TIME LIMIT EXCEEDED | -- | 2 | details |
#13 | TIME LIMIT EXCEEDED | -- | 2 | details |
#14 | TIME LIMIT EXCEEDED | -- | 2 | details |
#15 | ACCEPTED | 0.27 s | 1, 2 | details |
#16 | ACCEPTED | 0.24 s | 1, 2 | details |
#17 | ACCEPTED | 0.24 s | 1, 2 | details |
#18 | ACCEPTED | 0.27 s | 1, 2 | details |
#19 | ACCEPTED | 0.25 s | 1, 2 | details |
#20 | ACCEPTED | 0.26 s | 1, 2 | details |
#21 | TIME LIMIT EXCEEDED | -- | 2 | details |
#22 | TIME LIMIT EXCEEDED | -- | 2 | details |
#23 | TIME LIMIT EXCEEDED | -- | 2 | details |
#24 | ACCEPTED | 0.25 s | 1, 2 | details |
#25 | TIME LIMIT EXCEEDED | -- | 2 | details |
#26 | ACCEPTED | 0.27 s | 1, 2 | details |
#27 | TIME LIMIT EXCEEDED | -- | 2 | details |
#28 | ACCEPTED | 0.27 s | 1, 2 | details |
#29 | TIME LIMIT EXCEEDED | -- | 2 | details |
#30 | ACCEPTED | 0.25 s | 1, 2 | details |
#31 | TIME LIMIT EXCEEDED | -- | 2 | details |
Code
import java.util.Scanner;public class Sadonkorjuu {public static void main(String[] args) {Scanner scan = new Scanner(System.in);int n = scan.nextInt();int[] cities = new int[n];for(int i = 0; i < n; i++) {cities[i] = scan.nextInt();}//System.out.println("ok");int[][] connections = new int[2*(n-1)][3];for(int i = 0; i < n-1; i++) {connections[i][0] = scan.nextInt();connections[i][1] = scan.nextInt();connections[i][2] = scan.nextInt();connections[i+n-1][1] = connections[i][0];connections[i+n-1][0] = connections[i][1];connections[i+n-1][2] = connections[i][2];//System.out.println("ok");}int temp1 = 0, temp2 = 0, temp3 = 0;/*for(int i = 0; i < connections.length; i++) {for(int j = 0; j < connections.length - 1; j++) {if(connections[j][0] > connections[j+1][0]) {temp1 = connections[j][0];temp2 = connections[j][1];temp3 = connections[j][2];connections[j][0] = connections[j+1][0];connections[j][1] = connections[j+1][1];connections[j][2] = connections[j+1][2];connections[j+1][0] = temp1;connections[j+1][1] = temp2;connections[j+1][2] = temp3;}}}*/quickSort(connections, 0, connections.length - 1);/*for(int i = 0; i < connections.length; i++) {System.out.print(connections[i][0]);}*/int[] cityLocations = new int[n+1];temp1 = 1;for(int i = 0; i < connections.length; i++) {if(temp1 == connections[i][0]) {cityLocations[temp1 - 1] = i;temp1++;}}cityLocations[n] = connections.length;/*System.out.println("length: " + cityLocations.length);for(int k = 0; k < n; k++) {//System.out.println("city" + k + "connections:");//System.out.println(cityLocations[k] + " " + cityLocations[k+1] + ", ");for(int i = cityLocations[k]; i < cityLocations[k+1]; i++) {for(int j = 0; j < connections[0].length; j++) {System.out.print(connections[i][j] + ", ");}System.out.print("\n");}}for(int i = 0; i < cityLocations.length; i++) {System.out.print(cityLocations[i] + ", ");}*/int[] distances = new int[n];for(int i = 0; i < n; i++) {distances[i] = -1;}int sum = 0;for(int i = 0; i < n; i++) {if(cities[i] == 1) {distance(connections, cityLocations, cities, distances, i, 0, -1, i);if(distances[i] != -1) {sum += distances[i];}//System.out.println(distances[i]);//System.out.println("\n" + i);//sum += scan.nextInt();}}System.out.println(sum);/*for(int i = 0; i < n; i++) {System.out.print(", " + distances[i]);}*/}public static void distance(int[][] connections, int[] cityLocations, int[] cities, int[] distances, int cityNum, int distance, int prevCity, int startCity) {if(cities[cityNum] == 1) {for(int i = cityLocations[cityNum]; i < cityLocations[cityNum + 1]; i++) {if(connections[i][1]-1 != prevCity) {if(distance + connections[i][2]< distances[startCity] || distances[startCity] == -1) {//System.out.println( connections[i][0] + ", " + connections[i][1] + ", " + connections[i][2]);distance(connections, cityLocations, cities, distances, connections[i][1]-1, distance + connections[i][2], cityNum, startCity);}}}return;} else {if(distance < distances[startCity] || distances[startCity] == -1) {distances[startCity] = distance;return;}}return;}static void swap(int[][] connections, int i, int j){int temp1 = connections[i][0];int temp2 = connections[i][1];int temp3 = connections[i][2];connections[i][0] = connections[j][0];connections[i][1] = connections[j][1];connections[i][2] = connections[j][2];connections[j][0] = temp1;connections[j][1] = temp2;connections[j][2] = temp3;}static int partition(int[][] connections, int low, int high){int pivot = connections[high][0];int i = (low - 1);for (int j = low; j <= high - 1; j++) {if (connections[j][0] < pivot) {i++;swap(connections, i, j);}}swap(connections, i + 1, high);return (i + 1);}static void quickSort(int[][] connections, int low, int high){if (low < high) {int pi = partition(connections, low, high);quickSort(connections, low, pi - 1);quickSort(connections, pi + 1, high);}}}
Test details
Test 1
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1 0 |
correct output |
---|
0 |
user output |
---|
0 |
Test 2
Group: 1, 2
Verdict: ACCEPTED
input |
---|
5 0 0 0 0 0 1 2 1 2 3 2 3 4 3 ... |
correct output |
---|
0 |
user output |
---|
0 |
Test 3
Group: 1, 2
Verdict: ACCEPTED
input |
---|
4 1 0 1 1 1 2 10 2 3 20 2 4 30 |
correct output |
---|
60 |
user output |
---|
60 |
Test 4
Group: 1, 2
Verdict: ACCEPTED
input |
---|
5 0 1 1 1 0 1 2 10 2 3 20 3 4 30 ... |
correct output |
---|
80 |
user output |
---|
80 |
Test 5
Group: 1, 2
Verdict: ACCEPTED
input |
---|
5 0 1 0 1 1 1 2 1 2 3 5 3 4 3 ... |
correct output |
---|
6 |
user output |
---|
6 |
Test 6
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
5506363 |
user output |
---|
5506363 |
Test 7
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
1795118520 |
user output |
---|
(empty) |
Test 8
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 0 0 1 0 1 1 0 1 0 1 1 0 0 0 1 ... |
correct output |
---|
293576 |
user output |
---|
293576 |
Test 9
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
816932444 |
user output |
---|
(empty) |
Test 10
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ... |
correct output |
---|
3089 |
user output |
---|
3089 |
Test 11
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ... |
correct output |
---|
40839 |
user output |
---|
(empty) |
Test 12
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
5683983203973 |
user output |
---|
(empty) |
Test 13
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 0 1 1 1 1 1 1 0 0 0 1 1 0 1 0 ... |
correct output |
---|
58572993 |
user output |
---|
(empty) |
Test 14
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ... |
correct output |
---|
32755 |
user output |
---|
(empty) |
Test 15
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
126238345 |
user output |
---|
126238345 |
Test 16
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 0 0 0 1 0 1 1 1 0 0 1 0 1 1 0 ... |
correct output |
---|
278678 |
user output |
---|
278678 |
Test 17
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 0 0 0 1 0 0 0 1 0 0 0 0 0 0 ... |
correct output |
---|
34929 |
user output |
---|
34929 |
Test 18
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
1543963 |
user output |
---|
1543963 |
Test 19
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ... |
correct output |
---|
39606 |
user output |
---|
39606 |
Test 20
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 0 1 0 1 0 0 0 0 1 1 0 0 0 1 ... |
correct output |
---|
321598 |
user output |
---|
321598 |
Test 21
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
978670626 |
user output |
---|
(empty) |
Test 22
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ... |
correct output |
---|
375218 |
user output |
---|
(empty) |
Test 23
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 0 0 0 0 0 1 0 1 0 1 1 ... |
correct output |
---|
60422556 |
user output |
---|
(empty) |
Test 24
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
291990 |
user output |
---|
291990 |
Test 25
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
59607954 |
user output |
---|
(empty) |
Test 26
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
990 |
user output |
---|
990 |
Test 27
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
199982 |
user output |
---|
(empty) |
Test 28
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
7987 |
user output |
---|
7987 |
Test 29
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
3137875 |
user output |
---|
(empty) |
Test 30
Group: 1, 2
Verdict: ACCEPTED
input |
---|
1000 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
4657693 |
user output |
---|
4657693 |
Test 31
Group: 2
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
correct output |
---|
1652889357 |
user output |
---|
(empty) |