結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー pekempey
提出日時 2016-10-14 22:45:17
言語 C++11
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=gnu++11 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 75 ms / 4,000 ms
コード長 1,087 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,266 ms
コンパイル使用メモリ 196,912 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2026-05-16 10:42:33
合計ジャッジ時間 4,766 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

template<class T> using pqueue = priority_queue<T, vector<T>, greater<T>>;

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

	vector<int> S(N), P(N), U(N);
	vector<vector<int>> ls(11);
	for (int i = 0; i < N; i++) {
		scanf("%d %d %d", &S[i], &P[i], &U[i]);
		U[i]--;
		ls[S[i]].push_back(i);
	}

	vector<int> ord;
	vector<int> num(100000);
	for (int k = 10; k >= 0; k--) {
		vector<int> ptr(100000);
		vector<vector<pair<int, int>>> ll(100000);
		for (int id : ls[k]) {
			ll[U[id]].emplace_back(P[id], id);
		}
		pqueue<tuple<int, int, int>> q;
		for (int i = 0; i < 100000; i++) {
			if (!ll[i].empty()) {
				sort(ll[i].begin(), ll[i].end());
				q.emplace(num[i], ll[i][0].first, ll[i][0].second);
			}
		}
		while (!q.empty()) {
			int id = get<2>(q.top()); q.pop();
			int uid = U[id];
			ord.push_back(id);
			ptr[uid]++;
			num[uid]++;
			if (ptr[uid] < ll[uid].size()) {
				q.emplace(num[uid], ll[uid][ptr[uid]].first, ll[uid][ptr[uid]].second);
			}
		}
	}

	ord.resize(K);
	for (int i = 0; i < K; i++) {
		cout << ord[i] << endl;
	}
}
0