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

# [level 0] 로그인 성공? - 120883 [문제 링크]

 

### 성능 요약

       메모리: 10.2 MB, 시간: 0.00 ms

### 구분

       코딩테스트 연습 > 코딩테스트 입문

### 채점결과
       정확성: 100.0
       합계: 100.0 / 100.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입니다.

---

입출력 예

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"

---

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

입출력 예 #3

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

---

 

나의 답안

def solution(id_pw, db):
    count = [0,0,0]
    for i in range(len(db)):
        if db[i][0]==id_pw[0]:
            if db[i][1]==id_pw[1]:
                count[0] += 1
            else:
                count[1] +=1
        else:
            count[2] += 1
    
    if count[0]==1:
        return "login"
    elif count[1]==1:
        return "wrong pw"
    else: 
        return "fail"

---

다른 사람 답안

def solution(id_pw, db):
    if db_pw := dict(db).get(id_pw[0]):
        return "login" if db_pw == id_pw[1] else "wrong pw"
    return "fail"

내 코드와 비교 했을 때,,,, 

코드 길이 부터 차이가 나는 코드 ㅠㅜㅜ

언제쯤! 이렇게 짤 수 있을까?

 

:= 표현은 파이썬 3.8의 새로운 기능이다!

자세한 설명은 아래 docs 참고!!

https://docs.python.org/ko/3/whatsnew/3.8.html

---

다른 사람 답안(2)

def solution(id_pw, db):
    dic = dict(db)
    if dic.get(id_pw[0],-1) == id_pw[1]:
        return "login"
    elif dic.get(id_pw[0],-1) == -1:
        return "fail"
    else:
        return "wrong pw"

이 코드를 풀이해보도록 하겠다 

 

db 값을 dictionary로 바꾼다.

바꾼 값을 찍어서 확인해보면? 아래 사진과 같다!

get은 보면 아래와 같이 쓰인다

• get : 원하는 키key에 대한 값value을 할당할 수 있는 방법!
• 첫번째 인자 : 찾고자 하는 키key
• 두번째 인자 : 첫 번째 인자에 값이 없다면 넣고자 하는 값value

프린트 해보면 위와 같은 결과가 나온다!

dictionary dic에서 id_pw[0]에 해당하는 meosseugi 키는 값을 가지고 있으므로 가지고 있는 값인 1234를 반환한다.

 

이를 if 문으로 활용하여 확인해보면 로그인 성공? 문제를 해결할 수 있다.

 

먼저, 첫 번째 if 문

    if dic.get(id_pw[0],-1) == id_pw[1]:

          return "login"

dic 에 있는 pw와 id_pw의 pw를 비교하여 같으면 "login"을 반환한다!!

 

두 번째 elif 문

    elif dic.get(id_pw[0],-1) == -1:

        return "fail"

 

dic의 pw 값이 -1 이면 에초에 meosseugi라는 키에 값이 없어서 저장된 값이므로 "fail"을 반환한다!

 

 

마지막으로 else 문

    else:

        return "wrong pw"

키 값에 저장되어 있던 pw와 id_pw의 pw값이 다르면 "wrong pw"를 반환한다!