[코딩테스트 입문]Day23-4

로그인 성공?

문제 설명

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

---

문제 풀이

def solution(id_pw, db):
    for id, pw  in db:
        if id == id_pw[0]:
            if pw == id_pw[1]:
                return "login"
            else:
                return "wrong pw"
    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"

더 알아보기 :=

바다코끼리 연산자

바다코끼리 연산자로 불리는 Walrus Operator 는 Assignment Expressions라는 이름으로 정의되었습니다.

바다코끼리 연산자의 목적은 표현식에 이름을 부여하고 재사용할 수 있게 하는 것입니다.

numlist = [1, 2, 3, 4, 5]
n = len(numlist)
if n > 5:
    print(f'List is too long ({n} elements, expexted <= 5)')

len을 여러 번 호출하는 상황을 방지하기 위해, len의 출력값을 저장하는 변수 n을 사용하는데 라인 할 줄이 더 필요했습니다.

바다코끼리 연산자를 사용하면 아래와 같이 n을 조건문 안에서 선언하고 값을 대입할 수 있습니다.

numlist = [1, 2, 3, 4]
if (n := len(a)) > 5:
    print(f'List is too long ({n} elements, expected <= 5)')

조건문 이외에도 반복문에서는

while True:
    chunk = file.read(128)
    if chunk:
        break
    process(chunk)

위와 같은 코드를 아래처럼 변경할 수 있습니다.

while chunk := file.read(128)
    process(chunk)

출처: Python 3.8 기여운 바다코끼리를 드리겟슴미다 :=

1. 개념

표현식의 결과를 변수에 할당하고, 동시에 반환합니다.

즉, 변수 = 표현식을 하고, return 변수 기능을 합니다.