結果

問題 No.2220 Range Insert & Point Mex
ユーザー otamay6otamay6
提出日時 2023-02-17 23:32:16
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 3,655 bytes
コンパイル時間 8,604 ms
コンパイル使用メモリ 275,224 KB
最終ジャッジ日時 2025-02-10 18:17:31
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define REP(i,n) for(int i=0,i##_len=int(n);i<i##_len;++i)
#define rep(i,a,b) for(int i=int(a);i<int(b);++i)
#define All(x) (x).begin(),(x).end()
#define rAll(x) (x).rbegin(),(x).rend()
using namespace std;
using ll = long long;

void connect(vector<pair<int,int>> &ranges) {
    // rangesは昇順データ
    vector<pair<int,int>> nxt_range;
    sort(All(ranges));
    for(const auto &range: ranges) {
        if(nxt_range.size() == 0){
            nxt_range.push_back(range);
        }
        else {
            pair<int,int> &fr = nxt_range.back();
            if(range.first <= fr.second + 1) {
                fr.second = max(fr.second, range.second);
            }
            else {
                nxt_range.push_back(range);
            }
        }
    }
    ranges = nxt_range;
}

int main(){
    std::cin.tie(0)->sync_with_stdio(0);
    int n;std::cin>>n;
    vector<pair<int, pair<int,int>>> op(n);
    REP(i,n){
        int l, r,a;
        std::cin>>l>>r>>a;
        op[i] = {a, {l, r}};
    } 
    sort(All(op));
    // 数iを含む区間[l,r]の範囲
    vector<vector<pair<int,int>>> rgs(n);
    int mx = 0;
    int Lidx = 0;
    for(auto p:op){
        int a = p.first;
        pair<int,int> range = p.second;
        if(a >= n) {
            // どうせn以上のaは意味がない
            break;
        }
        else{
            rgs[a].push_back(range);
        }
    }
    vector<vector<pair<int,int>>> dp(n);
    // 全部の範囲を連結してから判定する
    dp[0] = rgs[0];
    connect(dp[0]);
    rep(a, 1, n){
        connect(rgs[a]);
        int Lidx = 0;
        for(const auto &range: rgs[a]){
            int l = range.first;
            int r = range.second;
            rep(i, Lidx, dp[a-1].size()) {
                pair<int, int> prev_range = dp[a-1][i];
                if(prev_range.second < l){
                    Lidx = i+1;
                    continue;
                }
                if(r < prev_range.first) {
                    break;
                }
                int nl = max(l, prev_range.first);
                int nr = min(r, prev_range.second);
                if(nl <= nr) {
                    dp[a].push_back({nl, nr});
                }
            }
        }
        sort(All(dp[a]));
        connect(dp[a]);
    }
    int q;std::cin>>q;
    REP(_,q){
        int x;std::cin>>x;
        // (l, r]の区間はSxに含まれない
        int l = -1, r = n;
        while(l+1 < r){
            int mid = (l+r) >> 1;
            if(dp[mid].size() == 0){
                r = mid;
                continue;
            }
            int lidx = -1, ridx = dp[mid].size();
            while(lidx + 1 < ridx) {
                int midx = (lidx + ridx) >> 1;
                const pair<int,int> &range = dp[mid][midx];
                if(x < range.first) {
                    ridx = midx;
                }
                else{
                    lidx = midx;
                }
            }
            if(lidx == -1){
                r = mid;
                continue;
            }
            bool inc = false;
            const pair<int,int> &range = dp[mid][lidx];
            // cerr << range.first << " " << x << endl;
            // assert(range.first <= x);
            // if(ridx+1 < dp[mid].size()){
            //     cerr << dp[mid][ridx+1].first << " " << x << endl;
            //     assert(x < dp[mid][ridx+1].first);
            // }
            if(range.first <= x && x <= range.second) {
                inc = true;
                l = mid;
            }
            if(!inc) r = mid;
        }
        cout << r << "\n";
    }
}
0