Nel primo codice basta scrivere

codice:
	if (i==0)
	{
		printf("\nInput non valido!!\n");
		while(getchar()!='\n');
	}