Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package Fundamentals.List.Exercise;
- import java.util.Arrays;
- import java.util.List;
- import java.util.Scanner;
- import java.util.stream.Collectors;
- public class PokemonDoNotGo {
- public static void main(String[] args) {
- Scanner scanner = new Scanner(System.in);
- List<Integer> inputNumbers = Arrays.stream(scanner.nextLine().split(" "))
- .map(Integer::parseInt).collect(Collectors.toList());
- int sumRemovedItems = 0;
- while (inputNumbers.size() > 0) {
- int index = Integer.parseInt(scanner.nextLine());
- if (index < 0) {
- int valueToReplace = inputNumbers.get(inputNumbers.size() - 1);
- sumRemovedItems += inputNumbers.get(0);
- for (int i = 0; i < inputNumbers.size(); i++) {
- if (valueToReplace < inputNumbers.get(i)) {
- inputNumbers.set(i, inputNumbers.get(i) - valueToReplace);
- } else {
- inputNumbers.set(i, inputNumbers.get(i) + valueToReplace);
- }
- }
- } else if (index >= inputNumbers.size()) {
- int valueToReplace = inputNumbers.get(0);
- sumRemovedItems += inputNumbers.get(inputNumbers.size() - 1);
- for (int i = 0; i < inputNumbers.size(); i++) {
- if (valueToReplace < inputNumbers.get(i)) {
- inputNumbers.set(i, inputNumbers.get(i) - valueToReplace);
- } else {
- inputNumbers.set(i, inputNumbers.get(i) + valueToReplace);
- }
- }
- } else {
- int indexValue = inputNumbers.get(index);
- sumRemovedItems += inputNumbers.get(index);
- inputNumbers.remove(index);
- for (int i = 0; i < inputNumbers.size(); i++) {
- if (indexValue < inputNumbers.get(i)) {
- inputNumbers.set(i, inputNumbers.get(i) - indexValue);
- } else {
- inputNumbers.set(i, inputNumbers.get(i) + indexValue);
- }
- }
- }
- }
- System.out.printf("%d", sumRemovedItems);
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment