Vainio pelasi Suomen maajoukkueessa 22 maaottelua, joissa hän teki kuusi maalia ja sai kolme syöttöpistettä. Hän edusti Suomea Squaw Valleyn olympialaisissa 1960 ja Sveitsin MM-kilpailuissa 1961.