Число подмассивов с медианой K

Дан массив различных чисел nums и число k (гарантированно присутствует в массиве). Медиана подмассива - средний элемент после сортировки (для чётной длины - меньший из двух средних). Верните количество непрерывных подмассивов, медиана которых равна ровно k. Подсказка: зафиксируйте позицию k. Замените элементы больше k на +1, меньше k на -1, а сам k на 0, и считайте баланс (префиксную сумму) влево и вправо от позиции k - подмассив годится, если суммарный баланс равен 0 (нечётная длина) или 1 (чётная длина, при этом k должен быть МЕНЬШИМ из двух средних).