자바 학생정보 등록 및 조회 프로그램

• 유튜브 알고리즘 기초100제 1/100
  https://www.youtube.com/watch?v=w023dXv03nk&list=PLVoihNyHW4xkm_KJ8_N8X7F6EQP4uSRyR&index=2

  1) main.class

package basic001;

import java.util.ArrayList;  
import java.util.Scanner;

public class main {

public static void main(String[] args) {

    Scanner scanner = new Scanner(System.in);
    ArrayList<Student> arr = new ArrayList<Student>();
    String ans = "";

    do {

        System.out.print("[1]학생정보 저장   [2]학생조회");
        int choice = scanner.nextInt();

        if (choice == 1) {
            System.out.print("등록할 학생이름 입력>> ");
            String name = scanner.next();

            System.out.print("등록할 학번 입력>>");
            int no = scanner.nextInt();

            Student std = new Student(name, no);
            arr.add(std);

        } else if (choice == 2) {

            System.out.print("검색할 학생 이름을 입력>> ");
            String searchName = scanner.next();

            if (arr.size() != 0) {
                for (int i = 0; i < arr.size(); i++) {
                    if (searchName.equals(arr.get(i).getName())) {
                        System.out.println(searchName + "의 학번은 " + arr.get(i).getNo() + "입니다.");
                    }
                }
            } else {
                System.out.println("해당 학생 정보가 존재하지 않습니다.");
            }

        } else {
            System.out.print("메뉴 번호를 확인하시고 다시 입력해주세요.");
        }

        System.out.print("계속하시겠습니까? [Y/N]");
        ans = scanner.next();
        if (ans.equals("N") || ans.equals("n"))
            break;
    } while (ans.equals("Y") || ans.equals("y"));
    System.out.println("학생조회 서비스를 종료합니다.");

}


}

2) Student.class

package basic001;

public class Student {

    private String name;
    private int no;

    public StudentDTO(String name, int no) {
        this.name = name;
        this.no = no;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getNo() {
        return no;
    }

    public void setNo(int no) {
        this.no = no;
    }

}