結果

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

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
#include <utility>
#include <set>
#include <map>
#include <queue>
#include <stack>
using namespace std;
int main()
{
    int N;
    cin >> N;
    vector<long long> X(N), A(N);
    for (int i = 0; i < N; i++)
        cin >> X[i];
    for (int i = 0; i < N; i++)
        cin >> A[i];
    vector<pair<long long, int>> a(N + 1);
    long long INF = 1000000000000000;
    vector<vector<int>> G(N);
    X.push_back(INF);
    for (int i = 0; i < N; i++)
    {
        int id = lower_bound(X.begin(), X.end(), A[i] + X[i]) - X.begin();
        if (X[id] == A[i] + X[i])
        {
            G[id].push_back(i);
        }
        id = lower_bound(X.begin(), X.end(), -A[i] + X[i]) - X.begin();
        if (X[id] == -A[i] + X[i])
        {
            G[id].push_back(i);
        }
    }
    priority_queue<pair<long long, int>> q;
    vector<long long> d(N);
    for (int i = 0; i < N; i++)
    {
        d[i] = A[i] + X[i];
        q.push({A[i] + X[i], i});
    }
    while (!q.empty())
    {
        auto [num, id] = q.top();
        q.pop();
        if (d[id] > num)
            continue;
        for (int to : G[id])
        {
            if (d[to] < num)
            {
                d[to] = num;
                q.push({num, to});
            }
        }
    }
    for (int i = 0; i < N; i++)
        cout << d[i] - X[i] << endl;
}
0