프로그래머스 - 2

프로그래머스 문제 풀이

프로그래머스

# Level1 #Python3

나누어 떨어지는 숫자 배열

문제 설명

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

제한사항

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

# mine
def solution(arr, divisor):
    answer = []
    for i in range(len(arr)):
        if arr[i] % divisor == 0:
            answer.append(arr[i])
    if len(answer) == 0:
        answer.append(-1)
    answer.sort()

    return answer

arr = [2, 36, 1, 3]
print(solution(arr, 1))

# someone else's 1
def solution(arr, divisor): return sorted([n for n in arr if n%divisor == 0]) or [-1]

# someone else's 2
def solution(arr, divisor):
    arr = [x for x in arr if x % divisor == 0];
    arr.sort();
    return arr if len(arr) != 0 else [-1];

def solution(arr, divisor): return sorted([n for n in arr if n%divisor == 0]) or [-1]

두 정수 사이의 합

문제 설명

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

제한 조건

• a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
• a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
• a와 b의 대소관계는 정해져있지 않습니다.

# mine
def solution(a, b):
    gab = abs(a-b)+1
    answer = 0
    if a < b:
        for i in range(gab):
            answer += a+i
    else:
        for i in range(gab):
            answer += b+i
    return answer

print(solution(3, 3))

# someone else's 1
def adder(a, b):
    if a > b: a, b = b, a
    return sum(range(a,b+1))

# someone else's 2
def adder(a, b):
    return (abs(a-b)+1)*(a+b)//2

if a > b: a, b = b, a

return (abs(a-b)+1)*(a+b)//2

문자열 내 마음대로 정렬하기

문제 설명

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

제한 조건

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

# mine
def solution(strings, n):
    return sorted(strings, key=lambda i : (i[n], i))

strings = ["abce", "abcd", "cdx"]
print(solution(strings, 2))

return sorted(strings, key=lambda i : (i[n], i))