Задачник по алгоритмам

Программа курса:

Задача 1: Следующий Великий Элемент I

Напишите определение функции next_greater_element(nums1, nums2), которая принимает:

  • nums1 — список целых чисел, являющийся подмножеством nums2.
  • nums2 — список целых чисел.

Функция должна для каждого элемента из списка nums1 найти индекс j, такой что nums1[i] == nums2[j], и определить следующий больший элемент nums2[j], находящийся справа в nums2. Если такого элемента нет, то вернуть -1.

Функция возвращает список ans, где ans[i] — это следующий больший элемент для nums1[i].

Примеры:

Пример 1:

  • Ввод:
    • nums1 = [4, 1, 2]
    • nums2 = [1, 3, 4, 2]
  • Вывод: [-1, 3, -1]
  • Объяснение:
    • Для числа 4 в nums2 = [1, 3, 4, 2] нет большего элемента справа, ответ -1.
    • Для числа 1 в nums2 = [1, 3, 4, 2] следующий больший элемент равен 3.
    • Для числа 2 в nums2 = [1, 3, 4, 2] нет большего элемента справа, ответ -1.

Пример 2:

  • Ввод:
    • nums1 = [2, 4]
    • nums2 = [1, 2, 3, 4]
  • Вывод: [3, -1]
  • Объяснение:
    • Для числа 2 в nums2 = [1, 2, 3, 4] следующий больший элемент равен 3.
    • Для числа 4 в nums2 = [1, 2, 3, 4] нет большего элемента справа, ответ -1.

Примечание:

Каждый элемент в nums1 гарантированно существует в nums2, и оба списка не содержат повторяющихся значений.

0
Вы должны Войти или Зарегистрироваться чтобы оставлять комментарии