최초 코드 - 시간 초과

def find(n):
    count = 0
    for i in range(1, int(n ** .5) + 1):
        if n % i == 0:
            count += 1
    return 1 if count == 1 else 0
            
def solution(n):
    answer = 0
    for i in range(2, n + 1):
        answer += find(i)
    return answer

 

최종 코드 

def solution(n):
    num = set(range(2, n + 1))

    for i in range(2, n + 1):
        if i in num:
            num -= set(range(2 * i, n + 1, i))
    return len(num)

나의 코드

def solution(num):
    return "Even" if num % 2 == 0 else "Odd"

 

 

눈여겨 볼만한 코드

def evenOrOdd(num):
    return ["Even", "Odd"][num & 1]

나의 코드

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        tmp = []
        for j in range(len(arr1[i])):
            tmp.append(arr1[i][j] + arr2[i][j])
        answer.append(tmp)
    return answer

 

참고할만한 코드

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

나의 코드

def solution(n):
    answer = [int(i) for i in reversed(str(n))]
    return answer

+ Recent posts