Anime where the school boy hero is good with puzzles, can fly with an item he collects

Phi Brain: Puzzle of God is a decent superficial match for "good with puzzles," but I'm not certain that he can fly.

No one loves puzzles more than Kaito Diamon, and when it comes to solving them, he's the best there is. But when a labyrinth underneath his school leads him to an mysterious armband that can unlock his brain's full potential, he is put on the Puzzle Of God organization's radar as one of the children of Phi-Brain. Now he must push his mind to the limit as the P.O.G. challenges him relentlessly to their deadly puzzles. With a plethora of dangerous, laughable, creative, or down-right demented puzzles being thrown at him, will Kaito have what it takes to solve his way to victory or will he fall against the puzzles he adores so much?

Trailer