Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package ListsExercise;
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.List;
- import java.util.Scanner;
- public class PokemonDontGo {
- public static void main(String[] args) {
- Scanner scanner = new Scanner(System.in);
- int[] input = Arrays.stream(scanner.nextLine().split("\\s+")).mapToInt(Integer::parseInt).toArray();
- List<Integer> pokemon = new ArrayList<>();
- for (int var : input) {
- pokemon.add(var);
- }
- int sum = 0;
- while (pokemon.size() > 0) {
- int indexToRemove = Integer.parseInt(scanner.nextLine());
- int indexAmount;
- if (indexToRemove < 0) {
- indexAmount = pokemon.get(0);
- pokemon.set(0, pokemon.get(pokemon.size() - 1));
- } else if (indexToRemove >= pokemon.size()) {
- indexAmount = pokemon.get(pokemon.size() - 1);
- pokemon.set(pokemon.size() - 1, pokemon.get(0));
- } else {
- indexAmount = pokemon.get(indexToRemove);
- pokemon.remove(indexToRemove);
- }
- sum += indexAmount;
- for (int i = 0; i < pokemon.size(); i++) {
- int temp = pokemon.get(i);
- if (pokemon.get(i) <= indexAmount) {
- temp += indexAmount;
- pokemon.set(i, temp);
- } else {
- temp -= indexAmount;
- pokemon.set(i, temp);
- }
- }
- }
- System.out.println(sum);
- }
- }
Add Comment
Please, Sign In to add comment