можно и на флексе, только не одним контейнером.. иногда вместо того чтобы мучатся, в тестовом задании пишешь "задача реализуется проще и с меньшей разметкой при помощи grid", если они упарываются и хотят флексы, пусть сами верстают, ибо для кого прогресс придуман, не понятно.