結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
ユーザー antaanta
提出日時 2016-10-14 22:42:35
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 69 ms / 4,000 ms
コード長 1,851 bytes
コンパイル時間 2,506 ms
コンパイル使用メモリ 188,488 KB
実行使用メモリ 11,096 KB
最終ジャッジ日時 2024-11-22 07:44:21
合計ジャッジ時間 5,638 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 63 ms
9,856 KB
testcase_01 AC 62 ms
9,860 KB
testcase_02 AC 53 ms
7,552 KB
testcase_03 AC 51 ms
6,816 KB
testcase_04 AC 50 ms
6,816 KB
testcase_05 AC 54 ms
7,872 KB
testcase_06 AC 69 ms
11,096 KB
testcase_07 AC 61 ms
9,796 KB
testcase_08 AC 58 ms
9,036 KB
testcase_09 AC 48 ms
6,816 KB
testcase_10 AC 41 ms
6,820 KB
testcase_11 AC 42 ms
6,820 KB
testcase_12 AC 40 ms
6,816 KB
testcase_13 AC 41 ms
6,820 KB
testcase_14 AC 38 ms
6,820 KB
testcase_15 AC 36 ms
6,820 KB
testcase_16 AC 42 ms
6,816 KB
testcase_17 AC 43 ms
6,820 KB
testcase_18 AC 42 ms
6,820 KB
testcase_19 AC 40 ms
6,816 KB
testcase_20 AC 38 ms
6,820 KB
testcase_21 AC 40 ms
6,820 KB
testcase_22 AC 41 ms
6,816 KB
testcase_23 AC 41 ms
6,816 KB
testcase_24 AC 40 ms
6,816 KB
testcase_25 AC 42 ms
6,820 KB
testcase_26 AC 38 ms
6,820 KB
testcase_27 AC 40 ms
6,816 KB
testcase_28 AC 39 ms
6,816 KB
testcase_29 AC 40 ms
6,820 KB
testcase_30 AC 2 ms
6,816 KB
testcase_31 AC 2 ms
6,820 KB
testcase_32 AC 43 ms
6,820 KB
testcase_33 AC 45 ms
6,816 KB
testcase_34 AC 51 ms
7,016 KB
testcase_35 AC 2 ms
6,820 KB
testcase_36 AC 2 ms
6,820 KB
testcase_37 AC 2 ms
6,816 KB
testcase_38 AC 2 ms
6,816 KB
testcase_39 AC 2 ms
6,820 KB
testcase_40 AC 2 ms
6,820 KB
testcase_41 AC 5 ms
6,820 KB
testcase_42 AC 5 ms
6,816 KB
testcase_43 AC 5 ms
6,816 KB
testcase_44 AC 5 ms
6,820 KB
testcase_45 AC 2 ms
6,820 KB
testcase_46 AC 6 ms
6,820 KB
testcase_47 AC 6 ms
6,816 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