java基础：do-while练习

题目:模拟ATM取款

/*题目:模拟ATM取款
声明变量balance并初始化为0，用以表示银行账户的余额，下面通过ATM机程序实现存款，取款等功能。
8========ATM=======
1、存款
2、取款
3、 显示余额
4、退出
请选择(1-4):
*/

import java.util.Scanner;

public class TestWhileD01 {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        // 定义银行账户余额
        int balance = 0;
        // 定义一个开关
        boolean flag = true;
        do {
            // 提示ATM机功能
            System.out.println("1、存款");
            System.out.println("2、取款");
            System.out.println("3、显示余额");
            System.out.println("4、退出");
            System.out.print("请选择（1-4）：");
            // 键盘输入
            int num = scan.nextInt();

            switch (num) {
                case 1:
                    System.out.println("请输入存款金额：");
                    // 键盘输入金额；
                    int depositNum = scan.nextInt();
                    // 余额加上
                    if (depositNum > 0) {
                        balance += depositNum;
                    }

                    break;
                case 2:
                    System.out.print("请输入取款金额：");
                    // 键盘输入金额；1
                    int withdrew = scan.nextInt();
                    if (withdrew > 0 && balance >= withdrew) {
                        // 余额减掉
                        balance -= withdrew;
                    } else {
                        System.out.println("您的当前余额不足");
                    }
                    break;
                case 3:
                    System.out.println("您的余额是：" + balance);
                    break;
                case 4:
                    flag = false;
                    System.out.println("谢谢使用");
                    break;
            }
        } while (flag);
        scan.close();
    }

}