結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー anta
提出日時 2016-10-14 22:42:35
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 69 ms / 4,000 ms
コード長 1,851 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,506 ms
コンパイル使用メモリ 188,488 KB
実行使用メモリ 11,096 KB
最終ジャッジ日時 2024-11-22 07:44:21
合計ジャッジ時間 5,638 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include "bits/stdc++.h"
using namespace std;
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#define rer(i,l,u) for(int (i)=(int)(l);(i)<=(int)(u);++(i))
#define reu(i,l,u) for(int (i)=(int)(l);(i)<(int)(u);++(i))
static const int INF = 0x3f3f3f3f; static const long long INFL = 0x3f3f3f3f3f3f3f3fLL;
typedef vector<int> vi; typedef pair<int, int> pii; typedef vector<pair<int, int> > vpii; typedef long long ll;
template<typename T, typename U> static void amin(T &x, U y) { if(y < x) x = y; }
template<typename T, typename U> static void amax(T &x, U y) { if(x < y) x = y; }

struct Team {
	int S, P, U;
};

int main() {
	int N; int K;
	while(~scanf("%d%d", &N, &K)) {
		int M = 0, MaxS = 0;
		vector<Team> teams(N);
		rep(i, N) {
			int S; int P; int U;
			scanf("%d%d%d", &S, &P, &U), -- U;
			amax(M, U + 1);
			amax(MaxS, S);
			teams[i] = { S,P,U };
		}
		vector<vi> byScore(MaxS + 1);
		rep(i, N)
			byScore[teams[i].S].push_back(i);
		vector<vpii> byUniversity(M);
		vector<int> universities;
		priority_queue<pair<ll,int>> pq;
		vector<int> list;
		vector<int> cnt(M, 0);
		for(int s = MaxS; s >= 0; -- s) {
			for(int i : byScore[s]) {
				int u = teams[i].U;
				if(byUniversity[u].empty())
					universities.push_back(u);
				byUniversity[u].emplace_back(teams[i].P, i);
			}
			for(int u : universities) {
				auto &v = byUniversity[u];
				sort(v.rbegin(), v.rend());
				pq.push(make_pair(-((ll)cnt[u] << 32 | v.back().first), u));
			}
			while(!pq.empty()) {
				int u = pq.top().second;
				pq.pop();
				auto &v = byUniversity[u];
				int i = v.back().second;
				list.push_back(i);
				++ cnt[u];
				v.pop_back();
				if(!v.empty())
					pq.push(make_pair(-((ll)cnt[u] << 32 | v.back().first), u));
			}
			universities.clear();
		}
		list.resize(K);
		for(int i = 0; i < K; ++ i)
			printf("%d\n", list[i]);
	}
	return 0;
}
0