結果

問題 No.953 席
ユーザー WarToksWarToks
提出日時 2020-01-03 01:14:20
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 121 ms / 2,000 ms
コード長 6,228 bytes
コンパイル時間 1,556 ms
コンパイル使用メモリ 150,092 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-11-30 16:02:16
合計ジャッジ時間 5,155 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 50 ms
5,248 KB
testcase_03 AC 46 ms
5,248 KB
testcase_04 AC 49 ms
5,248 KB
testcase_05 AC 44 ms
5,248 KB
testcase_06 AC 48 ms
5,248 KB
testcase_07 AC 121 ms
9,988 KB
testcase_08 AC 94 ms
8,592 KB
testcase_09 AC 103 ms
10,752 KB
testcase_10 AC 34 ms
5,632 KB
testcase_11 AC 75 ms
7,164 KB
testcase_12 AC 65 ms
6,668 KB
testcase_13 AC 83 ms
8,172 KB
testcase_14 AC 61 ms
6,148 KB
testcase_15 AC 59 ms
5,888 KB
testcase_16 AC 64 ms
6,144 KB
testcase_17 AC 81 ms
6,876 KB
testcase_18 AC 69 ms
6,520 KB
testcase_19 AC 96 ms
8,364 KB
testcase_20 AC 69 ms
6,368 KB
testcase_21 AC 53 ms
5,376 KB
testcase_22 AC 69 ms
7,412 KB
testcase_23 AC 53 ms
5,248 KB
testcase_24 AC 54 ms
5,248 KB
testcase_25 AC 92 ms
8,448 KB
testcase_26 AC 39 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <cstring>
#include <deque>
#include <functional>
#include <math.h>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <queue>
#include <vector>

#define FOR(i, a, b) for(int (i) = (a); (i) < (b); ++(i))
#define rFOR(i, a, b) for(int (i) = (b); (i) >= (a); --(i))
#define REP(i, n) FOR(i, 0, n)
#define rREP(i, n) rFOR(i, 0, (n-1))
#define SORT(A) std::sort((A).begin(), (A).end())
#define ALL(A) (A).begin(), (A).end()
// 座標圧縮 (for vector) : ソートしてから使うのが一般的 ; SORT(A) => COORDINATE_COMPRESSION(A)
#define COORDINATE_COMPRESSION(A) (A).erase(unique((A).begin(),(A).end()),(A).end())

using lli = long long int;
using pii = std::pair<int, int>;

void VintOut(std::vector<int>& A){
    const int n = A.size();
    if(n == 0){putchar('\n'); return;}
    printf("%d", A[0]);
    for(int i = 1; i < n; ++i) printf(" %d", A[i]);
    putchar('\n');
}
void VintOut(std::vector<long long int>& A){
    const int n = A.size();
    if(n == 0){ putchar('\n'); return;}
    printf("%lld", A[0]);
    for(int i = 1; i < n; ++i) printf(" %lld", A[i]);
    putchar('\n');
}

template <typename T>
inline bool chmin(T& a, T b){
    if(b < a){ a = b; return true;}
    return false;
}

template <typename T>
inline bool chmax(T& a, T b){
    if(a < b){ a = b; return true;}
    return false;
}


// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

constexpr int N = 1e5;
std::bitset<N> Invacant; // 座っているか否か
std::vector<char> Neighborhood; // 隣の人が空か
std::vector<int> Res; // 答え

std::vector<int> Priority; // 近さの順位
std::vector<std::pair<int, int>> Ppriority;


 
int main(void){
    int n, k1, k2; scanf("%d%d%d", &n, &k1, &k2); k1--; k2--;

    // 座席の優先順位の決定
    Ppriority.resize(n);
    REP(i, n){
        // (2 * k1 + k2)/3 - t = |2k1 + k2 - 3t|/3
        int u = 2 * k1 + k2 - 3 * i; if(u < 0) u *= -1;
        Ppriority[i] = {u, i};
    }
    std::sort(Ppriority.begin(), Ppriority.end());
    Priority.resize(n);
    REP(i, n) Priority[Ppriority[i].second] = i;


    Neighborhood.assign(n, 1);

    int q; scanf("%d", &q);
    Res.resize(q, -1);

    
    // 退店時刻, インデックス
    using datum = std::pair<lli, int>;
    std::queue<std::pair<int, int>> Q;
    std::priority_queue<datum, std::vector<datum>, std::greater<datum>> PQ;
    std::set<std::pair<bool, std::pair<int, int>>> S;

    REP(i, n) S.insert({false, {Priority[i], i}});

    std::function<void(int, bool)> update = [&](int seat, bool inc){
        if(seat < 0 or seat >= n) return; // 例外
        if(Invacant[seat]){
            if(inc) Neighborhood[seat]++; else Neighborhood[seat]--;
            return;
        }
        S.erase({Neighborhood[seat] <= 0, {Priority[seat], seat}});
        if(inc) Neighborhood[seat]++; else Neighborhood[seat]--;
        S.insert({Neighborhood[seat] <= 0, {Priority[seat], seat}});
    };

    REP(i, q){
        int a, b; scanf("%d%d", &a, &b);

        // i番目を座らせる前に起きる退店-待ちイベントを処理する
        while((not PQ.empty()) and PQ.top().first <= a){
            lli theTime = PQ.top().first;
            while((not PQ.empty()) and PQ.top().first == theTime){ // 同時刻に帰る客を全員帰らせる
                lli t; int idx;
                std::tie(t, idx) = PQ.top(); PQ.pop();
                // 退店処理
                int seat = Res[idx];
                Invacant[seat] = false;
                update(seat - 1, true); update(seat + 1, true);
                S.insert({Neighborhood[seat] <= 0, {Priority[seat], seat}});

                //printf("At %lld : Person %d | 退店 from %d\n", t, idx, seat);
            }
            
            while((not Q.empty()) and (not S.empty())){ // お待ちのお客様！
                // 入店処理
                auto val = *S.begin(); S.erase(S.begin());
                int seat = val.second.second;
                Invacant[seat] = true;
                update(seat - 1, false);
                update(seat + 1, false);

                int member, length;
                std::tie(member, length) = Q.front(); Q.pop();
                Res[member] = seat;
                PQ.push({theTime + length, member});

                //printf("At %lld : Person %d | 着席 at %d\n", theTime, member, seat);
            }
        }
        if(S.empty()) Q.push({i, b}); // 待つ場合
        else{
            auto val = *S.begin(); S.erase(S.begin());
            int seat = val.second.second;
            Invacant[seat] = true;
            update(seat - 1, false);
            update(seat + 1, false);
            Res[i] = seat;
            PQ.push({a + b, i});

            //printf("At %d : Person %d | 着席 at %d\n", a, i, seat);
        }
    }

    while(not Q.empty()){

            lli theTime = PQ.top().first;
            while((not PQ.empty()) and PQ.top().first == theTime){ // 同時刻に帰る客を全員帰らせる
                lli t; int idx;
                std::tie(t, idx) = PQ.top(); PQ.pop();
                // 退店処理
                int seat = Res[idx];
                Invacant[seat] = false;
                update(seat - 1, true); update(seat + 1, true);
                S.insert({Neighborhood[seat] <= 0, {Priority[seat], seat}});

                //printf("At %lld : Person %d | 退店 from %d\n", t, idx, seat);
            }

                    
        while((not Q.empty()) and (not S.empty())){ // お待ちのお客様！
            // 入店処理
            auto val = *S.begin(); S.erase(S.begin());
            int seat = val.second.second;


            Invacant[seat] = true;
            update(seat - 1, false);
            update(seat + 1, false);

            int member, length;
            std::tie(member, length) = Q.front(); Q.pop();
            Res[member] = seat;
            PQ.push({theTime + length, member});

            //printf("At %lld : Person %d | 着席 at %d\n", theTime, member, seat);
        }
    }

    REP(i, q) printf("%d\n", Res[i] + 1);


    
    
    return 0;
}
0