Блог

Я работаю над самой популярной задачей Quora для авторов на CodeSignal.

Учитывая 3 входных данных, нам нужно выполнить сортировку в зависимости от того, какой пользователь имеет наибольшее количество просмотров по ряду тем.

Вот мой алгоритм:

 def mostViewedWriters(topicIds, answerIds, views):
    finallist = []
    for x in enumerate(sorted(list(set(i for j in topicIds for i in j)))):
        returnlist = []
        for x in [q for p, q in enumerate(views) if q[0] in sorted([item for sublist in [z for q, z in enumerate(answerIds) if q in [i for i , y in enumerate(topicIds) if x[1] in y]] for item in sublist])]:
            sbl = [y for p, y in returnlist if p == x[1]]
            if not sbl:
                returnlist.append([x[1], x[2]])
            else:
                for idx, item in enumerate(returnlist):
                    if item[0] == x[1]:
                        added = item[1]   x[2]
                        returnlist[idx] = [x[1], added]
    
    res = sorted(returnlist, key=lambda x: x[1], reverse=True)
    finallist.append(res)
return finallist
 

К сожалению, приведенный выше алгоритм не выполняется для одного условия, которое я нашел в тексте задачи:

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

Так, например, один из моих 2D-списков, который я добавляю к объекту finallist, выглядит следующим образом:

 [[3,3],[2,3],[1,1]]
 

Поскольку значение привязано к первым двум спискам, мне нужно поменять местами эти два списка. Если бы их было 3 как таковых:

 [[3,3],[1,3],[2,3],[1,1]]
 

Их необходимо будет переупорядочить как таковые:

 [[3,3],[2,3],[1,3],[1,1]]
 

Сначала сортируется значение x [1], но затем значение x [0] сортируется следующим образом В ОБРАТНОМ ПОРЯДКЕ, x[0] — сортировка по убыванию, а x[1] — сортировка по возрастанию.

Как я могу запускать подмассивную сортировку каждый раз, когда есть связь со значением x[1]? Есть ли способ обновить лямбда-выражение в моей отсортированной функции в конце моего алгоритма, который позволит достичь этого?

Спасибо.

Это можно сделать двумя способами.

1. Поскольку в вашем массиве нет отрицательных чисел, вы можете сохранить идентификатор пользователя как отрицательный, а затем выполнить сортировку по убыванию в списке. Он автоматически получит идентификатор пользователя в порядке возрастания.
2. Используя key параметр, вы можете использовать простую лямбда-функцию, например

 l = [[1,2],[2,3],[2,1]]
l.sort(key=lambda x: (x[1], -x[0]), reverse=True)
print(l)  #[[2, 1], [2, 3], [1, 2]]
 

Используя предложение Abhinav с ключевым параметром в отрицательном значении для значения x[0], я поиграл с этим и придумал следующее:

 res = sorted(returnlist, key=lambda x: (x[1], -x[0]), reverse=True)
 

Это создает обратную сортировку как для значений x[0], так и для значений x[1], при этом сортировка x[1] является первичной, а значение -x[1] является вторичной или вспомогательной сортировкой из первого.

Код Abhinav был правильным, за исключением того, что в нем отсутствовал параметр reverse=True — необходимость в первую очередь выполнить обратную сортировку и обычную сортировку по возрастанию, поскольку целью была подмассив, и удаление обратного параметра привело к тому, что это больше не работало.