結果

問題 No.1950 片道きゃっちぼーる
ユーザー 👑 Nachia
提出日時 2022-05-20 23:03:43
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 170 ms / 3,000 ms
コード長 1,707 bytes
コンパイル時間 3,467 ms
コンパイル使用メモリ 127,260 KB
最終ジャッジ日時 2025-01-29 11:28:40
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <array>
#include <atcoder/modint>
using namespace std;
using i32 = int32_t;
using u32 = uint32_t;
using i64 = int64_t;
using u64 = uint64_t;
#define rep(i,n) for(int i=0; i<(int)(n); i++)

const i64 INF = 1001001001001001001;

using modint = atcoder::static_modint<998244353>;


int main(){
    int N; cin >> N;
    vector<int> X(N);
    rep(i,N) cin >> X[i];
    vector<int> A(N);
    rep(i,N) cin >> A[i];

    auto search_node = [&](int x) -> int {
        int p = lower_bound(X.begin(), X.end(), x) - X.begin();
        if(p == N) return -1;
        if(X[p] == x) return p;
        return -1;
    };

    vector<vector<int>> edges(N);
    rep(i,N){
        int to = search_node(X[i] - A[i]);
        if(to >= 0) edges[to].push_back(i);
    }
    rep(i,N){
        int to = search_node(X[i] + A[i]);
        if(to >= 0) edges[to].push_back(i);
    }

    vector<pair<int,int>> Q(N);
    rep(i,N) Q.push_back(make_pair(X[i]+A[i], i));
    sort(Q.begin(), Q.end());
    reverse(Q.begin(), Q.end());

    vector<int> ans(N, -1);
    vector<int> bfs;
    int bfsi = 0;
    for(auto [sd,s] : Q) if(ans[s] == -1){
        ans[s] = sd;
        bfs.push_back(s);
        while(bfsi < (int)bfs.size()){
            int p = bfs[bfsi++];
            for(int e : edges[p]) if(ans[e] == -1){
                bfs.push_back(e);
                ans[e] = ans[p];
            }
        }
    }

    rep(i,N) ans[i] -= X[i];
    rep(i,N) cout << ans[i] << '\n';
    return 0;
}


struct ios_do_not_sync{
    ios_do_not_sync(){
        std::ios::sync_with_stdio(false);
        std::cin.tie(nullptr);
    }
} ios_do_not_sync_instance;


0