import java.util.Random;
import java.util.Scanner;

public class FlashCards_SC
{
public static void main (String[] args)
{
Random generator = new Random();

int num1 = generator.nextInt(100);
int num2 = generator.nextInt(100);
System.out.println(num1 + " and " + num2);

Scanner input = new Scanner(System.in);

System.out.println("Which type of problem would you like to solve (please enter 0 for addition or 1 for subtraction): ");

int selection = input.nextInt();

int count = 0;
while(selection != 0 && selection != 1 && count < 2)
{
System.out.println("Sorry, you entered the wrong number (please enter 0 for addition or 1 for subtraction): ");
selection = input.nextInt();
count++;

if(count >= 2)
selection = 0; //Default case when user doesn't enter a valid input after two tries.
}

int correctAnswer = 0;
int guess = 0;

if (selection == 0)
{
correctAnswer = num1 + num2;
System.out.println("Ok, " + num1 + " + " + num2 + " = ");
guess = input.nextInt();

while (guess != correctAnswer)
{
System.out.println("Sorry, " + num1 + " + " + num2 + " is not " + guess + ", please try again. " + num1 + " + " + num2 + " = ");
guess = input.nextInt();
count++;

if (count >= 2)
System.out.println("Sorry, " + num1 + " + " + num2 + " is not " + guess + ". The correct answer is " + num1 + " + " + num2 + " = " + correctAnswer);
}

//Above is a second while loop to allow the user to make more than one guess when they are wrong.

if (guess == correctAnswer)
{
System.out.println("Congratulations, you got it right! " + num1 + " + " + num2 + " = " + guess);
}
}

int correctAnswer = 0;
int guess = 0;

if (selection == 1)
{
correctAnswer = num1 - num2;
System.out.println("Ok, " + num1 + " - " + num2 + " = ");
guess = input.nextInt();

while (guess != correctAnswer)
{
System.out.println("Sorry, " + num1 + " - " + num2 + " is not " + guess + ", please try again. " + num1 + " - " + num2 + " = ");
guess = input.nextInt();
count++;

if (count >= 2)
System.out.print("Sorry, " + num1 + " - " + num2 + " is not " + guess + ". The correct answer is " + num1 + " - " + num2 " = " + correctAnswer);
}

//Above is a second while loop to allow the user to make more than one guess when they are wrong.

if (guess == correctAnswer)
{
System.out.println("Congratulations, you got it right! " + num1 + " - " + num2 + " = " + guess);
}
}
}
}

System.out.print("Sorry, " + num1 + " - " + num2 + " is not " + guess + ". The correct answer is " + num1 + " - " + num2 " = " + correctAnswer);

System.out.print("Sorry, " + num1 + " - " + num2 + " is not " + guess + ". The correct answer is " + num1 + " - " + num2 + " = " + correctAnswer);