結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー pekempey
提出日時 2016-10-14 22:45:17
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 134 ms / 4,000 ms
コード長 1,087 bytes
コンパイル時間 1,856 ms
コンパイル使用メモリ 179,184 KB
実行使用メモリ 10,876 KB
最終ジャッジ日時 2024-11-22 07:56:24
合計ジャッジ時間 6,741 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:13:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   13 |                 scanf("%d %d %d", &S[i], &P[i], &U[i]);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#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