Как получить случайное число в диапазоне с исключениями?

Question

[bitbuff]

Например, в питоне есть функция randint(min, max), которая отдаёт случайное число в диапазоне от min до max.
Мне нужна функция, которая будет принимать массив из исключений: randint_se(min, max, exceptions), где exceptions - массив из тех числ, которые не должны попасть в диапазон. Массив exceptions может быть любой длины.

Answer 1

[bitbuff]

Я подумал и решил задачу. Может кому-нибудь пригодится.

import random

def randint_se(min, max, exceptions):

	possible = []
	exceptions.sort()

	for number in range(min, max+1):
		count = 0
		for exception_number in exceptions:
			if number == exception_number:
				break
			count += 1
		if count == len(exceptions):
			possible.append(number)


	return random.choice(possible)

Comments

[xdgadd]

Зачем вы сортируете массив исключений, если проверяете его линейно, а не бинпоиском?
Зачем делать массив possibles, а потосм выбирать число, если есть random.randint?
etc.

import random

# через while
def randint_se(min, max, exc):
    number = random.randint(min, max)
    while number in exc:
        number = random.randint(min, max)
    return number

# рекурсивно
def randint_se_r(min, max, exc):
    number = random.randint(min, max)
    if number in exc:
        return randint_se_r(min, max, exc)
    return number

[bitbuff]

xdgadd: Я конечно понимаю, что ваш код лаконичнее. А теперь представьте, что в цикле while number in exc раз десять выпадает число из исключений. Я пишу мини-игру в которой процессор должен пахать над реальными задачами

[xdgadd]

bitbuff: пока не выпадет 10^9 раз, не занимайтесь оптимизацией на спичках, потому что современные процессоры в среднем выполняют именно столько операций в секунду. А вероятность этого вообще от нуля неотличима. Беспокоиться бы стоило, если бы и exceptions, и range были огромными, например 10^6. Но такой массив просто-напросто не влезет к вам в память.

Answer 2

[NtropyNN]

Мой вариант такой:
нужно скорректировать максимальное значение с учётом длин всех исключённых диапазонов, получить случайное число для обновлённого диапазона, а затем применяя список исключённых диапазонов скорректировать значение.
Пример: задано вернуть число от 1 до 6 (игральная кость), но при этом число 4 считается запретным.
Решение: получать случайное число от 1 до 5 и если оно больше и равно 4 плюсовать к результату 1 (длину исключенного диапазона).