結果

問題 No.1014 competitive fighting
ユーザー NOSS
提出日時 2020-03-20 23:46:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,068 bytes
コンパイル時間 2,738 ms
コンパイル使用メモリ 188,300 KB
実行使用メモリ 14,080 KB
最終ジャッジ日時 2024-12-15 21:01:41
合計ジャッジ時間 15,183 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 45 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using ll = long long int;
using ull = unsigned long long int;
using P = pair<ll, ll>;
using P3 = pair<P,ll>;
using PP = pair<P, P>;
constexpr int INF = 1 << 30;
constexpr ll MOD = ll(1e9)+7;
constexpr int di[] = {0, 1, 0, -1};
constexpr int dj[] = {1, 0, -1, 0};
constexpr double EPS = 1e-9;

template <typename Monoid>
struct SegmentTree{
private:
    using F = function<Monoid(Monoid, Monoid)>;
    int N;
    vector<Monoid> node;
    F f;
    Monoid e;  // identity element

public:
    SegmentTree(){}
    SegmentTree(F f, Monoid e):f(f), e(e){}
    void init(int sz){
        N = 1;
        while(N < sz) N <<= 1;
        node.assign(2*N-1, e);
    }
    void build(vector<Monoid>& v){
        int sz = int(v.size());
        init(sz);
        for(int i=0; i<sz; i++){
            node[i+N-1] = v[i];
        }
        for(int i=N-2; i>=0; i--){
            node[i] = f(node[i*2+1], node[i*2+2]);
        }
    }
    void update(int k, Monoid x){
        k += N-1;
        node[k] = x;
        while(k > 0){
            k = (k-1)/2;
            node[k] = f(node[2*k+1], node[2*k+2]);
        }
    }
    // [a,b)
    Monoid query(int a, int b){return query(a, b, 0, 0, N);}
    Monoid query(int a, int b, int k, int l, int r){
        if(b <= l || r <= a) return e;
        if(a <= l && r <= b) return node[k];
        Monoid vl, vr;
        vl = query(a, b, 2*k+1, l, (l+r)/2);
        vr = query(a, b, 2*k+2, (l+r)/2, r);
        return f(vl, vr);
    }
};

SegmentTree<ll> rmq([](ll a, ll b){return max(a,b);},0);

int main(){
    int N;
    cin >> N;
    vector<P> A(N);
    vector<P3> ord(N);
    vector<ll> B(N), C(N), rev(N);
    for(int i=0;i<N;i++){
        cin >> A[i].first >> B[i] >> C[i];
        A[i].second = i;;
    }
    sort(A.begin(), A.end());
    vector<ll> r(N);  // [0,r[i])
    for(int i=0;i<N;i++){
        int k = A[i].second;
        rev[k] = i;
        r[i] = upper_bound(A.begin(), A.end(), P(B[k]-C[k],N)) - A.begin();
        ord[i] = P3(P(r[i], A[i].first),k);
    }
    sort(ord.begin(), ord.end());
    rmq.build(r);
    vector<bool> ban(N);
    int banmin = N+1;
    for(int i=0;i<N;i++){
        if(r[i] > banmin){
            ban[i] = true;
        }
        else if(r[i] <= i){
            if(rmq.query(0,r[i]) > i){
                ban[i] = true;
                banmin = min(banmin, i);
            }
        }else{
            if(max(rmq.query(0,i), rmq.query(i+1,r[i])) > i){
                ban[i] = true;
                banmin = min(banmin, i);
            }
        }
    }
    vector<ll> ans(N), tmp(N);
    rmq.init(N);
    // TODO ちゃんとトポソ順に更新する
    for(int i=0;i<N;i++){
        int idx = ord[i].second;
        int k = rev[idx];
        if(ban[k]) continue;
        ans[idx] = rmq.query(0,r[k]) + B[idx];
        tmp[k] = ans[idx];

        rmq.update(k, ans[idx]);
    }
    for(int i=0;i<N;i++){
        if(ban[rev[i]]){
            cout << "BAN" << endl;
        }else{
            cout << ans[i] << endl;
        }
    }
    return 0;
}
0