Тут банальный if стоит, можешь даже функцией делать из main(),проверку чтобы не было фигни в значениях от пользователя do{} while(). Нужно понять что в if прописывать, препода надо было слушать - что он дал, какие библиотеки сказал юзать, примеры может дал?