Статья Автор: Семенюк Александр

Антология ЕГЭ (2 кучи) - код

def pi(pos):
  x, y = pos
  return {(x + 1, y), (x, y + 1), (x * 3, y), (x, y * 3)} #1
def step(S, X, t):
  Y = set()
  for p in S:
    if (t % 2 == 1) and (pi(p) & X) : Y.add(p)
    if (t % 2 == 0) and (pi(p) <= X) : Y.add(p)
  return Y
swin, aa = 65, 6 #2 #3
k = 3 #4
G = set([(x, y) for x in range (1, swin) for y in range (1, swin - x)]) 
L = set([(x, y) for x in range (1,k * swin) for y in range (1,k * swin - x)]) 
L = L - G
W = set()
Start = set([(x, y) for x in range (aa, aa + 1) for y in range (1, swin - x)])
LW = {0 : W, 1 : L}
t = 0
while t < 24:
  t += 1
  R = step( G, LW[t % 2], t)
  G = G - R
  LW[(t + 1) % 2] = LW[(t + 1) % 2].union(R)
  print(t, len(R), sorted(R & Start)'[:10]') #если слишком много чисел при первом выводе

#1 - операции, которые можно применять к куче (определяются по первым строкам условия задачи)
#2 - суммарное кол-во камней в обеих кучах (выделено жирным черным в условии задачи)
#3 - начальное кол-во камней в одной из куч (так же выделено жирным черным в условии задачи)

Чтобы оставить комментарий нужна авторизация

Печать