통계러버

문제링크

• 타겟 넘버

문제 설명

n개의 음이 아닌 정수들이 있습니다. 이 정수들을 순서를 바꾸지 않고 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다.

-1+1+1+1+1 = 3
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3

사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.

제한 사항

• 주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
• 각 숫자는 1 이상 50 이하인 자연수입니다.
• 타겟 넘버는 1 이상 1000 이하인 자연수입니다.

입출력 예

입출력 예 설명

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

+4+1-2+1 = 4
+4-1+2-1 = 4

• 총 2가지 방법이 있으므로, 2를 return 합니다.

코드

DFS 풀이

def dfs(i, numbers, target, tot):
    global cnt
    n = len(numbers)
    if i== n:
        if tot == target:
            cnt += 1
            
    else:
        dfs(i+1, numbers, target, tot + numbers[i])
        dfs(i+1, numbers, target, tot + numbers[i]*(-1))
            
def solution(numbers, target):
    global cnt
    cnt = 0
    dfs(0,numbers, target, 0)
    return cnt

중복순열

• 중복조합은 from itertools import combinations_with_replacement를 쓰는 반면 중복순열은 from itertools import product를 쓴다
• combinations_with_replacement(뽑을 곳, 반복횟수)을 써주기만 하면 되는 반면 product(뽑을 곳, repeat = 반복횟수)임 만약 product에서 repeat을 안쓰면 오류가 뜸 

ex)

from itertools import combinations_with_replacement

li = [1,2]
for com in combinations_with_replacement(li, 2):
    print(com)

>>>(1, 1)
   (1, 2)
   (2, 2)

from itertools import product

li = [1,2]
for pro in product(li,2):
    print(pro)

>>> TypeError

from itertools import product

li = [1,2]
for pro in product(li,repeat = 2):
    print(pro)

>>> (1, 1)
    (1, 2)
    (2, 1)
    (2, 2)

중복순열을 이용한 풀이

def solution(numbers, target):
    from itertools import product
    w = [-1,1]
    n = len(numbers)
    cnt = 0
    for per in  product(w, repeat = n):
        sol = 0
        for i in range(n):
            sol += per[i]*numbers[i]
        if sol == target:
            cnt +=1
    return cnt

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제링크

• JadenCase 문자열 만들기

문제 설명

JadenCase란 모든 단어의 첫 문자가 대문자이고, 그 외의 알파벳은 소문자인 문자열입니다. 단, 첫 문자가 알파벳이 아닐 때에는 이어지는 알파벳은 소문자로 쓰면 됩니다. (첫 번째 입출력 예 참고)
문자열 s가 주어졌을 때, s를 JadenCase로 바꾼 문자열을 리턴하는 함수, solution을 완성해주세요.

제한조건

• s는 길이 1 이상 200 이하인 문자열입니다.
• s는 알파벳과 숫자, 공백문자(" ")로 이루어져 있습니다.
  • 숫자는 단어의 첫 문자로만 나옵니다.
  • 숫자로만 이루어진 단어는 없습니다.
  • 공백문자가 연속해서 나올 수 있습니다.

입출력 예

참고

https://kmsstat01.tistory.com/entry/stringsplit

‘문자열’.upper() 혹은 ‘문자열’.lower()는 ‘숫자’에 적용될 경우 그대로 나옴. 따라서 위 문제의 조건을 따로 고려하지 않아도 됨. 주의! upper 혹은 lower 뒤에 ()를 안하면 값이 생성되지 않으므로 꼭 ()를 뒤에 붙이자
ex) '1b'.upper() >>> ‘1B’ , '1B'.upper() >>> ‘1b’

코드

def solution(s):
    sol = []
    li = s.split(' ')  # 문자열을 split(' ')하면 공백을 기준으로 잘라서 리스트로 만든다. 이때 공백이 연속으로 있으면 공백의 수 -1 개의 ''이 리스트의 원소가 됨
    for k in li:
        
        if  k == '':    # 공백이 두개 이상 연속으로 있을 경우 리스트의 원소중에 ''이 있을 것. 이를 고려해야함
            sol.append(k)
        else:
            sol.append(k[0].upper() +k[1:].lower())   # '문자열'.upper()는 '숫자'.upper()을 할 경우에도 에러 없이 그대로 출력함 
    return(' '.join(sol))

아래의 코드는 하드코드이므로 비추천함....

def solution(s):
    sol = []
    li = s.split(' ')  # 문자열을 split(' ')하면 공백을 기준으로 잘라서 리스트로 만든다. 이때 공백이 연속으로 있으면 공백의 수 -1 개의 ''이 리스트의 원소가 됨
    for i in li:
        if i == '':     # 공백이 두개 이상 연속으로 있을 경우 리스트의 원소중에 ''이 있을 것. 이를 고려해야함
            w = ''
        else:
            w = ''    
            for j in range(len(i)):
                if i[0]  in 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ' :
                    if j ==0:
                        w = w + i[0].upper()
                    else:    
                        w = w + i[j].lower()
                else:
                    if j == 0:
                        w = w + i[0]
                    else:    
                        w = w + i[j].lower()
        sol.append(w)

    return(' '.join(sol))

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

• N개의 최소공배수

문제 설명

두 수의 최소공배수(Least Common Multiple)란 입력된 두 수의 배수 중 공통이 되는 가장 작은 숫자를 의미합니다. 예를 들어 2와 7의 최소공배수는 14가 됩니다. 정의를 확장해서, n개의 수의 최소공배수는 n 개의 수들의 배수 중 공통이 되는 가장 작은 숫자가 됩니다. n개의 숫자를 담은 배열 arr이 입력되었을 때 이 수들의 최소공배수를 반환하는 함수, solution을 완성해 주세요.

제한사항

• arr은 길이 1이상, 15이하인 배열입니다.
• arr의 원소는 100 이하인 자연수입니다.

입출력 예

풀이

• 최대공약수는 math 라이브러리에서 gcd함수를 사용하면 됨
• 최소공배수는 두값을 곱하고 최대 공약수로 나누었을 때의 값임. 주의! math 라이브러이에서 gcd는 정수값만 읽음 하지만 나눗셈을 하면 float이 되므로 int값으로 바꿔줘야함
• 3개의 수에 대한 최대공약수는 두개에 대한 최대공약수와 나머지 수 사이에 다시 최대공약수를 구하면 됨
• 3개의 수에 대한 최소공배수는 두개에 대한 최소공배수와 나머지 수 사이에 다시 최소공배수를 구하면 됨

코드

sol1)  최대 공약수를 함수로 만들어서 푸는 풀이

# N개의 최소공배수

def gcd(x,y):
    while x!= 0:
        y , x = x , y%x   # 주의!     y = x
                          #           x = y%x  이렇게 두줄로 쓰면 오답이 나옴. 그 이유는 위에서 할당된 y가 밑에서 적용되기 때문에
    return y    

def lcm(x,y):
    return int(x*y/gcd(x,y))

def solution(arr):
    m = arr.pop()
    n = arr.pop()
    
    l_c_m = lcm(m,n)
    while arr:
        l_c_m = lcm(l_c_m, arr.pop())
    return l_c_m

sol2)  math함수를 이용한 풀이

from math import gcd


def solution(num):      
    answer = num[0]
    for n in num:
        answer = int(n * answer / gcd(n, answer))

    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

• 피보나치 수 문제 

문제 설명

피보나치 수는 F(0) = 0, F(1) = 1일 때, 1 이상의 n에 대하여 F(n) = F(n-1) + F(n-2) 가 적용되는 수 입니다.

예를 들어

F(2) = F(0) + F(1) = 0 + 1 = 1
F(3) = F(1) + F(2) = 1 + 1 = 2
F(4) = F(2) + F(3) = 1 + 2 = 3
F(5) = F(3) + F(4) = 2 + 3 = 5

와 같이 이어집니다.

2 이상의 n이 입력되었을 때, n번째 피보나치 수를 1234567으로 나눈 나머지를 리턴하는 함수, solution을 완성해 주세요.

제한 사항

• n은 1이상, 100000이하인 자연수입니다.
  
  

입출력 예

입출력 예 설명

피보나치 수는 0번째부터 0, 1, 1, 2, 3, 5, ... 와 같이 이어집니다.

코드

• 시간초과한 풀이

재귀함수는 시간이 매우 오래 걸림

# 피보나치 수 시간 초과 풀이

def Fibonacci(n):
    if n == 0:
        return 0
    elif n == 1:
        return 1
    else :
        return Fibonacci(n-1) + Fibonacci(n-2)
    
def solution(n):
    return Fibonacci(n)%1234567

• test 통과한 풀이 - 리스트를 이용하여 점화식 만들기

리스트 끝부분에 append로 원소를 추가하는 것은 O(1)이므로 O(N). N은 100000이 최대 이므로 1초안에 출력 가능함

# 피보나치 수 성공한 풀이
def solution(n):
    sol = [0,1]
    for i in range(2, n+1):
        sol.append(sol[i-2] + sol[i-1])
    return sol[-1]%1234567

• test 통과한 풀이 - 딕셔너리를 이용해서 점화식 만들기

def solution(n):
    answer ={}
    answer[0] = 0
    answer[1] = 1
    for i in range(2, n+1):
        answer[i] = answer[i-1] + answer[i-2]
    return answer[n]%1234567

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

•  가장 큰 수 문제

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

제한사항

• numbers의 길이는 1 이상 100,000 이하입니다.
• numbers의 원소는 0 이상 1,000 이하입니다.
• 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

입출력 예

문제풀이

• int형의 list를 map을 사용하여 string으로 치환한 뒤, list로 변환한다.
• 변환된 num을 sort()를 사용하여 key 조건에 맞게 정렬한다. lambda x : x * 4은 num 인자 각각의 문자열을 4번 반복한다는 뜻이다.
  • (x * 4)[:4]을 하는 이유?
    • num의 인수값이 1000 이하이므로 4자리수로 맞춘 뒤, 비교하겠다는 뜻.
• 문자열 비교는 ASCII 값으로 치환되어 정렬된다. 먼저 첫번째 인덱스에 대해 비교하고, 그 때 승부가 안나면 두번째 인덱스에 대해 비교하고, 그 때도 승부가 안나면 세번째 인덱스를 비교한다(승부가 날 때까지). [’6’, ‘10’, ‘2’]일 때를 예를 들면 x*4이므로 ‘6666’, ‘10101010’, ‘2222’의 첫번째 인덱스 값으로 비교한다.
  • 6 = 86, 1 = 81, 2 = 82 이므로 6 > 2 > 1순으로 크다.(해당 예시는 해당 안되지만 만일 첫번째 인덱스에서 승부가 안나면 다음 인덱스로 가서 승부를 한다…이를 승부가 날 때까지 반복한다)
  • sort()의 기본 정렬 기준은 오름차순이다. reverse = True 전의 sort된 결과값은 10, 2, 6이다.
  • 이를 reverse = True를 통해 내림차순 해주면 6,2,10이 된다. 이것을 ‘‘.join(num)을 통해 문자열을 합쳐주면 된다.
  • int로 변환한 뒤, 또 str로 변환해주는 이유?
    • 모든 값이 0일 때(즉, ‘000’일때 이를 0으로 바꿔주기 위해서) int로 변환한 뒤, 다시 str로 변환한다.
       ex) int('000') >>> 0 

코드

def solution(numbers):
    numbers = list(map(str, numbers)) # O(n)
    numbers.sort(key = lambda x: (x*4)[:4], reverse=True) #O(nlogn)
    return str(int(''.join(numbers)))  # O(n)

sol2

def solution(numbers):
    numbers = list(map(str, numbers))  # O(n)
    numbers.sort(key = lambda x: (x*4)[:4], reverse=True)  #O(nlogn)
    if numbers[0] == '0': # 가장 큰수가 '0'이라는 것은 모든 수가 '0'인 것
        answer = '0'
    else:
        answer = ''.join(numbers)   # O(n)
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges