P9_PokemonDontGo*_Lists_Ex

package feb20_Lists_Exercises;

import java.util.Arrays;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

public class P9_PokemonDontGo_B {
    public static void main(String[] args) {
        @SuppressWarnings("resource")

        Scanner sc = new Scanner(System.in);

        List<Integer> pokemons = Arrays.stream(sc.nextLine().split("\\s+")).map(Integer::parseInt)
                .collect(Collectors.toList());

        int sum = 0;

        while (!pokemons.isEmpty()) {

            int indexToRemove = Integer.parseInt(sc.nextLine());

            if (indexToRemove >= 0 && indexToRemove < pokemons.size()) {
                int element = pokemons.get(indexToRemove);
                sum += element;
                pokemons.remove(pokemons.get(indexToRemove));
                changeListElements(pokemons, element);
            } else if (indexToRemove < 0) {
                int element = pokemons.get(0);
                sum += element;
                pokemons.set(0, pokemons.get(pokemons.size() - 1));
                changeListElements(pokemons, element);
            } else {
                int element = pokemons.get(pokemons.size() - 1);
                sum += element;
                pokemons.set(pokemons.size() - 1, pokemons.get(0));
                changeListElements(pokemons, element);
            }
        }

        System.out.println(sum);
    }

    public static void changeListElements(List<Integer> pokemons, int element) {
        for (int i = 0; i < pokemons.size(); i++) {
            if (pokemons.get(i) <= element) {
                pokemons.set(i, (pokemons.get(i) + element));
            } else {
                pokemons.set(i, (pokemons.get(i) - element));
            }
        }
    }
}