Mail.ruПочтаМой МирОдноклассникиВКонтактеИгрыЗнакомстваНовостиКалендарьОблакоЗаметкиВсе проекты

Помогите решить задачу на питоне

あなたの自我を放 っておいてください Ученик (41), на голосовании 1 месяц назад
Перед вами в ряд лежит
n
шаров. У вас есть краски
k
разных цветов. Вы можете покрасить любой шар в любой цвет, красок каждого цвета бесконечно. Сколько существует способов раскрасить шары так, чтобы никакие два соседних шара не были покрашены в одинаковый цвет? Каждый шар должен быть покрашен в один из
k
цветов.

Формат входных данных
Даны два целых числа
n
и
k
, по одному на строке (
1

n

1000
,
2

k

1000
).

Формат результата
Выведите одно число — ответ на задачу.

Примеры
Входные данные
2
2
Результат работы
2
Голосование за лучший ответ
기억 Сострадай Мастер (1196) 2 месяца назад
Ну там крч всякие input's for i range и тому подобное, удачи брад
あなたの自我を放 っておいてくださいУченик (41) 2 месяца назад
круто, но нужен код без for, if , else
Татьяна Просветленный (26991) 2 месяца назад
 
# Считываем входные данные
n = int(input())
k = int(input())
# Инициализируем список для хранения количества способов раскрасить шары
dp = [0] * (n + 1)
# Базовый случай: 1 способ раскрасить 1 шар в k цветов
dp[1] = k
# Заполняем список dp с использованием динамического программирования
list(map(lambda i: dp.__setitem__(i, (k - 1) * (dp[i - 1] + dp[i - 2])), range(2, n + 1)))

# Выводим результат
print(dp[n])
и упрощенном коде
 n = int(input()) 
k = int(input())
dp = [0] * (n + 1)
dp[1] = k
list(map(lambda i: dp.__setitem__(i, (k - 1) * (dp[i - 1] + dp[i - 2])), range(2, n + 1)))
print(((k - 1) * (dp[n - 1] + dp[n - 2])) if n > 1 else k)


Результат
あなたの自我を放 っておいてくださいУченик (41) 2 месяца назад
а можно проще написать?
Татьяна Просветленный (26991) Hajsjdbdjs Jsjdhd, написан там и код в 1 строку
Светлана Мещенко Ученик (158) 1 месяц назад
воспользуйся формулой
k * (k-1)**(n-1)
Похожие вопросы