結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
ユーザー simansiman
提出日時 2023-01-30 20:08:51
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 183 ms / 4,000 ms
コード長 2,384 bytes
コンパイル時間 5,638 ms
コンパイル使用メモリ 106,688 KB
実行使用メモリ 11,200 KB
最終ジャッジ日時 2023-09-12 17:35:02
合計ジャッジ時間 14,860 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 170 ms
10,104 KB
testcase_01 AC 150 ms
9,800 KB
testcase_02 AC 114 ms
9,684 KB
testcase_03 AC 140 ms
9,864 KB
testcase_04 AC 137 ms
9,684 KB
testcase_05 AC 128 ms
9,840 KB
testcase_06 AC 183 ms
11,200 KB
testcase_07 AC 124 ms
9,804 KB
testcase_08 AC 113 ms
10,800 KB
testcase_09 AC 118 ms
9,672 KB
testcase_10 AC 113 ms
9,176 KB
testcase_11 AC 131 ms
9,072 KB
testcase_12 AC 123 ms
9,404 KB
testcase_13 AC 83 ms
9,096 KB
testcase_14 AC 101 ms
9,132 KB
testcase_15 AC 70 ms
8,584 KB
testcase_16 AC 139 ms
8,988 KB
testcase_17 AC 138 ms
9,652 KB
testcase_18 AC 143 ms
9,604 KB
testcase_19 AC 83 ms
9,180 KB
testcase_20 AC 95 ms
9,132 KB
testcase_21 AC 91 ms
9,192 KB
testcase_22 AC 123 ms
9,200 KB
testcase_23 AC 107 ms
9,452 KB
testcase_24 AC 84 ms
9,156 KB
testcase_25 AC 117 ms
9,120 KB
testcase_26 AC 91 ms
8,680 KB
testcase_27 AC 113 ms
8,924 KB
testcase_28 AC 82 ms
9,148 KB
testcase_29 AC 123 ms
8,760 KB
testcase_30 AC 5 ms
6,700 KB
testcase_31 AC 5 ms
6,724 KB
testcase_32 AC 142 ms
10,296 KB
testcase_33 AC 147 ms
9,304 KB
testcase_34 AC 115 ms
9,992 KB
testcase_35 AC 6 ms
6,744 KB
testcase_36 AC 5 ms
6,632 KB
testcase_37 AC 6 ms
6,632 KB
testcase_38 AC 6 ms
6,628 KB
testcase_39 AC 6 ms
6,600 KB
testcase_40 AC 7 ms
6,776 KB
testcase_41 AC 13 ms
7,068 KB
testcase_42 AC 13 ms
7,016 KB
testcase_43 AC 13 ms
6,868 KB
testcase_44 AC 13 ms
6,900 KB
testcase_45 AC 7 ms
6,840 KB
testcase_46 AC 27 ms
6,984 KB
testcase_47 AC 27 ms
6,972 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <climits>
#include <map>
#include <queue>
#include <set>
#include <cstring>
#include <vector>

using namespace std;
typedef long long ll;

struct Node {
  int id;
  int solved_cnt;
  int penalty;
  int uid;

  Node(int id = -1, int solved_cnt = -1, int penalty = -1, int uid = -1) {
    this->id = id;
    this->solved_cnt = solved_cnt;
    this->penalty = penalty;
    this->uid = uid;
  }

  bool operator>(const Node &n) const {
    if (solved_cnt == n.solved_cnt) {
      return penalty > n.penalty;
    } else {
      return solved_cnt < n.solved_cnt;
    }
  }
};

struct SubNode {
  int id;
  int uid;
  int selected_cnt;
  int penalty;

  SubNode(int id = -1, int uid = -1, int selected_cnt = -1, int penalty = -1) {
    this->id = id;
    this->uid = uid;
    this->selected_cnt = selected_cnt;
    this->penalty = penalty;
  }

  bool operator>(const SubNode &n) const {
    if (selected_cnt == n.selected_cnt) {
      return penalty > n.penalty;
    } else {
      return selected_cnt > n.selected_cnt;
    }
  }
};

int main() {
  int N, K;
  cin >> N >> K;

  priority_queue <Node, vector<Node>, greater<Node>> pque_list[100010];
  for (int i = 0; i < N; ++i) {
    int S, P, U;
    cin >> S >> P >> U;

    pque_list[U].push(Node(i, S, P, U));
  }

  vector<int> selected_ids;
  vector<int> selected_counter(100010, 0);

  for (int s = 10; s >= 0; --s) {
    priority_queue <SubNode, vector<SubNode>, greater<SubNode>> pque;

    for (int uid = 1; uid <= 100000; ++uid) {
      if (pque_list[uid].empty()) continue;
      if (pque_list[uid].top().solved_cnt < s) continue;

      Node node = pque_list[uid].top();
      pque_list[uid].pop();

      pque.push(SubNode(node.id, uid, selected_counter[uid], node.penalty));
    }

    while (not pque.empty() && selected_ids.size() < K) {
      SubNode s_node = pque.top();
      pque.pop();

      int uid = s_node.uid;
      selected_ids.push_back(s_node.id);
      selected_counter[uid]++;

      if (pque_list[uid].size() > 0 && pque_list[uid].top().solved_cnt >= s) {
        Node node = pque_list[uid].top();
        pque_list[uid].pop();

        pque.push(SubNode(node.id, uid, selected_counter[uid], node.penalty));
      }
    }
  }

  for (int id : selected_ids) {
    cout << id << endl;
  }

  return 0;
}
0