Advertisement
maxim_shlyahtin

Untitled

Nov 19th, 2020
171
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.53 KB | None | 0 0
  1. n, k = list(map(int, input().split()))
  2. s = list(map(int, input().split()))
  3. if k % 2 != 0:
  4.         k += 1
  5. def test():
  6.     count_1, count_2 = 0, 0
  7.     for i in range(len(s)):
  8.         if s[i] == 1:
  9.             count_1 += 1
  10.             if count_1 == (k // 2):
  11.                 return i + 1
  12.         elif s[i] == 2:
  13.             count_2 += 1
  14.             if count_2 == (k // 2):
  15.                 return i + 1
  16.  
  17.  
  18. f = test()
  19. for i in range(len(s)):
  20.     if i < f:
  21.         s[i] = 0
  22.     else:
  23.         s[i] = 1
  24. print(f)
  25. print(*s, end=' ')
  26.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement