Cronin effect and energy conservation constraints in pA collisions at LHC and RHIC