結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー femto
提出日時 2016-10-15 00:26:39
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 187 ms / 4,000 ms
コード長 1,473 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,591 ms
コンパイル使用メモリ 112,104 KB
実行使用メモリ 75,064 KB
最終ジャッジ日時 2024-11-22 08:57:45
合計ジャッジ時間 9,381 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <cstring>
#include <string>
#include <algorithm>
#include <iomanip>
#include <cmath>
#include <cassert>
#include <map>
#include <set>
#include <queue>
using namespace std;

struct T {
	int s, p, u, id;
	bool operator < (const T& t) const {
		if(s != t.s) return s > t.s;
		return p < t.p;
	}
};

deque<T> ts[100010];
int sum[11];
int sel[100010];
vector<int> rnk[100010];

int main() {
	cin.tie(0);
	ios::sync_with_stdio(false);

	int N, K;
	cin >> N >> K;

	for(int i = 0; i < N; i++) {
		T t;
		cin >> t.s >> t.p >> t.u;
		t.id = i;
		ts[t.u].push_back(t);
		sum[t.s]++;
	}
	for(int i = 0; i < 100010; i++) {
		sort(ts[i].begin(), ts[i].end());
	}

	vector<int> ans;
	while(1) {
		for(int s = 10; s >= 0; s--) {
			if(sum[s] == 0) continue;
			for(int u = 0; u < 100010; u++) {
				if(ts[u].size() && ts[u].front().s == s) rnk[sel[u]].push_back(u);
			}
			for(int r = 0; r < 100010; r++) {
				if(rnk[r].size() == 0) continue;
				vector<T> temp;
				for(int u : rnk[r]) {
					temp.push_back(ts[u][0]);
				}
				sort(temp.begin(), temp.end());
				for(T t : temp) {
					ans.push_back(t.id);
					if(ans.size() == K) {
						for(int v : ans) {
							cout << v << endl;
						}
						return 0;
					}

					sum[s]--;
					ts[t.u].pop_front();
					sel[t.u]++;
					if(ts[t.u].size() && ts[t.u].front().s == s) {
						rnk[sel[t.u]].push_back(t.u);
					}
				}
				rnk[r].clear();
				if(sum[s] == 0) break;
			}
		}
	}
}
0