문제
무지가 먹방을 시작한 지 K 초 후에 네트워크 장애로 인해 방송이 잠시 중단되었다. 무지는 네트워크 정상화 후 다시 방송을 이어갈 때, 몇 번 음식부터 섭취해야 하는지를 알고자 한다. 각 음식을 모두 먹는데 필요한 시간이 담겨있는 배열 food_times, 네트워크 장애가 발생한 시간 K 초가 매개변수로 주어질 때 몇 번 음식부터 다시 섭취하면 되는지 return 하도록 solution 함수를 완성하라. programmers
코드
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
struct Node {
int value;
int idx;
};
bool compare_by_value(Node a, Node b) {
return a.value < b.value;
}
bool compare_by_idx(Node a, Node b) {
return a.idx < b.idx;
}
int solution(vector<int> food_times, long long k) {
int answer = 0;
vector<Node> min_idx;
for (int i = 0; i < food_times.size(); i++) {
Node n = { food_times[i], i };
min_idx.push_back(n);
}
sort(min_idx.begin(), min_idx.end(), compare_by_value);
long long cur_time = 0;
int l = min_idx.size();
for (int i = 0; i < min_idx.size(); i++) {
long long new_time = (long long)(min_idx[i].value - cur_time) * l;
if (new_time == 0) {
l--;
continue;
}
if (new_time <= k) {
k -= new_time;
l--;
cur_time = min_idx[i].value;
} else {
k %= l;
vector<Node> remain(min_idx.begin() + i, min_idx.end());
sort(remain.begin(), remain.end(), compare_by_idx);
return remain[k].idx + 1;
}
}
return -1;
}
풀이
먹는데 필요한 시간이 적은 순으로 정렬하고, 가장 작은 것부터 그 시간을 한번에 단축시킨다.