#848036 (C++17) No.2242 Cities and Teleporters

提出ソース

結果

問題	No.2242 Cities and Teleporters
ユーザー	hotman78
提出日時	2023-03-10 22:57:37
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2,434 ms / 3,000 ms
コード長	9,118 bytes
コンパイル時間	4,357 ms
コンパイル使用メモリ	245,204 KB
実行使用メモリ	32,240 KB
最終ジャッジ日時	2023-10-18 08:40:43
合計ジャッジ時間	25,739 ms
ジャッジサーバーID （参考情報）	judge13 / judge12

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 4,348 KB
testcase_01	AC	2 ms 4,348 KB
testcase_02	AC	2 ms 4,348 KB
testcase_03	AC	2 ms 4,348 KB
testcase_04	AC	2 ms 4,348 KB
testcase_05	AC	1,296 ms 32,240 KB
testcase_06	AC	837 ms 32,240 KB
testcase_07	AC	1,276 ms 32,240 KB
testcase_08	AC	2,323 ms 32,240 KB
testcase_09	AC	2,434 ms 32,240 KB
testcase_10	AC	359 ms 32,240 KB
testcase_11	AC	403 ms 32,240 KB
testcase_12	AC	411 ms 32,240 KB
testcase_13	AC	520 ms 32,240 KB
testcase_14	AC	830 ms 32,240 KB
testcase_15	AC	535 ms 32,240 KB
testcase_16	AC	822 ms 32,240 KB
testcase_17	AC	1,492 ms 32,240 KB
testcase_18	AC	697 ms 31,712 KB
testcase_19	AC	728 ms 31,712 KB
testcase_20	AC	496 ms 30,920 KB
testcase_21	AC	431 ms 30,920 KB
testcase_22	AC	637 ms 30,920 KB
testcase_23	AC	752 ms 32,240 KB
testcase_24	AC	693 ms 32,240 KB
testcase_25	AC	856 ms 32,240 KB

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

ソースコード

raw source code

// author: hotman78
// date: 2023/03/10-22:57:20

// --- begin raw code -----------------
// #include"cpplib/util/template.hpp"
// 
// int main(){
//     int n;
//     cin>>n;
//     vector<int> h(n),t(n);
//     rep(i,n)cin>>h[i];
//     rep(i,n)cin>>t[i];
//     int q;
//     cin>>q;
//     vector<tuple<int,int,int>>v(n);
//     vector<int> v2(n);
//     {
//         rep(i,n)v[i]=make_tuple(h[i],-t[i],i);
//         int cnt=-1,idx=-1;
//         sort(all(v));
//         rep(i,n){
//             if(chmax(cnt,-get<1>(v[i])))idx=get<2>(v[i]);
//             v2[i]=idx;
//         }
// 
//     }
//     vector<vector<int>> tb(30,vector<int>(n));
//     rep(i,n){
//         int idx=lower_bound(all(v),make_tuple(t[i],IINF,IINF))-v.begin()-1;
//         if(idx==-1)tb[0][i]=i;
//         else tb[0][i]=v2[idx];
//     }
//     rep(j,29){
//         rep(i,n){
//             tb[j+1][i]=tb[j][tb[j][i]];
//         }
//     }
//     auto get=[&](int x,int fst)->int
//     {
//         int cnt=0,res=fst;
//         while(x){
//             if(x%2){
//                 res=tb[cnt][res];
//             }
//             x/=2;
//             cnt++;
//         }
//         return res;  
//     };
//     while(q--){
//         int p,q;
//         cin>>p>>q;
//         p--;q--;
//         if(t[p]>=h[q]){
//                 cout<<1<<endl;
//         }else{
//             int ng=0,ok=1LL<<20;
//             while(ok-ng>1){
//                 lint m=(ok+ng)/2;
//                 if(t[get(m,p)]>=h[q]){
//                     ok=m;
//                 }else{
//                     ng=m;
//                 }
//             }
//             if(ok>=1LL<<19){
//                 ok=-2;
//             }
//             cout<<ok+1<<endl;
//         }
//     }
// }
// --- end raw code -----------------

#line 2 "cpplib/util/template.hpp"
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx2")
#include <bits/stdc++.h>
using namespace std;
#line 1 "cpplib/util/ioutil.hpp"
// template <class Head,class... Args>
// std::ostream& output(std::ostream& out,const Head& head,const Args&... args){
//     out>>head;
//     return output(head,args...);
// }
// template <class Head>
// std::ostream& output(std::ostream& out,const Head& head){
//     out>>head;
//     return out;
// }

template <typename T, typename E>
std::ostream &operator<<(std::ostream &out, std::pair<T, E> v) {
    out << "(" << v.first << "," << v.second << ")";
    return out;
}

// template <class... Args>
// ostream& operator<<(ostream& out,std::tuple<Args...>v){
//     std::apply(output,v);
//     return out;
// }
#line 8 "cpplib/util/template.hpp"
struct __INIT__ {
    __INIT__() {
        cin.tie(0);
        ios::sync_with_stdio(false);
        cout << fixed << setprecision(15);
    }
} __INIT__;
typedef long long lint;
constexpr long long INF = 1LL << 60;
constexpr int IINF = 1 << 30;
constexpr double EPS = 1e-10;
#ifndef REACTIVE
#define endl '\n';
#endif
typedef vector<lint> vec;
typedef vector<vector<lint>> mat;
typedef vector<vector<vector<lint>>> mat3;
typedef vector<string> svec;
typedef vector<vector<string>> smat;
template <typename T> using V = vector<T>;
template <typename T> using VV = V<V<T>>;
template <typename T> inline void output(T t) {
    bool f = 0;
    for (auto i : t) {
        cout << (f ? " " : "") << i;
        f = 1;
    }
    cout << endl;
}
template <typename T> inline void output2(T t) {
    for (auto i : t)
        output(i);
}
template <typename T> inline void debug(T t) {
    bool f = 0;
    for (auto i : t) {
        cerr << (f ? " " : "") << i;
        f = 1;
    }
    cerr << endl;
}
template <typename T> inline void debug2(T t) {
    for (auto i : t)
        debug(i);
}
#define loop(n) for (long long _ = 0; _ < (long long)(n); ++_)
#define _overload4(_1, _2, _3, _4, name, ...) name
#define __rep(i, a) repi(i, 0, a, 1)
#define _rep(i, a, b) repi(i, a, b, 1)
#define repi(i, a, b, c)                                                       \
    for (long long i = (long long)(a); i < (long long)(b); i += c)
#define rep(...) _overload4(__VA_ARGS__, repi, _rep, __rep)(__VA_ARGS__)
#define _overload3_rev(_1, _2, _3, name, ...) name
#define _rep_rev(i, a) repi_rev(i, 0, a)
#define repi_rev(i, a, b)                                                      \
    for (long long i = (long long)(b)-1; i >= (long long)(a); --i)
#define rrep(...) _overload3_rev(__VA_ARGS__, repi_rev, _rep_rev)(__VA_ARGS__)

#define all(n) begin(n), end(n)
template <typename T, typename E> bool chmin(T &s, const E &t) {
    bool res = s > t;
    s = min<T>(s, t);
    return res;
}
template <typename T, typename E> bool chmax(T &s, const E &t) {
    bool res = s < t;
    s = max<T>(s, t);
    return res;
}
const vector<lint> dx = {1, 0, -1, 0, 1, 1, -1, -1};
const vector<lint> dy = {0, 1, 0, -1, 1, -1, 1, -1};
#define SUM(v) accumulate(all(v), 0LL)
#if __cplusplus >= 201703L
template <typename T, typename... Args>
auto make_vector(T x, int arg, Args... args) {
    if constexpr (sizeof...(args) == 0)
        return vector<T>(arg, x);
    else
        return vector(arg, make_vector<T>(x, args...));
}
#endif
#define extrep(v, ...) for (auto v : __MAKE_MAT__({__VA_ARGS__}))
#define bit(n, a) ((n >> a) & 1)
vector<vector<long long>> __MAKE_MAT__(vector<long long> v) {
    if (v.empty())
        return vector<vector<long long>>(1, vector<long long>());
    long long n = v.back();
    v.pop_back();
    vector<vector<long long>> ret;
    vector<vector<long long>> tmp = __MAKE_MAT__(v);
    for (auto e : tmp)
        for (long long i = 0; i < n; ++i) {
            ret.push_back(e);
            ret.back().push_back(i);
        }
    return ret;
}
using graph = vector<vector<int>>;
template <typename T> using graph_w = vector<vector<pair<int, T>>>;

#if __cplusplus >= 201703L
constexpr inline long long powll(long long a, long long b) {
    long long res = 1;
    while (b--)
        res *= a;
    return res;
}
#endif

template <typename T, typename E>
pair<T, E> &operator+=(pair<T, E> &s, const pair<T, E> &t) {
    s.first += t.first;
    s.second += t.second;
    return s;
}
template <typename T, typename E>
pair<T, E> &operator-=(pair<T, E> &s, const pair<T, E> &t) {
    s.first -= t.first;
    s.second -= t.second;
    return s;
}
template <typename T, typename E>
pair<T, E> operator+(const pair<T, E> &s, const pair<T, E> &t) {
    auto res = s;
    return res += t;
}
template <typename T, typename E>
pair<T, E> operator-(const pair<T, E> &s, const pair<T, E> &t) {
    auto res = s;
    return res -= t;
}
#define BEGIN_STACK_EXTEND(size)                                               \
    void *stack_extend_memory_ = malloc(size);                                 \
    void *stack_extend_origin_memory_;                                         \
    char *stack_extend_dummy_memory_ = (char *)alloca(                         \
        (1 + (int)(((long long)stack_extend_memory_) & 127)) * 16);            \
    *stack_extend_dummy_memory_ = 0;                                           \
    asm volatile("mov %%rsp, %%rbx\nmov %%rax, %%rsp"                          \
                 : "=b"(stack_extend_origin_memory_)                           \
                 : "a"((char *)stack_extend_memory_ + (size)-1024));
#define END_STACK_EXTEND                                                       \
    asm volatile("mov %%rax, %%rsp" ::"a"(stack_extend_origin_memory_));       \
    free(stack_extend_memory_);
#line 2 "main.cpp"

int main() {
    int n;
    cin >> n;
    vector<int> h(n), t(n);
    rep(i, n) cin >> h[i];
    rep(i, n) cin >> t[i];
    int q;
    cin >> q;
    vector<tuple<int, int, int>> v(n);
    vector<int> v2(n);
    {
        rep(i, n) v[i] = make_tuple(h[i], -t[i], i);
        int cnt = -1, idx = -1;
        sort(all(v));
        rep(i, n) {
            if (chmax(cnt, -get<1>(v[i])))
                idx = get<2>(v[i]);
            v2[i] = idx;
        }
    }
    vector<vector<int>> tb(30, vector<int>(n));
    rep(i, n) {
        int idx =
            lower_bound(all(v), make_tuple(t[i], IINF, IINF)) - v.begin() - 1;
        if (idx == -1)
            tb[0][i] = i;
        else
            tb[0][i] = v2[idx];
    }
    rep(j, 29) {
        rep(i, n) { tb[j + 1][i] = tb[j][tb[j][i]]; }
    }
    auto get = [&](int x, int fst) -> int {
        int cnt = 0, res = fst;
        while (x) {
            if (x % 2) {
                res = tb[cnt][res];
            }
            x /= 2;
            cnt++;
        }
        return res;
    };
    while (q--) {
        int p, q;
        cin >> p >> q;
        p--;
        q--;
        if (t[p] >= h[q]) {
            cout << 1 << endl;
        } else {
            int ng = 0, ok = 1LL << 20;
            while (ok - ng > 1) {
                lint m = (ok + ng) / 2;
                if (t[get(m, p)] >= h[q]) {
                    ok = m;
                } else {
                    ng = m;
                }
            }
            if (ok >= 1LL << 19) {
                ok = -2;
            }
            cout << ok + 1 << endl;
        }
    }
}

yukicoder

結果

テストケース

ソースコード