Advertisement
PlotnikovPhilipp

Untitled

Nov 18th, 2019
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.65 KB | None | 0 0
  1. n, k, x, y = [int(i) for i in input().split()]
  2. amountOfRooms = int(input())
  3. arr = [int(i) for i in input().split()]
  4. resultArr = [[] for i in range(n)]
  5. amountOfEdge = 0
  6. i = 0
  7. while (i + 1) <= max(arr):
  8.     if amountOfEdge == n:
  9.         amountOfEdge = 0
  10.     if (amountOfEdge + 1) % k == 0:
  11.         for j in range(x):
  12.             resultArr[amountOfEdge].append(i + 1)
  13.             i += 1
  14.     else:
  15.         for j in range(y):
  16.             resultArr[amountOfEdge].append(i + 1)
  17.             i += 1
  18.     amountOfEdge += 1
  19. print(resultArr)
  20. for i in arr:
  21.     for j in resultArr:
  22.         if i in j:
  23.             print(resultArr.index(j) + 1)
  24.             break
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement