結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
ユーザー antaanta
提出日時 2016-10-14 22:42:35
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 64 ms / 4,000 ms
コード長 1,851 bytes
コンパイル時間 2,413 ms
コンパイル使用メモリ 186,632 KB
実行使用メモリ 10,760 KB
最終ジャッジ日時 2023-08-14 11:18:09
合計ジャッジ時間 5,947 ms
ジャッジサーバーID
(参考情報)
judge15 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 59 ms
9,320 KB
testcase_01 AC 60 ms
9,632 KB
testcase_02 AC 50 ms
7,312 KB
testcase_03 AC 48 ms
6,332 KB
testcase_04 AC 47 ms
6,064 KB
testcase_05 AC 51 ms
7,960 KB
testcase_06 AC 64 ms
10,760 KB
testcase_07 AC 58 ms
9,572 KB
testcase_08 AC 55 ms
8,708 KB
testcase_09 AC 46 ms
6,256 KB
testcase_10 AC 38 ms
6,128 KB
testcase_11 AC 39 ms
5,904 KB
testcase_12 AC 38 ms
5,484 KB
testcase_13 AC 37 ms
5,988 KB
testcase_14 AC 36 ms
5,644 KB
testcase_15 AC 34 ms
5,780 KB
testcase_16 AC 39 ms
5,832 KB
testcase_17 AC 40 ms
6,000 KB
testcase_18 AC 40 ms
5,992 KB
testcase_19 AC 36 ms
6,104 KB
testcase_20 AC 36 ms
5,488 KB
testcase_21 AC 37 ms
5,928 KB
testcase_22 AC 39 ms
5,788 KB
testcase_23 AC 38 ms
5,888 KB
testcase_24 AC 37 ms
6,068 KB
testcase_25 AC 39 ms
5,844 KB
testcase_26 AC 35 ms
6,172 KB
testcase_27 AC 37 ms
5,836 KB
testcase_28 AC 37 ms
6,108 KB
testcase_29 AC 37 ms
6,164 KB
testcase_30 AC 1 ms
4,384 KB
testcase_31 AC 2 ms
4,380 KB
testcase_32 AC 40 ms
5,640 KB
testcase_33 AC 42 ms
6,200 KB
testcase_34 AC 48 ms
6,776 KB
testcase_35 AC 1 ms
4,380 KB
testcase_36 AC 2 ms
4,384 KB
testcase_37 AC 2 ms
4,384 KB
testcase_38 AC 2 ms
4,380 KB
testcase_39 AC 2 ms
4,384 KB
testcase_40 AC 2 ms
4,380 KB
testcase_41 AC 5 ms
4,384 KB
testcase_42 AC 5 ms
4,380 KB
testcase_43 AC 5 ms
4,384 KB
testcase_44 AC 5 ms
4,380 KB
testcase_45 AC 2 ms
4,380 KB
testcase_46 AC 6 ms
4,384 KB
testcase_47 AC 6 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#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