[프로그래머스(Java)] 로그인 성공? / Map, map.containsKey()

 

Level. 0

 

문제

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

 

풀이

import java.util.*;

class Solution {
    public String solution(String[] id_pw, String[][] db) {
        Map<String, String> map = new HashMap<>();
        for(int i=0; i<db.length; i++){
            map.put(db[i][0], db[i][1]);
        }
        
        if(!map.containsKey(id_pw[0]))
            return "fail";
        if(!map.get(id_pw[0]).equals(id_pw[1]))
            return "wrong pw";
        return "login";
    }
}

 

 

1. id를 key로, password를 value로 저장하기 위해 map 자료구조를 사용하였다.
2. containsKey는 자료구조에 파라미터로 입력한 Key가 없다면 false를 반환한다. 이 땐 아이디가 일치하는 회원이 없다는 뜻이므로 "fail"을 출력한다.
3. key는 있으나 value가 다르다면 비밀번호가 틀리다는 뜻이므로  "wrong pw" 를 출력한다. 
4. 위 조건을 통과했다면 id 와 pw가 일치했다는 뜻이므로 "login"을 출력한다. 

 

---

다른 풀이

class Solution {
    public String solution(String[] id_pw, String[][] db) {
        String answer = "";

        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";
    }
}

---

https://school.programmers.co.kr/learn/courses/30/lessons/120883

프로그래머스