At maximum height, the y-component of velocity is equal to zero.
Since there is no component of force acting in the x direction, the x component of velocity remains constant throughout the flight. The total momentum there must be equal to the x component momentum of the ball.
