구구단 출력 프로그램 (while문)

import java.util.Scanner;

public class Quiz_03 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);

		int input=0;
		int su=0;
		int button;

		button=1;

		//스위치역할변수 없이 무한히 반복하려면 while(true)라고 하면 됨

		while(button==1) {
			System.out.println("=== 구구단 출력 프로그램 ===");
			System.out.print("1~9단 중 선택 : ");
			input=Integer.parseInt(sc.nextLine());
			System.out.println();
			
			if(input>=10 || 1>input) {
				System.out.println("입력이 잘못되었습니다.");
			}else {
				while(su<9) {
					su++;
					System.out.println(input+" * " +su + " = " + (input*su));
				}
				su=0;
			}
			System.out.println("계속하시겠습니까?  1.네  2.아니오");
			System.out.print(">");
			button=Integer.parseInt(sc.nextLine());
			if(button==2) {
				System.out.println("구구단 종료");
			}
		}

	}
}