[프로그래머스 Python] 코딩 기초 트레이닝 Day 20 - 함수(메서드)

---

1. 배열의 길이를 2의 거듭제곱으로 만들기

정수 배열 arr이 매개변수로 주어집니다. arr의 길이가 2의 정수 거듭제곱이 되도록 arr 뒤에 정수 0을 추가하려고 합니다. arr에 최소한의 개수로 0을 추가한 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
• 1 ≤ arr의 원소 ≤ 1,000

코드 작성

def solution(arr):
    if len(arr) == 1:
        return arr
    # 배열의 길이가 2 이상인 경우
    l = min([2**i for i in range(1, len(arr)//2+1) if 2**i >= len(arr)])
    for _ in range(l-len(arr)):
        arr.append(0)
    return arr

2. 배열 비교하기

이 문제에서 두 정수 배열의 대소관계를 다음과 같이 정의합니다.

• 두 배열의 길이가 다르다면, 배열의 길이가 긴 쪽이 더 큽니다.
• 배열의 길이가 같다면 각 배열에 있는 모든 원소의 합을 비교하여 다르다면 더 큰 쪽이 크고, 같다면 같습니다.
  두 정수 배열 arr1과 arr2가 주어질 때, 위에서 정의한 배열의 대소관계에 대하여 arr2가 크다면 -1, arr1이 크다면 1, 두 배열이 같다면 0을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr1의 길이 ≤ 100
• 1 ≤ arr2의 길이 ≤ 100
• 1 ≤ arr1의 원소 ≤ 100
• 1 ≤ arr2의 원소 ≤ 100
• 문제에서 정의한 배열의 대소관계가 일반적인 프로그래밍 언어에서 정의된 배열의 대소관계와 다를 수 있는 점에 유의해주세요.

코드 작성

def solution(arr1, arr2):
    if len(arr1) != len(arr2):
        return 1 if len(arr1) > len(arr2) else -1
    if sum(arr1) == sum(arr2):
        return 0
    return 1 if sum(arr1) > sum(arr2) else -1

3. 문자열 묶기

문자열 배열 strArr이 주어집니다. strArr의 원소들을 길이가 같은 문자열들끼리 그룹으로 묶었을 때 가장 개수가 많은 그룹의 크기를 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ strArr의 길이 ≤ 100,000
  • 1 ≤ strArr의 원소의 길이 ≤ 30
  • strArr의 원소들은 알파벳 소문자로 이루어진 문자열입니다.

코드 작성

def solution(strArr):
    dic = {len(s):0 for s in strArr}
    for s in strArr:
        dic[len(s)] += 1
    return max(dic.values())

코드 작성(리스트 ver)

def solution(strArr):
    arr = [0] * 31
    for s in strArr:
        arr[len(s)] += 1
    return max(arr)

제한사항을 보면 원소의 길이는 1 ~ 30의 값을 가진다. 이 점을 이용해서 arr에 원소의 길이 범위를 0으로 넣었다. 그 다음 반복문을 통해 원소의 길이를 arr의 인덱스로 위치를 지정해 1을 더했다. 배열(arr) 내에 문자열의 길이에 맞는 그룹을 만들었고 가장 많이 등장한 길이값을 반환하기 위해 max 함수를 사용했다.

4. 배열의 길이에 따라 다른 연산하기

정수 배열 arr과 정수 n이 매개변수로 주어집니다. arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을, arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위치에 n을 더한 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
• 1 ≤ arr의 원소 ≤ 1,000
• 1 ≤ n ≤ 1,000

코드 작성

def solution(arr, n):
    odd = [j+n if i%2==0 else j for i, j in enumerate(arr)]
    even = [j+n if i%2!=0 else j for i, j in enumerate(arr)]
    return even if len(arr) % 2 == 0 else odd

5. 뒤에서 5등까지

정수로 이루어진 리스트 num_list가 주어집니다. num_list에서 가장 작은 5개의 수를 오름차순으로 담은 리스트를 return하도록 solution 함수를 완성해주세요.

제한사항

• 6 ≤ num_list의 길이 ≤ 30
• 1 ≤ num_list의 원소 ≤ 100

코드 작성

def solution(num_list):
    return sorted(num_list)[:5]

📍 문제 출처

• 배열의 길이를 2의 거듭제곱으로 만들기
• 배열 비교하기
• 문자열 묶기
• 배열의 길이에 따라 다른 연산하기
• 뒤에서 5등까지

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.