프로그래머스 입문 Day 23 파이썬

프로그래머스

특이한 정렬

문제 설명

정수 n을 기준으로 n과 가까운 수부터 정렬하려고 합니다.

이때 n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치합니다.

정수가 담긴 배열 numlist와 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ n ≤ 10,000
• 1 ≤ numlist의 원소 ≤ 10,000
• 1 ≤ numlist의 길이 ≤ 100
• numlist는 중복된 원소를 갖지 않습니다.

입출력 예

numlist	n	result
[1, 2, 3, 4, 5, 6]	4	[4, 5, 3, 6, 2, 1]
[36, 40, 20, 47, 10, 6, 7000, 10000]	30	[36, 40, 20, 47, 10, 6, 7000, 10000]

입출력 예 설명

입출력 예 #1

• 4에서 가까운 순으로 [4, 5, 3, 6, 2, 1]을 return합니다.
• 3과 5는 거리가 같으므로 더 큰 5가 앞에 와야 합니다.
• 2와 6은 거리가 같으므로 더 큰 6이 앞에 와야 합니다.

입출력 예 #2

• 30에서 가까운 순으로 [36, 40, 20, 47, 10, 6, 7000, 10000]을 return합니다.
• 20과 40은 거리가 같으므로 더 큰 40이 앞에 와야 합니다.

solution.py

def solution(numlist, n):
    return sorted(numlist, key=lambda x: (abs(x - n), -x))

---

등수 매기기

문제 설명

영어 점수와 수학 점수의 평균 점수를 기준으로 학생들의 등수를 매기려고 합니다.

영어 점수와 수학 점수를 담은 2차원 정수 배열 score가 주어질 때, 영어 점수와 수학 점수의 평균을 기준으로 매긴 등수를 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 0 ≤ score[0], score[1] ≤ 100
• 1 ≤ score의 길이 ≤ 10
• score의 원소 길이는 2입니다.
• score는 중복된 원소를 갖지 않습니다.

입출력 예

score	result
[[80, 70], [90, 50], [40, 70], [50, 80]]	[1, 2, 4, 3]
[[80, 70], [70, 80], [30, 50], [90, 100], [100, 90], [100, 100], [10, 30]]	[4, 4, 6, 2, 2, 1, 7]

입출력 예 설명

입출력 예 #1

• 평균은 각각 75, 70, 55, 65 이므로 등수를 매겨 [1, 2, 4, 3]을 return합니다.

입출력 예 #2

• 평균은 각각 75, 75, 40, 95, 95, 100, 20 이므로 [4, 4, 6, 2, 2, 1, 7] 을 return합니다.
• 공동 2등이 두 명, 공동 4등이 2명 이므로 3등과 5등은 없습니다.

solution.py

def solution(score):
    averages = [(i, (s[0] + s[1]) / 2) for i, s in enumerate(score)]

    sorted_avg = sorted(averages, key=lambda x: x[1], reverse=True)

    ranks = [0] * len(score)
    rank = 1
    for i in range(len(score)):
        if i > 0 and sorted_avg[i][1] == sorted_avg[i - 1][1]:
            ranks[sorted_avg[i][0]] = ranks[sorted_avg[i - 1][0]]
        else:
            ranks[sorted_avg[i][0]] = rank
        rank += 1

    return ranks

---

옹알이 (1)

문제 설명

머쓱이는 태어난 지 6개월 된 조카를 돌보고 있습니다.

조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음을 최대 한 번씩 사용해 조합한(이어 붙인) 발음밖에 하지 못합니다.

문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ babbling의 길이 ≤ 100
• 1 ≤ babbling[i]의 길이 ≤ 15
• babbling의 각 문자열에서 "aya", "ye", "woo", "ma"는 각각 최대 한 번씩만 등장합니다.
  • 즉, 각 문자열의 가능한 모든 부분 문자열 중에서 "aya", "ye", "woo", "ma"가 한 번씩만 등장합니다.
• 문자열은 알파벳 소문자로만 이루어져 있습니다.

입출력 예

babbling	result
["aya", "yee", "u", "maa", "wyeoo"]	1
["ayaye", "uuuma", "ye", "yemawoo", "ayaa"]	3

입출력 예 설명

입출력 예 #1

• ["aya", "yee", "u", "maa", "wyeoo"]에서 발음할 수 있는 것은 "aya"뿐입니다. 따라서 1을 return합니다.

입출력 예 #2

• ["ayaye", "uuuma", "ye", "yemawoo", "ayaa"]에서 발음할 수 있는 것은 "aya" + "ye" = "ayaye", "ye", "ye" + "ma" + "woo" = "yemawoo"로 3개입니다. 따라서 3을 return합니다.

---

유의사항

• 네 가지를 붙여 만들 수 있는 발음 이외에는 어떤 발음도 할 수 없는 것으로 규정합니다. 예를 들어 "woowo"는 "woo"는 발음할 수 있지만 "wo"를 발음할 수 없기 때문에 할 수 없는 발음입니다.

문제 분석

1 발음 가능한 단어 리스트에 저장
2 발음 가능한 단어를 세기 위한 변수 count 0으로 초기화
3 babbling 배열의 각 문자열 i에 대해 반복
  a answer 리스트의 각 단어 j에 대해
    문자열 i에서 j를 한 번 공백으로 대체
  b 문자열 i의 앞뒤 공백을 제거하고 빈 문자열이 되었는지 확인
    빈 문자열이면 count 1 증가
4 최종 count 반환

solution.py

def solution(babbling):
    answer = ["aya", "ye", "woo", "ma"]
    count = 0

    for i in babbling:
        for j in answer:
            i = i.replace(j, " ", 1)
        if i.strip() == "":
            count += 1

    return count

---

로그인 성공?

문제 설명

머쓱이는 프로그래머스에 로그인하려고 합니다.

머쓱이가 입력한 아이디와 패스워드가 담긴 배열 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합니다.

solution.py

def solution(id_pw, db):
    for i in db:
        id, pw = i
        if id == id_pw[0]:
            if pw == id_pw[1]:
                return "login"
            else:
                return "wrong pw"
    return "fail"