Имея список индексов и ноль или более списков целых чисел, выведите списки целых чисел, отсортированные в порядке возрастания, с приоритетом ключа из первого ввода.
пример
Пусть будет ввод ключей [1, 0, 2]
, а ввод списков будет [[5, 3, 4], [6, 2, 1], [5, 2, 1]]
. Эти списки должны быть отсортированы по их второму элементу, затем первому элементу, затем третьему элементу в порядке возрастания:
- Сначала мы сортируем значения по индексу
1
:[[6, 2, 1], [5, 2, 1], [5, 3, 4]]
- Далее мы разрываем любые связи из первой сортировки, используя значения в индексе
0
:[[5, 2, 1], [6, 2, 1], [5, 3, 4]]
- Наконец, мы разрываем все оставшиеся связи с индексами в индексах
2
(это на самом деле ничего не меняет, потому что связей не осталось).
Детали
- Сортировка стабильна: если два элемента сравниваются равными по отношению к заданным ключам сортировки, они должны оставаться в одном и том же относительном порядке в выходных данных. Например, если
A
иB
при заданных ключах сортировки равны, а вход был[..., A, ..., B, ...]
,A
должен быть помещен передB
в выводе. - Ключ сортировки никогда не будет ссылаться на несуществующий элемент в одном из списков ввода.
- Ключ сортировки не будет повторяться. Таким образом,
[1, 2, 1]
неверный список ключей сортировки. - Любые элементы, на которые не ссылаются ключи сортировки, не учитываются в порядке сортировки. Только начальный относительный порядок и значения элементов, на которые ссылаются ключи сортировки, определяют порядок вывода.
- Вы можете выбрать, будут ли ключи сортировки с нулевой или одной индексацией.
- В ключах сортировки не будет отрицательных значений. Если вы решите использовать одноиндексирование, в ключах сортировки также не будет нулей.
- Целочисленные значения не будут превышать естественного представления вашего языка. Если выбранный вами язык обладает способностью к целым числам произвольной точности (например, Python), тогда любое целочисленное значение может присутствовать во входных данных с учетом ограничений памяти.
Реализация ссылок (Python 2)
#!/usr/bin/env python
keys = input()
lists = input()
print sorted(lists, key=lambda l:[l[x] for x in keys])
Тестовые случаи
Формат: keys lists -> output
. Все ключи сортировки имеют нулевую индексацию.
[1, 0, 2] [[5, 3, 4], [6, 2, 1], [5, 2, 1]] -> [[5, 2, 1], [6, 2, 1], [5, 3, 4]]
[1, 2] [[5, 3, 4], [6, 2, 1], [5, 2, 1]] -> [[6, 2, 1], [5, 2, 1], [5, 3, 4]]
[0, 1] [[1, 2], [2, 1]] -> [[1, 2], [2, 1]]
[1, 0] [[1, 2], [2, 1]] -> [[2, 1], [1, 2]]
[0] [[4], [10, 11, -88], [-2, 7]] -> [[-2, 7], [4], [10, 11, -88]]
[2] [[-1, -5, 8, -1, -4, -10, -5, 4, 4, 6, -8, 4, 2], [-7, 6, 2, -8, -7, 7, -3, 3, 0, -6, 1], [-9, 8, -5, -1, -7, -8, -5, -6, 5, -6, 6]] -> [[-9, 8, -5, -1, -7, -8, -5, -6, 5, -6, 6], [-7, 6, 2, -8, -7, 7, -3, 3, 0, -6, 1], [-1, -5, 8, -1, -4, -10, -5, 4, 4, 6, -8, 4, 2]]
[2, 1] [[9, 2, -2, -10, -6], [3, -4, -2]] -> [[3, -4, -2], [9, 2, -2, -10, -6]]
[2, 4, 8] [[5, -3, 4, -6, -1, -2, -2, -4, 5], [-2, -3, 6, -4, -1, -4, -4, -5, 8, 9, 9, -3, 3, -9, -3], [2, 0, 10, -10, -1, 2, -1, 5, -1, 10, -5], [-7, -8, -6, 7, 3, 8, 6, -7, -2, 0, -6, -4, 4, -3, 2, -3]] -> [[-7, -8, -6, 7, 3, 8, 6, -7, -2, 0, -6, -4, 4, -3, 2, -3], [5, -3, 4, -6, -1, -2, -2, -4, 5], [-2, -3, 6, -4, -1, -4, -4, -5, 8, 9, 9, -3, 3, -9, -3], [2, 0, 10, -10, -1, 2, -1, 5, -1, 10, -5]]
[1, 2, 3, 4, 5] [[-7, 3, -8, 3, 5, -1, 6, -6, 9, 8], [-9, -1, -7, -9, -10, -2, -8, -10, -10, -3], [5, 3, -6, -5, -4, -4, -8, 2], [9, -4, 1, -1, -3, -2], [-6, -10, 4, -10, 6, 6, -1, 3, 0, 0], [1, -2, -7, -6, -7, -7, -1, 0, -4, 3, 3], [7, -1, -7, 2, -2, 9, 7, 5, -6, -8], [1, -5, -3, -10, -7, 9, -8, -5, -1], [-9, 4, -1, -1, 2, 4]] -> [[-6, -10, 4, -10, 6, 6, -1, 3, 0, 0], [1, -5, -3, -10, -7, 9, -8, -5, -1], [9, -4, 1, -1, -3, -2], [1, -2, -7, -6, -7, -7, -1, 0, -4, 3, 3], [-9, -1, -7, -9, -10, -2, -8, -10, -10, -3], [7, -1, -7, 2, -2, 9, 7, 5, -6, -8], [-7, 3, -8, 3, 5, -1, 6, -6, 9, 8], [5, 3, -6, -5, -4, -4, -8, 2], [-9, 4, -1, -1, 2, 4]]
[8, 7, 3, 2, 4, 9, 1] [[8, -5, 1, -6, -1, -4, 6, 10, 10, 6, 9, 5], [4, -8, 6, -10, -2, -3, 2, -6, 9, 5, 4, 10, 2, 3], [10, -1, 3, 0, -4, 1, -5, -4, -1, -7, 9, -9, -1, -5, 7, 8, 9, 6, -3], [0, -9, -7, -2, 2, -5, 7, 4, 6, -4, 1, 8, -7, 10], [5, 6, -9, 0, -1, 5, 4, 7, 5, 10, 2, 5, 7, -9]] -> [[10, -1, 3, 0, -4, 1, -5, -4, -1, -7, 9, -9, -1, -5, 7, 8, 9, 6, -3], [5, 6, -9, 0, -1, 5, 4, 7, 5, 10, 2, 5, 7, -9], [0, -9, -7, -2, 2, -5, 7, 4, 6, -4, 1, 8, -7, 10], [4, -8, 6, -10, -2, -3, 2, -6, 9, 5, 4, 10, 2, 3], [8, -5, 1, -6, -1, -4, 6, 10, 10, 6, 9, 5]]