[백준 1590번] 캠프가는 영식

[백준 1590번] 캠프가는 영식

www.acmicpc.net/problem/1590

1590번: 캠프가는 영식

 

 

 문제

영식이는 민식이와 함게 고속버스를 타고 캠프를 가야 하지만, 민식이는 영식이를 깨우지 않고 혼자 버스를 타고 캠프에 가버렸다.

영식이는 혼자 고속버스터미널까지 가서 캠프에 오려고 한다. 터미널에는 캠프 장소까지 운행하는 N가지의 버스가 있다. 각각의 버스는 시작 시각, 간격, 대수의 정보를 가지고 있다. 예를 들어, 어떤 버스의 시작 시각이 특점 시점을 기준으로 10분 후이고, 간격은 10분이고, 대수가 5대이면, 이 버스는 10분, 20분, 30분, 40분, 50분에 한 대씩 출발한다.

영식이는 버스터미널에 T분에 도착했다. 영식이가 버스를 타려면 최소 몇 분을 더 기다려야 하는지 구하는 프로그램을 작성하시오.

 

 입력

첫째 줄에 버스의 개수 N과 영식이가 버스터미널에 도착하는 시간 T가 주어진다. 둘째 줄부터 총 N개의 줄에 각 버스의 시작 시각, 간격, 대수가 공백을 사이에 두고 주어진다. 버스의 개수와 각 버스의 대수는 100,000보다 작거나 같은 자연수이며, T와 각 버스의 시작 시각과 간격은 1,000,000,000보다 작거나 같은 자연수이다.

 

 

 출력

첫째 줄에 영식이가 기다려야 하는 시간을 출력한다. 영식이가 도착하는 동시에 버스가 출발하면 정답은 0이다. 만약 버스가 없어서 캠프에 갈 수 없으면 -1을 출력한다. 정답은 231보다 작다.

 

 

 풀이

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<long long> v;
int main() {
	long long N, T; cin >> N >> T;
	for(int i=0;i<N;i++){
		int a, b, c; cin >> a >> b >> c;
		for(int i=0;i<c;i++){
			long long t = a + b*i;
			v.push_back(t);
		}
	}
	sort(v.begin(), v.end());
	
	int first = 0;
	int last = v.size();
	while(first < last){
		int mid = (first + last)/2;
		if(v[mid] >= T) last = mid;
		else first = mid + 1;
	}
	if(v[first] - T < 0) cout << -1;
	else cout << v[first] - T;
	return 0;
}

 버스 종류마다 시작 시간, 간격, 대수를 고려하여 버스의 모든 출발 시간을 벡터에 담아둔다. 여기에서 영식이가 버스터미널에 도착하는 시간인 T와 보다 큰 수 중에 가장 작은 수를 찾는다. 이를 위해 벡터 정렬 후 이분탐색으로 값을 구한다. 아마 upper_bound를 사용해도 될 것 같다. 수의 크기를 고려하여 long long을 써주면 문제없다.