結果

問題 No.1950 片道きゃっちぼーる
ユーザー take000
提出日時 2022-05-21 00:14:52
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,559 bytes
コンパイル時間 2,211 ms
コンパイル使用メモリ 199,352 KB
最終ジャッジ日時 2025-01-29 12:17:20
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < n; ++i)
typedef long long ll;
using namespace std;

int main() {
    cin.tie(0);
    ios_base::sync_with_stdio(false);

    int N;
    cin >> N;
    vector<ll> X(N + 1), A(N + 1);
    rep(i, N) cin >> X[i];
    rep(i, N) cin >> A[i];
    X[N] = 1e18, A[N] = 0;

    vector<bool> seen(N);
    vector<ll> ans(N);

    rep(i, N) {
        auto calc = [&](auto &&self, int idx) -> ll {
            if (seen[idx]) return ans[idx];
            seen[idx] = true;

            // 右に投げて終わり
            ans[idx] = A[idx];

            auto rid = lower_bound(X.begin(), X.end(), X[idx] + A[idx]) - X.begin();
            auto lid = lower_bound(X.begin(), X.end(), X[idx] - A[idx]) - X.begin();

            // 右に投げて更に投げる
            if (X[rid] == X[idx] + A[idx])
                ans[idx] = max(ans[idx], A[idx] + self(self, rid));

            // 左に投げて更に投げる
            if (X[lid] == X[idx] - A[idx])
                ans[idx] = max(ans[idx], -A[idx] + self(self, lid));

            // 右に投げて更に投げる
            if (X[rid] == X[idx] + A[idx])
                ans[idx] = max(ans[idx], A[idx] + self(self, rid));

            // 左に投げて更に投げる
            if (X[lid] == X[idx] - A[idx])
                ans[idx] = max(ans[idx], -A[idx] + self(self, lid));

            return ans[idx];
        };

        seen[i] = false;
        calc(calc, i);
    }

    for (ll a : ans)
        cout << a << "\n";

    return 0;
}
0