MATL , 17 13 байт
:tt!/XR6#uG))
Попробуйте онлайн! Или проверьте все тестовые случаи .
Размер ввода может быть ограничен точностью с плавающей запятой. Все тесты дают правильный результат.
объяснение
Это генерирует все дроби k/m
с k
, m
в [1 2 ...n]
виде n
× n
матрицы. Строка указывает числитель, а столбец - знаменатель. На самом деле матричная запись содержит обратную дробь m/k
вместо k/m
, но это не имеет значения и может быть проигнорировано в остальной части объяснения.
Матричные записи неявно считаются отсортированными в порядке столбцов. В этом случае это соответствует требуемому порядку: знаменатель, затем числитель.
Три типа записей должны быть проигнорированы из этой матрицы:
- Записи
k/m
, k>m
которые имеют то же значение, что и предыдущая запись (например, 2/4
игнорируются, потому что они совпадают с 1/2
)
- Записи
k/k
, k>1
. Записи, числитель которых превышает знаменатель
- Записи
k/m
, k<m
(они не являются частью этой проблемы).
Игнорирование записей осуществляется с помощью unique
функции, которая стабильно удаляет повторяющиеся значения и выводит индексы сохранившихся записей. При этом записи типа 1 выше автоматически удаляются. Для работы с типами 2 и 3, записи матрицы по диагонали и ниже установлены в 0
. Таким образом, будут удалены все нулевые записи, кроме первой (соответствующей действительной дроби 1/1
).
Рассмотрим ввод 4
в качестве примера.
: % Input n implicitly. Push range [1 2 ...n]
% STACK: [1 2 3 4]
t % Duplicate
% STACK: [1 2 3 4], [1 2 3 4]
t! % Duplicate and transpose
% STACK: [1 2 3 4], [1 2 3 4], [1; 2; 3; 4]
/ % Divide element-wise with broadcast: gives matrix with all pairs
% STACK: [1 2 3 4], [1 2 3 4;
0.5000 1 1.5000 2;
0.3333 0.6667 1 1.3333;
0.2500 0.5000 0.7500 1 ]
XR % Upper triangular part above the diagonal. This sets to 0 all entries
% corresponding to fractions that equal or exceed 1. (Since the matrix
% actually contains the inverse fractions, nonzero entries will contain
% values greater than 1)
% STACK: [1 2 3 4], [0 2 3 4;
0 0 1.5000 2;
0 0 0 1.3333;
0 0 0 0 ]
6#u % Indices of first appearance of unique elements
% STACK: [1 2 3 4], [1; 5; 9; 10; 13; 15]
G % Push input n again
% STACK: [1 2 3 4], [1; 5; 9; 10; 13; 15], 4
) % Index: get the n-th entry from the array of indices of unique elements
% STACK: [1 2 3 4], 10
) % Index (modular): get the corresponding real part. Display implicitly
% STACK: 2