結果

問題 No.1950 片道きゃっちぼーる
ユーザー MazesobaMazesoba
提出日時 2022-05-25 06:37:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,150 ms / 3,000 ms
コード長 1,640 bytes
コンパイル時間 4,557 ms
コンパイル使用メモリ 262,288 KB
最終ジャッジ日時 2025-01-29 15:14:14
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>

using namespace std;

//#define DISABLE_PRINT

#if defined(ENABLE_PRINT) && !defined(DISABLE_PRINT)

#define P(...) fprintf(stderr, __VA_ARGS__)
#define P2(fmt) fprintf(stderr, fmt)
#define LP fprintf(stderr, "L: %d\n", __LINE__)

#else

#define P(...) ((void)0)
#define P2(fmt) ((void)0)
#define LP ((void)0)

#endif

#define rep(i, n) for(int i = 0; i < (int)(n); ++i)
#define ALL(x) x.begin(),x.end()

using ll = long long;
using ull = unsigned long long;

int main(int, const char**)
{
    int N; cin >> N;
    vector<vector<int>> g(N);
    vector<ll> X(N), A(N);
    map<ll, int> posToIdx;
    map<ll, vector<int>> distToIdx;
    rep(i, N) {
        cin >> X[i];
        posToIdx[X[i]] = i;
    }
    rep(i, N) cin >> A[i];
    rep(i, N) {
        auto d0 = X[i] + A[i];
        if(posToIdx.find(d0) != posToIdx.end()) {
            g[posToIdx[d0]].push_back(i);
        }
        distToIdx[d0].push_back(i);

        auto d1 = X[i] - A[i];
        if(posToIdx.find(d1) != posToIdx.end()) {
            g[posToIdx[d1]].push_back(i);
        }
    }

    vector<ll> ans(N, -1);
    function<void(int, ll)> f = [&](int x, ll d) {
        if(ans[x] != -1) return;
        ans[x] = d;
        assert(ans[x] >= X[x]);
        for(auto nx : g[x]) {
            f(nx, d);
        }
    };

    for(auto itr = distToIdx.rbegin(); itr != distToIdx.rend(); ++itr) {
        auto& l = itr->second;
        for(auto idx : l) {
            f(idx, itr->first);
        }
    }

    rep(i, N) {
        auto a = ans[i] - X[i];
        assert(a >= 0);
        cout << a << endl;
    }

    return 0;
}
0