Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Main {
- public static void main(String[] args) {
- int n, k;
- List<List<Integer>> a;
- // Считали ети списки
- // n - общее
- // k - кол-во списков
- // храню в сете пару (число, номер списка из которого это число)
- // сортировка в первую очередь будет происходить по первому числу
- // это нам и нужно
- TreeSet<Pair<Integer, Integer>> set = new TreeSet();
- for (int i = 0; i < k; ++i) {
- // добавляю в сет начальный элемент каждого из списков
- set.add(new Pair<>(a.get(i).get(0), i));
- a.get(i).remove(a.get(i).size() - 1);
- }
- int res [] = new int[n];
- for (int i = 0; i < n; ++i) {
- // беру из сета минимальный элемент
- // ставлю его в результирующий массив
- // удаляю из сета и добавляю в сет следующий элемент того списка
- // если список не пустой
- // удаляю из списка то что добавил в сет
- Pair<Integer, Integer> current = set.first();
- res[i] = current.getKey();
- set.remove(set.first());
- int ind = current.getValue();
- if (a.get(ind).isEmpty()) continue;
- current = new Pair<>(a.get(ind).get(a.get(ind).size() - 1), ind);
- a.get(ind).remove(a.get(ind).size() - 1);
- set.add(current);
- }
- // результат это res
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement