[프로그래머스] 로그인 성공?

문제설명

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

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

class Solution {
    fun solution(id_pw: Array<String>, db: Array<Array<String>>): String {
        val id = id_pw[0]
        val pw = id_pw[1]

        for(index in 0 until db.size) {           
            if (db[index][0] == id) {                
                if(db[index][1] == pw) return "login"                
                return "wrong pw"
            }
        }               
        
        return "fail"
    }
}

풀이방법

// a. id_pw의 배열의 0번 원소와 1번 원소 모두 존재할 경우 "login"
// b. id만 존재할 경우 "wrongpw"
// c. 다 없는 경우 "fail"

// db길이 만큼 반복문을 돌며 it[0]에 값이 있다면 it[1]도 값을 체크 하며 다 동일할경우 a리턴
// db길이 만큼 반복문을 돌며 it[0]에 값이 있지만 it[1]에 값이 없다면 b리턴
// 가장 바깥 리턴에 fail