вопрос же был как решить задачу наиболее оптимальным путем, можно решить через if но все полетит, можно решить через цикл, но вопрос будет по затрате времени, можно решить через индексы (но тоже полетит при изменении логики), можно пробовать find ... много решений но интересует оптимальное