puzzle one's brain upon a problem

ломать себе голову над проблемой (one's head over a question, oneself over the matter, etc., и т.д.)