취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 3079번 문제인 입국심사이다.
문제는 아래 링크를 확인하자.

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

구하는 시간 T의 범위를 [1, 1000000000000000000]으로 잡고, 심사를 T 안에 통과할 수 있는가 없는가로 이분탐색을 해나가는 것으로 문제를 해결할 수 있다.

다만, 이분탐색의 max범위가 너무 커 전체 시간동안 심사할 수 있는 사람의 수의 계산과정에서 오버플로우가 일어날 수 있으니 구현에 유의하여야한다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;
typedef long long ll;

ll arr[100000];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	ll N, M; cin >> N >> M;
	for (int i = 0; i < N; i++) cin >> arr[i];

	ll L = 1, R = 1000000000000000000LL;
	while (L < R) {
		ll temp = 0;
		ll mid = (L + R) / 2;
		for (int i = 0; i < N; i++) temp = min(temp + mid / arr[i], 1000000000000000000LL);
		if (temp >= M) R = mid;
		else L = mid + 1;
	}

	cout << L;
}