ATM[4단계]

# ATM[4단계]
# 1. 가입
# . 계좌번호와 비밀번호를 입력받아 가입
# . 계좌번호 중복검사
# 2. 탈퇴
# . 계좌번호를 입력받아 탈퇴

accs = [1001, 1002, 0, 0, 0]
pws  = [1111, 2222, 0, 0, 0]

cnt = 2

run = True
while run:
    print("[1]가입")
    print("[2]탈퇴")
    print("[0]종료")

    choice = int(input("메뉴 선택 : "))

    if choice == 1:
        num = int(input("사용할 계좌번호 입력: "))
        i = 0
        count = 0
        while i < cnt:
            if accs[i] == num:
                count += 1
                break
            i += 1
        if count != 0:
            print("중복입니다.")
            continue
        accs[cnt] = num       
        pw = int(input("사용할 비밀번호 입력: "))
        pws[cnt] = pw
        cnt += 1
        
    elif choice == 2:
        idx = -1
        pw = int(input("비밀번호를 입력하세요: "))
        while i < cnt:
            if pws[i] == pw:
                count += 1
                idx = i
                break
            i += 1
        if count != 0:
            print("탈퇴완료.")
            accs[idx] = 0
            pws[idx] = 0
            cnt -= 1
        else:
            print("존재하지 않는 비밀번호입니다.")
    else:
        run = False
        print("종료")