結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー tottoripaper
提出日時 2016-10-14 22:59:38
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 71 ms / 4,000 ms
コード長 1,921 bytes
コンパイル時間 1,995 ms
コンパイル使用メモリ 189,588 KB
実行使用メモリ 10,280 KB
最終ジャッジ日時 2024-11-22 08:48:06
合計ジャッジ時間 5,683 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define fst(t) std::get<0>(t)
#define snd(t) std::get<1>(t)
#define thd(t) std::get<2>(t)
#define fth(t) std::get<3>(t)

using ll = long long;
using P = std::tuple<int,int,int,int>;
using P3 = std::tuple<int,int,int>;

const int dx[8] = {-1, 1, 0, 0, -1, -1, 1, 1}, dy[8] = {0, 0, -1, 1, -1, 1, -1, 1};

int N, K, nxt[100000], prv[100000];
P ps[100000];
std::vector<P3> us[100000];
std::priority_queue<P, vector<P>, greater<P>> pq;

int main(){
    //*
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);
    /*/
      /*/

    std::cin >> N >> K;

    for(int i=0;i<N;++i){
        std::cin >> fst(ps[i]) >> snd(ps[i]) >> thd(ps[i]);
        fst(ps[i]) = -fst(ps[i]);
        fth(ps[i]) = i;
    }

    sort(ps, ps+N);

    int n = 0;
    for(int i=0,j=0;i<N;){
        vector<int> is;
        
        while(j < N && fst(ps[j]) == fst(ps[i])){
            int p, u, index;
            tie(ignore, p, u, index) = ps[j];

            us[u].emplace_back(p, u, index);
            is.emplace_back(u);
            
            ++j;
        }

        sort(is.begin(), is.end());
        is.erase(unique(is.begin(), is.end()), is.end());
        
        for(int u : is){
            sort(us[u].begin() + prv[u], us[u].end());
            pq.push(std::make_tuple(nxt[u], fst(us[u][nxt[u]]), snd(us[u][nxt[u]]), thd(us[u][nxt[u]])));
            ++nxt[u];
            prv[u] = us[u].size();
        }

        int _n = 0;
        while(_n < j - i){
            int u, index;
            tie(ignore, ignore, u, index) = pq.top();
            pq.pop();

            printf("%d\n", index);
            if(++n == K){return 0;}

            if(nxt[u] < us[u].size()){
                pq.push(std::make_tuple(nxt[u], fst(us[u][nxt[u]]), snd(us[u][nxt[u]]), thd(us[u][nxt[u]])));
                ++nxt[u];
            }
            ++_n;
        }

        i = j;
    }
}
0