結果

問題 No.1965 Heavier
ユーザー srjywrdnprktsrjywrdnprkt
提出日時 2023-05-18 00:20:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 872 ms / 2,000 ms
コード長 3,057 bytes
コンパイル時間 2,216 ms
コンパイル使用メモリ 137,892 KB
実行使用メモリ 52,084 KB
最終ジャッジ日時 2024-05-09 07:31:01
合計ジャッジ時間 18,694 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 731 ms
47,116 KB
testcase_07 AC 718 ms
46,980 KB
testcase_08 AC 490 ms
52,048 KB
testcase_09 AC 689 ms
34,972 KB
testcase_10 AC 690 ms
35,000 KB
testcase_11 AC 687 ms
35,048 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 745 ms
35,132 KB
testcase_16 AC 755 ms
35,192 KB
testcase_17 AC 572 ms
28,436 KB
testcase_18 AC 826 ms
35,860 KB
testcase_19 AC 723 ms
45,492 KB
testcase_20 AC 758 ms
46,464 KB
testcase_21 AC 660 ms
43,184 KB
testcase_22 AC 657 ms
42,636 KB
testcase_23 AC 444 ms
29,020 KB
testcase_24 AC 591 ms
39,024 KB
testcase_25 AC 354 ms
52,068 KB
testcase_26 AC 346 ms
52,084 KB
testcase_27 AC 872 ms
47,512 KB
testcase_28 AC 871 ms
47,504 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cmath>
#include <map>
#include <set>
#include <iomanip>
#include <queue>
#include <algorithm>
#include <numeric>
#include <deque>
#include <complex>

using namespace std;

template<class S, S (*op)(S, S), S (*e)()>struct SegTree {
    vector<S> seg;
    long long N = 1;

    SegTree (long long n) : SegTree(vector<S>(n, e())) {}
    SegTree (const vector<S> &v){
        long long n = v.size();
        while (N < n) N *= 2;
        seg.resize(N*2-1, e());
        for (long long i=0; i<n; i++) seg[i+N-1] = v[i];
        for (long long i=N-2; i>=0; i--){
            seg[i] = op(seg[i*2+1], seg[i*2+2]);
        }
    }

    void set(long long loc, S val){
        loc += N-1;
        seg[loc] = val;
        while (loc != 0){
            loc = (loc-1)/2;
            seg[loc] = op(seg[loc*2+1], seg[loc*2+2]);
        }
    }

    //op(a[l], ..., a[r])
    S prod (long long l, long long r) const{
        return _prod(l, r, 0, 0, N-1);
    }

    S all_prod() const{
        return seg[0];
    }

    S _prod (long long l, long long r, long long idx, long long bitl, long long bitr) const{
        if (r < bitl || l > bitr) return e();
        if (l <= bitl && bitr <= r) return seg[idx];

        long long bitm = (bitl+bitr)/2;
        return op(_prod(l, r, idx*2+1, bitl, bitm), _prod(l, r, idx*2+2, bitm+1, bitr));
    }

    S get (long long i) const{
        return seg[i+N-1];
    }

    void show() const{
        for (int i=N-1; i<N*2-1; i++) cout << seg[i] << " ";
        cout << endl;
    }
};

long long op(long long a, long long b){
    return max(a, b);
}

const long long INF = (1LL<<62) - 1;

long long e(){
    return -INF;
}

template <typename T>
map<T, T> compress(vector<T> &A){

    map<T, T> comp;
    int N = A.size(), i=0;
    for (int i=0; i<N; i++) comp[A[i]];

    for (auto &e : comp){
        e.second = i;
        i++;
    }

    return comp;
}

using ll = long long;
 
int main(){

    ll N, l=0, ans=0;
    cin >> N;
    vector<ll> A(N), B(N);
    for (int i=0; i<N; i++) cin >> A[i];
    for (int i=0; i<N; i++) cin >> B[i];

    vector<multiset<ll>> v(N), w(N);
    map<ll, ll> mp = compress(B);
    SegTree<long long, op, e> mi(N+1), pl(N+1);

    auto judge = [&](auto a, auto b){
        return (pl.prod(mp[b], N) < a+b) && (mi.prod(0, mp[b]-1) < a-b);
    };

    for (int r=0; r<N; r++){
        while(l<r && !judge(A[r], B[r])){
            v[mp[B[l]]].erase(v[mp[B[l]]].find(A[l]-B[l]));
            w[mp[B[l]]].erase(w[mp[B[l]]].find(A[l]+B[l]));
            if (v[mp[B[l]]].size() == 0) mi.set(mp[B[l]], e());
            else mi.set(mp[B[l]], *v[mp[B[l]]].rbegin());
            if (w[mp[B[l]]].size() == 0) pl.set(mp[B[l]], e());
            else pl.set(mp[B[l]], *w[mp[B[l]]].rbegin());
            l++;
        }
        if (judge(A[r], B[r])) ans += r-l;
        mi.set(mp[B[r]], A[r]-B[r]);
        pl.set(mp[B[r]], A[r]+B[r]);
        v[mp[B[r]]].insert(A[r]-B[r]);
        w[mp[B[r]]].insert(A[r]+B[r]);
    }

    cout << ans << endl;
    return 0;
}
0