Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.List;
- import java.util.Scanner;
- import java.util.stream.Collectors;
- public class Train {
- public static void main(String[] args) {
- Scanner scanner = new Scanner(System.in);
- List<Integer> listOfWagons = Arrays
- .stream(scanner.nextLine().split(" "))
- .map(Integer::parseInt)
- .collect(Collectors.toList());
- int wagonCapacity = Integer.parseInt(scanner.nextLine());
- String command = scanner.nextLine();
- while (!command.equals("end")) {
- String[] input = command.split(" ");
- if (input[0].equals("Add")) {
- int numberPassengers = Integer.parseInt(input[1]);
- listOfWagons.add(numberPassengers);
- } else {
- int passengersAdd = Integer.parseInt(input[0]);
- for (int i = 0; i < listOfWagons.size(); i++) {
- int currentWagon = listOfWagons.get(i);
- if (currentWagon + passengersAdd <= wagonCapacity) {
- listOfWagons.set(i, currentWagon + passengersAdd);
- break;
- }
- }
- }
- command = scanner.nextLine();
- }
- for (int wagon : listOfWagons) {
- System.out.print(wagon + " ");
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment