Next Самое большое число одинаковых цифр

Question

Я не вижу ничего плохого в моем коде, но я не могу возвращать -1, когда вход не может произвести следующее большее число, то есть вход 531, который спускается.

import itertools as it 
def next_bigger(n): 
    if sorted("531", reverse = True) == list("531"): 
     return -1 
    s = tuple(str(n)) 
    for x in it.dropwhile(lambda x: x <= s, it.permutations(sorted(s))): 
     return int(''.join(x)) 
    return s 

Может кому-то помочь?

Answer 1

Вы можете просто использовать оператор if в начале вашей функции, чтобы проверить, находится ли номер уже в порядке сортировки в обратном порядке. Если он отсортирован return -1 немедленно:

>>> sorted("531", reverse = True) == list("531") 
True