Sram crankset compatibility
If you look up SRAM S350-1 (single chain ring) on SRAM’s website it says that it is a direct replacement for Rival 1 and compatible with 130, 135, and 142 OLD frames.  Rival 1 should be a straight swap assuming you get the correct axle variant (i.e., GPX).