Con un poco de cosmética lo que esta haciendo es un máquina de estado, pero le falta mas forma.
Es claro que se trata de un programa propuesto como tarea escolar, así que las opciones para resolverlas deberían ser bien recibidas y según su entusiasmo y motivación aprender lo que mejor le sirva.
Usar máquina de estados es la mejor opción siempre, pero preparar tu cerebro para de aquí en mas hacerlo siempre de ese modo lleva tiempo.