Как исправить логическую ошибку в коде?

Question

[Михаил С.]

Я написал цикл, который по задумке должен был определять с помощью рандомно составленного списка цифр где будет заложена бомба, но в итоге встретился с проблемой в индексации. Как мне исправить эту ошибку не затрагивая установленное количество списков(markup) - 5.

import random
bombID = sorted(random.sample(range(0, 25), 5))

for y in range(5):
    markup = []

    for x in range(5):
        idx = y * 5 + x
        if bombID[y] == idx:
            markup.append(f'bomb{idx}')
        else:
            markup.append(str(idx))
    print(markup)

Comments

[Dmitrii]

А в каком случае возникает?

[Михаил С.]

Dmitrii,

[1, 4, 10, 12, 23] - это индексы, которые сгенерировал модуль рандома
['0', 'bomb1', '2', '3', '4']----|
['5', '6', '7', '8', '9']----|
['bomb10', '11', '12', '13', '14']----|      А это все куда он смог пробраться, то есть, например 4 он пропустил, 12 пропустил
['15', '16', '17', '18', '19']----|
['20', '21', '22', 'bomb23', '24']----|

[Dmitrii]

Михаил С.,
Тогда я бы посоветовал использовать просто список 0 до 24 и уже по индексам устанавливать бомбы. Не нужно было бы перебирать все элементы. Временная сложность у этого подхода ниже, но не соответствует требованию в вопросе).

Answer 1

[Denkuwus]

Можно попробовать немного изменить оператор if

import random

bombID = sorted(random.sample(range(0, 25), 5))

for y in range(5):
     markup = []

     for x in range(5):
         idx = y * 5 + x
         if idx in bombID:
             markup.append(f'bomb{idx}')
         else:
             markup.append(str(idx))
     print(markup)

Теперь оператор if должен быть истинным только в том случае, если текущий индекс (idx) является одним из чисел в списке bombID

Comments

[Михаил С.]

Спасибо)