Surface area of the fin stack matters. An air cooler will always be limited by the space available around the CPU. A water cooling radiator has more flexibility to be placed in around the case.
That said, having less than a 360mm AIO is probably a waste. Also, higher end Intel chips these days are so power hungry that they can’t be physically cooled properly with the surface area available on the package.