로그인 성공?

/* 로그인 성공?

 * 아이디와 패스워드가 담긴 배열 id_pw와 회원들의 정보가 담긴 2차원 배열 db가 주어질 때,

 * 다음과 같이 로그인 성공, 실패에 따른 메시지를 return

 *

 * id_pw                        db                                                                                  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"

 */

 

import java.util.*;
public class programmer_0_56 {
    static String[] a1 = {"meosseugi", "1234"};     static String[][] b1 = {{"rardss", "123"},{"yyoom", "1234"},{"meosseugi", "1234"}};
    static String[] a2 = {"programmer01", "15789"}; static String[][] b2 = {{"programmer02", "111111"},{"programmer00", "134"},{"programmer01", "1145"}};
    static String[] a3 = {"rabbit04", "98761"};     static String[][] b3 = {{"jaja11", "98761"},{"krong0313", "29440"},{"rabbit00", "111333"}};
    public String solution(String[] id_pw, String[][] db) {
        String answer = "";
        Map<String, String> mp = new HashMap<String, String>();
        for(String[] n : db){
            mp.put(n[0], n[1]);
        }
        if(mp.containsKey(id_pw[0])){
            if(mp.get(id_pw[0]).equals(id_pw[1])){
                return "login";
            }
            else    return "wrong pw";
        }
        else    return "fail";
        /*
        for(int i = 0; i < db.length; i++){
            if(db[i][0].equals(id_pw[0])){
                if(db[i][1].equals(id_pw[1])){
                    return "login";
                }
                return "wrong pw";
            }
        }
        return "fail";
         */
    }
    public static void main(String args[]){
        programmer_0_56 t = new programmer_0_56();
 
        System.out.println("---------------------------------------");
        System.out.println("result = " + t.solution(a1,b1));
        System.out.println("---------------------------------------");
        System.out.println("result2 = " + t.solution(a2,b2));
        System.out.println("---------------------------------------");
        System.out.println("result3 = " + t.solution(a3,b3));
        System.out.println("---------------------------------------");
    }
}