코딩테스트 스터디 4주차

 

 

1. 2016년

 

def solution(a, b):
    date = ['FRI','SAT','SUN','MON','TUE','WED','THU']
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    cnt = 0
    
    for i in range(a-1):
        cnt += month[i]
    ans = (cnt + b) % 7
    
    return date[ans-1]

 

 

 

2. 문자열 다루기 기본

 

def solution(s):
    if len(s) == 4 and s.isdigit():
        return True
    elif len(s) == 6 and s.isdigit():
        return True
    else:
        return False

 

 

 

3. 문자열 내림차순으로 배치하기 

 

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

 

 

 

4. 약수의 합

 

def solution(n):
    arr = []
    
    for i in range(1,n+1):
        if n % i == 0:
            arr.append(i)
    return sum(arr)

 

 

 

5. 서울에서 김서방 찾기 

 

def solution(seoul):
        answer = "김서방은 " + str(seoul.index("Kim")) + "에 있다"
        return answer

 

 

 

6. 수박수박수박수박수박수?

 

def solution(n):
    ans = ''
    for i in range(n):
        if i % 2 == 0:
            ans += '수'
        else:
            ans += '박'
    return ans

 

 

 

7. 문자열을 정수로 바꾸기 

 

def solution(s):
    if s[0] == "-":
        k = int(s[1:]) * -1
        return k
    else:
        k = int(s[0:])
        return k

 

 

 

 

8. 소수 찾기 

 

import itertools
import math

def solution(numbers):
    arr = []
    imp = []
    ans = 0
    
    for i in range(1,len(numbers)+1):
        num = list(map(''.join, itertools.permutations(numbers,i)))
        arr.append(num)
    
    arr = sum(arr,[])
    arr = list(set(arr))
    
    for i in arr:
        imp.append(int(i))
    imp = list(set(imp))
    imp.sort()
    
    for k in imp:
        cnt = 0
        sqrt = int(math.sqrt(k))
        if k != 0 and k != 1:
            for i in range(2,sqrt+1):
                    if k % i == 0:
                        cnt += 1
            if cnt == 0:
                ans += 1
    return ans

분석 > 해당 수를 제곱근 이하의 수로 하나씩 나누어 봤을 때, 나누어 떨어지는 수가 0개라면 이 수가 소수임을 알 수 있다는 것이 중요한 포인트인 문제였다. 

 

 

 

9. 시저 암호 

 

def solution(s, n):
    ans = ''
    
    for i in s:
        if i.isupper():
            if ord(i) + n > ord('Z'):
                ans += chr(ord(i) + n - ord('Z') + ord('A') - 1)
            else:
                ans += chr(ord(i) + n)
        elif i.islower():
            if ord(i) + n > ord('z'):
                ans += chr(ord(i) + n - ord('z') + ord('a') - 1)
            else:
                ans += chr(ord(i) + n)
        else:
            ans += ' '
                
    return ans

 

 

 

10. 최소 직사각형 

 

def solution(sizes):
    w = []
    h = []
    for i in range(len(sizes)):
        if sizes[i][0] > sizes[i][1]:
            w.append(sizes[i][0])
            h.append(sizes[i][1])
        else:
            w.append(sizes[i][1])
            h.append(sizes[i][0])
    return max(w) * max(h)