동캄의 코딩도장

프로그래머스 level0 [다음에 올 숫자] 파이썬 본문

코테/프로그래머스 level0

프로그래머스 level0 [다음에 올 숫자] 파이썬

동 캄 2023. 3. 5. 08:10

https://school.programmers.co.kr/learn/courses/30/lessons/120924

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

엄청 오랜만에 글을 쓰는거 같다. 꾸준히 연습해야지...!

def solution(common):
    answer = 0
    if(common[2]*common[0]==common[1]**2):
        if(common[0]==0 and common[1]==0):
            answer=0
        else:
            answer=common[-1]*(common[1]//common[0])
    else:
        answer=common[-1]+common[1]-common[0]
    return answer

먼저 등비수열인지 확인하고, 등비 수열이 아니면 등차가 0인지, 0이 아닌지 확인한다.