프로그래머스/JAVA/로그인 성공?

문제설명

머쓱이는 프로그래머스에 로그인하려고 합니다. 머쓱이가 입력한 아이디와 패스워드가 담긴 배열 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입니다.

입출력 예

my_string		result
["meosseugi", "1234"]	[["rardss", "123"], ["yyoom", "1234"], ["meosseugi", "1234"]]	"login"
["programmer01", "15789"]	[["programmer02", "111111"], ["programmer00", "134"], ["programmer01", "1145"]]	"wrong pw"
["rabbit04", "98761"]	[["jaja11", "98761"], ["krong0313", "29440"], ["rabbit00", "111333"]]	"fail"

입출력 예 설명

입출력 예 #1

• db에 같은 정보의 계정이 있으므로 "login"을 return합니다.

입출력 예 #2

• db에 아이디는 같지만 패스워드가 다른 계정이 있으므로 "wrong pw"를 return합니다.

입출력 예 #3

• db에 아이디가 맞는 계정이 없으므로 "fail"을 return합니다.

풀이

HashMap에 id값인 db[i][0]을 key로 pw값인 db[i][1]을 value값으로 넣어준 뒤,
id_pw의 id_pw[0]을key값으로 입력하여 출력된 value값을 id_pw[1]과 비교한다.
두 값이 일치히면 "login"을 리턴, 두 값이 일치하지 않을 때 "wrong pw",
리턴값이 공백 문자열 일 때 "fail"을 리턴.
하려고 했으나 마지막에 일치하는 key값이 없을 때 리턴되는 값은 공백 문자열이 아니라
null이었다.

null값이 나오게 되면 예외가 발생하므로 try-catch문으로 처리 하여 nullPointException이 발생하면 
"fail"을 리턴하도록 했다.

import java.util.HashMap;
import java.util.Map;

class Solution {
	public String solution(String[] id_pw, String[][] db) {
		Map<String, String> memberTable = new HashMap<>();

		for (int i = 0; i < db.length; i++) {
			memberTable.put(db[i][0], db[i][1]);
		}
		try {
			if (memberTable.get(id_pw[0]).equals(id_pw[1])) {
				return "login";
			} else if (!memberTable.get(id_pw[0]).equals(id_pw[1])) {
				return "wrong pw";
			} else {
				return "fail";
			}
		} catch (Exception e) {
			return "fail";
		}
	}
}

참고

• 로그인 성공?