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 | } |