자바 알고리즘 - 로그인 성공?

문제 설명

머쓱이는 프로그래머스에 로그인하려고 합니다. 머쓱이가 입력한 아이디와 패스워드가 담긴 배열 id_pw와 회원들의 정보가 담긴 2차원 배열 db가 주어질 때, 다음과 같이 로그인 성공, 실패에 따른 메시지를 return하도록 solution 함수를 완성해주세요.

아이디와 비밀번호가 모두 일치하는 회원정보가 있으면 "login"을 return합니다.
로그인이 실패했을 때 아이디가 일치하는 회원이 없다면 “fail”를, 아이디는 일치하지만 비밀번호가 일치하는 회원이 없다면 “wrong pw”를 return 합니다.

 

제한 사항

• 회원들의 아이디는 문자열입니다.
• 회원들의 아이디는 알파벳 소문자와 숫자로만 이루어져 있습니다.
• 회원들의 패스워드는 숫자로 구성된 문자열입니다.
• 회원들의 비밀번호는 같을 수 있지만 아이디는 같을 수 없습니다.
• id\_pw의 길이는 2입니다.
• id\_pw와 db의 원소는 [아이디, 패스워드] 형태입니다.
• 1 ≤ 아이디의 길이 ≤ 15
• 1 ≤ 비밀번호의 길이 ≤ 6
• 1 ≤ db의 길이 ≤ 10
• db의 원소의 길이는 2입니다.

 

답안

class Solution {
    public String solution(String[] id_pw, String[][] db) {
        String answer = "";
        for(int i=0; i<db.length; i++) {
            if(id_pw[0].equals(db[i][0])) {
                if(id_pw[1].equals(db[i][1])) {
                    answer = "login";
                    break;
                }
                else {
                    answer = "wrong pw";
                }
            }
        }

        return answer.length() > 0? answer : "fail";
    }
}

먼저 for문을 통해 DB 배열의 길이를 돌면서 입력한 id_pw의 일치, 불일치 여부를 판단한다. 이 때 체크해야 하는 부분은 결과값에 우선순위가 있다는 점이다.

ID와 PW가 모두 일치하면 login, ID는 일치하고 PW가 일치하지 않으면 wrong pw, 둘다 불일치시 fail을 리턴해야하는데 이전 DB 원소에서 wrong pw가 담겼다가 다음 원소 체크시 모두 불일치해서 fail이 담기면 에러가 뜬다는 점이다.

 

그래서 fail은 return 직전 login도 아니고 wrong pw도 아닐 때 넘겨주는 식으로 처리했다.