파이썬 문제: 3/29

프로그래머스 Level1 : 두 정수 사이의 합

두 정수 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의 대소관계는 정해져있지 않습니다.

입출력 예

a	b	return
3	5	12
3	3	3
5	3	12

해설

처음 내 풀이는 다음과 같다.

def solution(a, b):
    return sum([i for i in range(a, b+1)]) if a<=b else sum([i for i in range(b, a+1)])

If문을 안쓰고 min, max로 구현하는 방법이 있더라. range에 바로 sum이 적용되는지도 몰랐다.

def solution(arr):
    return sum(range(min(a,b),max(a,b)+1))

프로그래머스 Level1 : 문자열 내 마음대로 정렬하기

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

제한 조건

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

입출력 예

strings	n	return
[sun, bed, car]	1	[car, bed, sun]
[abce, abcd, cdx]	2	[abcd, abce, cdx]

입출력 예 설명

입출력 예 1

sun, bed, car의 1번째 인덱스 값은 각각 u, e, a 입니다. 이를 기준으로 strings를 정렬하면 [car, bed, sun] 입니다.

입출력 예 2

abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

해설

처음 내 풀이는 다음과 같다.

def solution(strings, n):
    strings.sort()
    strings.sort(key=lambda x : x[n])
    return strings

다음과 같이 간단하게도 된다.

def solution(strings, n):
	return sorted(sorted(strings), key=lambda x: x[n])

프로그래머스 Level1 : 문자열 내 p와 y의 개수

대문자와 소문자가 섞여있는 문자열 s가 주어집니다. s에 ‘p’의 개수와 ‘y’의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution를 완성하세요. ‘p’, ‘y’ 모두 하나도 없는 경우는 항상 True를 리턴합니다. 단, 개수를 비교할 때 대문자와 소문자는 구별하지 않습니다.

예를 들어 s가 pPoooyY면 true를 return하고 Pyy라면 false를 return합니다.

제한사항

• 문자열 s의 길이 : 50 이하의 자연수
• 문자열 s는 알파벳으로만 이루어져 있습니다.

---

입출력 예

s	answer
pPoooyY	true
Pyy	false

입출력 예 설명

입출력 예 #1

‘p’의 개수 2개, ‘y’의 개수 2개로 같으므로 true를 return 합니다.

입출력 예 #2

‘p’의 개수 1개, ‘y’의 개수 2개로 다르므로 false를 return 합니다.

해설

처음 내 풀이는 다음과 같다.

def solution(s):
    answer = 0
    for string in s.lower():
        if string == 'p':
            answer += 1
        elif string == 'y':
            answer -= 1
    return True if answer == 0 else False

count가 있는지 생각이 안났다. 다음처럼 간단히 풀 수 있다.

def solution(s):
	return s.lower().count('p') == s.lower().count('y')

프로그래머스 Level1 : 문자열 내림차순으로 배치하기

문자열 s에 나타나는 문자를 큰것부터 작은 순으로 정렬해 새로운 문자열을 리턴하는 함수, solution을 완성해주세요. s는 영문 대소문자로만 구성되어 있으며, 대문자는 소문자보다 작은 것으로 간주합니다.

제한 사항

• str은 길이 1 이상인 문자열입니다.

입출력 예

s	return
“Zbcdefg”	“gfedcbZ”

해설

처음 내 풀이는 다음과 같다. 정답과 같더라.

def solution(s):
    return ''.join(sorted(list(s), reverse=True))

프로그래머스 Level1 : 문자열 다루기 기본

문자열 s의 길이가 4 혹은 6이고, 숫자로만 구성돼있는지 확인해주는 함수, solution을 완성하세요. 예를 들어 s가 a234이면 False를 리턴하고 1234라면 True를 리턴하면 됩니다.

제한 사항

• s는 길이 1 이상, 길이 8 이하인 문자열입니다.

입출력 예

s	return
a234	false
1234	true

해설

처음 내 풀이는 다음과 같다. 정답과 같더라.

def solution(s):
    if len(s) == 4 or len(s) == 6:
        try:
            int(s)
            return True
        except: return False
    return False

isdigit과 in을 사용해서 간략히 표현할 수 있더라. in 활용도 고려해야 할듯

def solution(s):
	return s.isdigit() and len(s) in (4, 6)

프로그래머스 Level1 : 소수 찾기

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.

소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다.(1은 소수가 아닙니다.)

제한 조건

• n은 2이상 1000000이하의 자연수입니다.

입출력 예

n	result
10	4
5	3

입출력 예 설명

입출력 예 #1 1부터 10 사이의 소수는 [2,3,5,7] 4개가 존재하므로 4를 반환

입출력 예 #2 1부터 5 사이의 소수는 [2,3,5] 3개가 존재하므로 3를 반환

해설

소수는 찾는데 효율성에서 자꾸 실패해서, 웹 서핑으로 에라토스테네스의 체라는 방식을 알아냈다. 그리고 집합 연산을 위한 객체 set도 발견해서 이를 이용해 코드를 구성했다.

import math
def solution(n):
    answer = set([i  for i in range(2, n+1)])
    for i in range(2, int(math.sqrt(n)+1)):   
        if i in answer:
            answer -= set([j for j in range(i*2, n+1, i)])
    return len(answer)+1

1부터 $n$까지 수 중에 소수를 찾을때는 $\sqrt n$보다 작은 수들의 배수만 지워도 되므로 위와 같이 코드를 작성했다. 1레벨 풀이중에 오랜만에 막힌 문제인듯.

프로그래머스 Level1 : 수박수박수박수박수?

길이가 n이고, 수박수박수박수….와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 수박수박을 리턴하고 3이라면 수박수를 리턴하면 됩니다.

제한 조건

• n은 길이 10,000이하인 자연수입니다.

입출력 예

n	return
3	“수박수”
4	“수박수박”

해설

엄청 간단하게 풀었다.

def solution(n):
    return "수박"*(n//2)+"수"*(n%2)