It has more to do with the balls actually. As a Medvedev fan I actually disagree with his statement that the super heavy hard court balls introduced at the start of 2023 have helped him (though he’s definitely correct that they’ve helped Jannik), because look at his HC results the last two years compared to before.
He gets impatient when he can’t easily hit through his opponent and tries to rip the cover off it without using any spin. As far as the courts go, yes, players can expose him by rushing him more on faster courts, but the dead balls exacerbate this by skidding low instead of bouncing to a comfortable height and giving him easy shots to attack. If you pay close attention, he always does much worse at the end of ball cycles than at the start.
Oh clay/grass, the ball bounces higher, plus with the less felt used on them are also lighter=much more success.