백준BOJ/Java/Python : 2292번 벌집

2292번 : 벌집 원본

알고리즘 분류

• 수학

문제

위의 그림과 같이 육각형으로 이루어진 벌집이 있다.
그림에서 보는 바와 같이 중앙의 방 1부터 시작해서 이웃하는 방에 돌아가면서
1씩 증가하는 번호를 주소로 매길 수 있다. 숫자 N이 주어졌을 때,
벌집의 중앙 1에서 N번 방까지 최소 개수의 방을 지나서 갈 때
몇 개의 방을 지나가는지(시작과 끝을 포함하여)를 계산하는 프로그램을 작성하시오.
예를 들면, 13까지는 3개, 58까지는 5개를 지난다.

입력

첫째 줄에 N(1 ≤ N ≤ 1,000,000,000)이 주어진다.

출력

입력으로 주어진 방까지 최소 개수의 방을 지나서 갈 때 몇 개의 방을 지나는지 출력한다.

풀이

1에서부터 최소 이동 거리의 겹이 쌓일 때마다 1, 6 * 1, 6 * 2, 6 * 3...의 6의 배수만큼 늘어난다.

소스 코드

Java

import java.io.*;

public class Baek2292 {
    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));

        int num = Integer.parseInt(reader.readLine());
        int cnt = 1;
        int a = 1;

        while (true) {
            if (num <= a) {
                break;
            } else {
                a += (6 * cnt);
                cnt++;
            }
        }
        System.out.println(cnt);
    }
}

Python

def cal(n):
    if n == 1:
        return 1
    else:
        cnt = 1
        n -= 1
        while n > 0:
            n -= cnt * 6
            cnt += 1
        return cnt


num = int(input())
print(cal(num))