Либо я что-то не понимаю, либо это очень простая задача которая решается в 1 строку