문제

무지가 먹방을 시작한 지 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;
}

풀이

먹는데 필요한 시간이 적은 순으로 정렬하고, 가장 작은 것부터 그 시간을 한번에 단축시킨다.