Ristolainen pelasi alle 20-vuotiaiden MM-kilpailuissa vuosina 2012, 2013 ja 2014. Hän oli vuoden 2013 turnauksessa Suomen tehokkain puolustaja Ville Pokan kanssa tehoilla 2+4=6. Ristolainen toimi myös joukkueen kapteenina turnausta edeltäneissä harjoitusotteluissa. Vuoden 2014 turnauksessa hän oli Suomen varakapteeni ja ratkaisi joukkueelle maailmanmestaruuden tehtyään jatkoajalla 2–3-voittomaalin turnauksen isäntäjoukkue Ruotsia vastaan. Ristolainen valittiin koko turnauksen parhaaksi puolustajaksi ja tähdistökentälliseen.