#961507 (C++17) No.2690 A present from B (Hard)

提出ソース

結果

問題	No.2690 A present from B (Hard)
ユーザー	👑 Nachia
提出日時	2024-03-17 23:24:11
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	254 ms / 2,000 ms
コード長	5,846 bytes
コンパイル時間	1,208 ms
コンパイル使用メモリ	104,092 KB
実行使用メモリ	16,000 KB
最終ジャッジ日時	2024-03-17 23:24:17
合計ジャッジ時間	5,531 ms
ジャッジサーバーID （参考情報）	judge11 / judge13

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,548 KB
testcase_01	AC	2 ms 6,548 KB
testcase_02	AC	2 ms 6,548 KB
testcase_03	AC	2 ms 6,548 KB
testcase_04	AC	2 ms 6,548 KB
testcase_05	AC	2 ms 6,548 KB
testcase_06	AC	231 ms 16,000 KB
testcase_07	AC	254 ms 16,000 KB
testcase_08	AC	234 ms 16,000 KB
testcase_09	AC	11 ms 6,548 KB
testcase_10	AC	13 ms 6,548 KB
testcase_11	AC	11 ms 6,548 KB
testcase_12	AC	2 ms 6,548 KB
testcase_13	AC	2 ms 6,548 KB
testcase_14	AC	1 ms 6,548 KB
testcase_15	AC	2 ms 6,548 KB
testcase_16	AC	2 ms 6,548 KB
testcase_17	AC	197 ms 9,856 KB
testcase_18	AC	101 ms 15,872 KB
testcase_19	AC	56 ms 15,744 KB
testcase_20	AC	181 ms 6,784 KB
testcase_21	AC	44 ms 15,744 KB
testcase_22	AC	40 ms 9,600 KB
testcase_23	AC	95 ms 15,744 KB
testcase_24	AC	121 ms 6,656 KB
testcase_25	AC	36 ms 15,744 KB
testcase_26	AC	152 ms 15,872 KB
testcase_27	AC	177 ms 16,000 KB
testcase_28	AC	176 ms 16,000 KB
testcase_29	AC	184 ms 16,000 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#ifdef NACHIA
#define _GLIBCXX_DEBUG
#else
#define NDEBUG
#endif
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <utility>
#include <queue>
#include <array>
#include <cmath>
#include <atcoder/modint>
using i64 = long long;
using u64 = unsigned long long;
#define rep(i,n) for(i64 i=0; i<(i64)(n); i++)
#define repr(i,n) for(i64 i=(i64)(n)-1; i>=0; i--)
const i64 INF = 1001001001001001001;
const char* yn(bool x){ return x ? "Yes" : "No"; }
template<typename A> void chmin(A& l, const A& r){ if(r < l) l = r; }
template<typename A> void chmax(A& l, const A& r){ if(l < r) l = r; }
template<typename A> using nega_queue = std::priority_queue<A,std::vector<A>,std::greater<A>>;
using Modint = atcoder::static_modint<998244353>;
// #include "nachia/vec.hpp"
using namespace std;


namespace nachia{

template<
    class S,
    class F,
    S op(S l, S r),
    F composition(F f, F x),
    S mapping(F f, S x)
>
struct LazySegtree {
private:

    struct Node { S s; F f; bool propagated; };
    int N;
    int logN;
    int xN;
    std::vector<Node> A;

    void mapf(Node& a, F f){
        a.propagated = false;
        a.f = composition(f, a.f);
        a.s = mapping(f, a.s);
    }
    void mergev(int i){
        if(i<N) A[i].s = op(A[i*2].s, A[i*2+1].s);
    }
    void spread(int i){
        if(A[i].propagated || !(i < N)) return;
        mapf(A[i*2], A[i].f);
        mapf(A[i*2+1], A[i].f);
        A[i].f = A[0].f;
        A[i].propagated = true;
    }
    
    // bool cmp(S)
    template<class E>
    int minLeft2(int r, E cmp, int a = 0, int b = 0, int i = -1){
        static S x;
        if(i == -1){ a=0; b=N; i=1; x = A[0].s; }
        if(r <= a) return a;
        if(b <= r){
            S nx = op(A[i].s, x);
            if(cmp(nx)){ x = nx; return a; }
        }
        if(b-a == 1) return b;
        spread(i);
        int q = minLeft2(r, cmp, (a+b)/2, b, i*2+1);
        if(q > (a+b)/2) return q;
        return minLeft2(r, cmp, a, (a+b)/2, i*2);
    }
    // bool cmp(S)
    template<class E>
    int maxRight2(int l, E cmp, int a = 0, int b = 0, int i = -1){
        static S x;
        if(i == -1){ a=0; b=N; i=1; x = A[0].s; }
        if(b <= l) return b;
        if(l <= a){
            S nx = op(x, A[i].s);
            if(cmp(nx)){ x = nx; return b; }
        }
        if(b - a == 1) return a;
        spread(i);
        int q = maxRight2(l, cmp, a, (a+b)/2, i*2);
        if(q < (a+b)/2) return q;
        return maxRight2(l, cmp, (a+b)/2, b, i*2+1);
    }
public:

    LazySegtree() : N(0), logN(-1), xN(0){}
    LazySegtree(int n, S e, F id){
        N=1; logN=0; xN=n;
        while(N<n){ N *= 2; logN++; }
        A.assign(N*2, { e, id, true });
    }
    LazySegtree(const std::vector<S>& a, S e, F id)
        : LazySegtree(a.size(), std::move(e), std::move(id)){
        for(std::size_t i=0; i<a.size(); i++) A[i+N].s = a[i];
        for(int i=N-1; i>=1; i--) mergev(i);
    }

    void set(int p, S x){
        p += N;
        for(int d=logN; d; d--) spread(p >> d);
        A[p].s = x;
        for(int d=1; d<=logN; d++) mergev(p >> d);
    }
    S get(int p){
        p += N;
        for(int d=logN; d; d--) spread(p >> d);
        return A[p].s;
    }
    void apply(int p, F f){ set(p, mapping(f, get(p))); }
    void apply(int l, int r, F f){
        if(!(l < r)) return;
        if(l == 0 && r == N){ mapf(A[1], f); return; }
        l += N; r += N;
        for(int d=logN; d; d--){
            if((l >> d) << d != l) spread(l >> d);
            if((r >> d) << d != r) spread(r >> d);
        }
        int lp = l, rp = r;
        while(l < r){
            if(l&1){ mapf(A[l++], f); } l /= 2;
            if(r&1){ mapf(A[--r], f); } r /= 2;
        }
        for(int d=1 ; d<=logN; d++){
            if((lp >> d) << d != lp) mergev(lp >> d);
            if((rp >> d) << d != rp) mergev(rp >> d);
        }
    }
    S prod(int l, int r){
        if(!(l < r)) return A[0].s;
        l += N; r += N;
        for(int d=logN; d; d--){
            if((l >> d) << d != l) spread(l >> d);
            if((r >> d) << d != r) spread(r >> d);
        }
        S q1 = A[0].s, q2 = A[0].s;
        while(l < r){
            if(l&1){ q1 = op(q1, A[l++].s); } l /= 2;
            if(r&1){ q2 = op(A[--r].s, q2); } r /= 2;
        }
        return op(q1, q2);
    }
    S allProd() const { return A[1].s; }

    // bool cmp(S)
    template<class E>
    int minLeft(int r, E cmp){
        return minLeft2(r, cmp);
    }

    // bool cmp(S)
    template<class E>
    int maxRight(int l, E cmp){
        int x = maxRight2(l, cmp);
        return x > xN ? xN : x;
    }
};

} // namespace nachia;

i64 lst_chmin(i64 f, i64 x){ return min(f, x); }
using RangeChminPointGet = nachia::LazySegtree<i64, i64, lst_chmin, lst_chmin, lst_chmin>;

void testcase(){
    int N, M; cin >> N >> M;
    RangeChminPointGet ds_ru(N, INF, INF);
    RangeChminPointGet ds_lu(N, INF, INF);
    auto getVal = [&](i64 i){ return min(ds_ru.get(i) + i, ds_lu.get(i) - i); };
    auto applyVal = [&](i64 i, i64 x){
        ds_ru.apply(i, N, x-i);
        ds_lu.apply(0, i+1, x+i);
    };
    vector<int> A(M); rep(i,M){ cin >> A[i]; A[i]--; }
    reverse(A.begin(), A.end());
    applyVal(0, 0);
    for(int a : A){
        auto qa = getVal(a);
        auto qb = getVal(a+1);
        ds_ru.set(a, INF); ds_lu.set(a, INF);
        ds_ru.set(a+1, INF); ds_lu.set(a+1, INF);
        if(a != 0) chmin(qb, getVal(a-1) + 1);
        if(a+2 < N) chmin(qa, getVal(a+2) + 1);
        applyVal(a, qb);
        applyVal(a+1, qa);
    }
    rep(i,N-1){
        if(i) cout << ' ';
        cout << getVal(i+1);
    } cout << '\n';
}

int main(){
    ios::sync_with_stdio(false); cin.tie(nullptr);
    #ifdef NACHIA
    int T; cin >> T; for(int t=0; t<T; T!=++t?(cout<<'\n'),0:0)
    #endif
    testcase();
    return 0;
}

yukicoder

結果

テストケース

ソースコード