Ответ: Придумайте более быстрый алгоритм.
Это непростая проблема (помечена как «сложная» на Hackerrank, хотя на самом деле это «средняя» IMHO), поэтому не ожидайте, что буквальная реализация алгоритма, как описано, решит ее. Этот алгоритм занимает O (нм) время, но есть простой метод, который занимает всего O (n + m) время. Я кодировал его на Hackerrank в 10 строках кода: он прошел все тестовые случаи с первой попытки.
Я не даю вам решение здесь, так как вы должны решить это самостоятельно, но, кроме инициализации, есть только один проход по m запросам, принимая O ( 1) объем работы для каждого, за которым следует один проход по элементам n , снова принимая O (1) для каждого. Подсказка: думайте дифференцированно.