View difference between Paste ID: Nfhmm8FX and ViV9se0b
SHOW: | | - or go back to the newest paste.
1
import java.util.Arrays;
2
import java.util.List;
3
import java.util.Scanner;
4
import java.util.stream.Collectors;
5
6
public class _09_PokemonDontGo {
7
    public static void main(String[] args) {
8
        Scanner scanner = new Scanner(System.in);
9
10
        List<Integer> pok = Arrays.stream(scanner.nextLine()
11
                .split("\\s+")).map(Integer::parseInt)
12
                .collect(Collectors.toList());
13
14
        int sum = 0;
15
16
        while (!pok.isEmpty()) {
17
            int ind = Integer.parseInt(scanner.nextLine());
18
19
            if (ind >= 0 && ind <= pok.size() - 1) {
20
                int element = pok.get(ind);
21
                sum += element;
22
                pok.remove(ind);
23
                changeListContent(pok, element);
24
            } else if (ind < 0) {
25
                int element = pok.get(0);
26
                sum += element;
27
                pok.set(0, pok.get(pok.size()-1));
28
                changeListContent(pok, element);
29
            } else {
30
                int element = pok.get(pok.size()-1);
31
                sum += element;
32
                pok.set((pok.size()-1), pok.get(0));
33
                changeListContent(pok, element);
34
            }
35
        }
36
        System.out.println(sum);
37
    }
38
39
    static void changeListContent(List<Integer> numbersList, int element) {
40
        for (int i = 0; i < numbersList.size(); i++) {
41
            if (numbersList.get(i) <= element){
42
                numbersList.set(i, numbersList.get(i) + element);
43
            } else {
44
                numbersList.set(i, numbersList.get(i) - element);
45
            }
46
        }
47
    }
48
}