[파이썬/python] 백준 - 11564 점프왕 최준민

문제

https://www.acmicpc.net/problem/11564

---

문제 설명

1.   수직선상에 초콜릿이 존재한다.
   1. 초콜릿은 a 이상 b 이하의 모든 정수 좌표 위에 존재한다.
2. 점프력이 K인 준민이는 항상 점프 거리가 K가 되도록 점프한다.
3. 0에서 시작해서 최대 몇 개의 초콜릿을 얻을 수 있는지 구한다.

 

---

풀이

단순 수식 계산 문제라고 생각했다.

먼저 두 숫자가 0을 기준으로 한쪽에 몰려있는지 판단했다.

 

만약 하나는 음수, 하나는 양수라면, 0의 위치에 초콜릿을 먹고 시작한다. 그 후 각각 0을 기준으로 a의 배수, b의 배수만큼 초콜릿을 먹을 수 있다.

 

두 숫자 모두 양수 혹은 음수라면 사이에 존재하는 초콜릿을 판별하여 결과에 합산했다.

 

1. A*B가 음수일 때
   1. 두 수의 절댓값을 K로 나누어 초콜릿의 수를 구한다.
2. A*B가 양수일 때
   1. a와 b의 절댓값 중 최소 최대를 구해 범위를 구한다.
   2. 범위 사이에서 초콜릿이 총 몇개가 나올 수 있는지 구한다.

---

코드

import sys
input = sys.stdin.readline

k,a,b = map(int,input().split())
if a*b < 0:
    cnt = 1
    a,b = abs(a), abs(b)
    cnt += (a//k + b//k)
else:
    a,b = min(abs(a), abs(b)),max(abs(a), abs(b))
    cnt = 0
    cnt += ((b//k) - ((a+k)//k) + 1)
    if a % k == 0:
        cnt += 1
print(cnt)

---

시간복잡도

• 단순 O(1)로 해결이 가능하다.