다음에 올 숫자

쉬웠지만 어려웠던 문제...

📝 문제

💡 해결..?

쉽다고 생각했다. 한 1분만에 다 써내리고 체점을 했는데.. 테스트 4만 막혀서
넘어가질 않았다.

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

생각해보니 음수일 경우에는 작동하지 않을것 같아 test 4의 경우가
그 경우라고 생각하고 수정했으나 그럼에도...먹히지 않았다.

그래서

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

으로 짰더니 성공.....
common[2] - common[1]이 common[0]이 아닌 것도 있는지,,,, 흠냐...

근데 이것 코드도 음수는 안되는거 아닌가?

알수가 없다...아무튼...해결!