desislava_topuzakova

01. Train

Feb 14th, 2021
462
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.54 KB | None | 0 0
  1. import java.util.*;
  2. import java.util.stream.Collectors;
  3.  
  4. public class demo {
  5. public static void main(String[] args) {
  6. Scanner scanner = new Scanner(System.in);
  7. List<Integer> wagons = Arrays.stream(scanner.nextLine().split("\\s+"))
  8. .map(Integer::parseInt).collect(Collectors.toList());
  9. int maxCapacity = Integer.parseInt(scanner.nextLine());
  10.  
  11. String input = scanner.nextLine();
  12.  
  13. while(!input.equals("end")) {
  14. //"Add 10" -> split -> ["Add", "10"]
  15. //"30" -> split -> ["30"]
  16. String [] tokens = input.split("\\s+");
  17. if(tokens.length == 2) {
  18. int newWagon = Integer.parseInt(tokens[1]);
  19. wagons.add(newWagon);
  20. } else {
  21. int passengersToAdd = Integer.parseInt(tokens[0]);
  22. for (int index = 0; index <= wagons.size() - 1; index++) {
  23. int passengersInWagon = wagons.get(index);
  24. //мога да добавя ако не стигам макс. капацитет
  25. if (passengersInWagon + passengersToAdd <= maxCapacity) {
  26. wagons.set(index, passengersInWagon + passengersToAdd);
  27. break;
  28. }
  29. }
  30. }
  31. input = scanner.nextLine();
  32. }
  33.  
  34. printList(wagons);
  35.  
  36. }
  37.  
  38.  
  39. private static void printList(List<Integer> wagons) {
  40. for (int wagon : wagons) {
  41. System.out.print(wagon + " ");
  42. }
  43. }
  44. }
Advertisement
Add Comment
Please, Sign In to add comment