Didart

Train - Lists

Oct 17th, 2021
115
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 1.41 KB | None | 0 0
  1. import java.util.ArrayList;
  2. import java.util.Arrays;
  3. import java.util.List;
  4. import java.util.Scanner;
  5. import java.util.stream.Collectors;
  6.  
  7. public class Train {
  8.     public static void main(String[] args) {
  9.         Scanner scanner = new Scanner(System.in);
  10.  
  11.         List<Integer> listOfWagons = Arrays
  12.                 .stream(scanner.nextLine().split(" "))
  13.                 .map(Integer::parseInt)
  14.                 .collect(Collectors.toList());
  15.  
  16.         int wagonCapacity = Integer.parseInt(scanner.nextLine());
  17.  
  18.         String command = scanner.nextLine();
  19.         while (!command.equals("end")) {
  20.             String[] input = command.split(" ");
  21.  
  22.             if (input[0].equals("Add")) {
  23.                 int numberPassengers = Integer.parseInt(input[1]);
  24.                 listOfWagons.add(numberPassengers);
  25.             } else {
  26.                 int passengersAdd = Integer.parseInt(input[0]);
  27.  
  28.                 for (int i = 0; i < listOfWagons.size(); i++) {
  29.                     int currentWagon = listOfWagons.get(i);
  30.                     if (currentWagon + passengersAdd <= wagonCapacity) {
  31.                         listOfWagons.set(i, currentWagon + passengersAdd);
  32.                         break;
  33.                     }
  34.                 }
  35.             }
  36.             command = scanner.nextLine();
  37.         }
  38.         for (int wagon : listOfWagons) {
  39.             System.out.print(wagon + " ");
  40.         }
  41.     }
  42. }
  43.  
Advertisement
Add Comment
Please, Sign In to add comment