코딩테스트

24/11/27 Python

helena1129 2024. 11. 28. 15:22

1. 다음 큰 숫자

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

자연수 n이 주어졌을 때, n의 다음 큰 숫자는 다음과 같이 정의 합니다.
조건 1. n의 다음 큰 숫자는 n보다 큰 자연수 입니다.조건 2. n의 다음 큰 숫자와 n은 2진수로 변환했을 때 1의 갯수가 같습니다.조건 3. n의 다음 큰 숫자는 조건 1, 2를 만족하는 수 중 가장 작은 수 입니다.
예를 들어서 78(1001110)의 다음 큰 숫자는 83(1010011)입니다.
자연수 n이 매개변수로 주어질 때, n의 다음 큰 숫자를 return 하는 solution 함수를 완성해주세요.
def solution(n):
    answer = 0
    i = n

    while True:
      i += 1
      if bin(i)[2:].count('1') == bin(n)[2:].count('1'):
        answer = i
        break

    return answer

 


 

2. 피보나치 수

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

피보나치 수는 F(0) = 0, F(1) = 1일 때, 1 이상의 n에 대하여 F(n) = F(n-1) + F(n-2) 가 적용되는 수 입니다.
예를들어
F(2) = F(0) + F(1) = 0 + 1 = 1F(3) = F(1) + F(2) = 1 + 1 = 2F(4) = F(2) + F(3) = 1 + 2 = 3F(5) = F(3) + F(4) = 2 + 3 = 5
와 같이 이어집니다.
2 이상의 n이 입력되었을 때, n번째 피보나치 수를 1234567으로 나눈 나머지를 리턴하는 함수, solution을 완성해 주세요.
# Dynamic Programming
def fibo(n):
  dp = [0] * (n+1)
  dp[0], dp[1] = 0, 1
  for i in range(2, n+1):
    dp[i] = dp[i-2] + dp[i-1]
  return dp[n]

def solution(n):
    return fibo(n) % 1234567
# 더 간단한 방법
def solution(n):
    a, b = 0, 1

    for _ in range(1, n):
        a, b = b, (a + b) % 1234567

    return b