結果

問題 No.1170 Never Want to Walk
ユーザー hiro71687k
提出日時 2023-02-27 05:10:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 491 ms / 2,000 ms
コード長 1,531 bytes
コンパイル時間 4,551 ms
コンパイル使用メモリ 258,400 KB
最終ジャッジ日時 2025-02-10 23:51:03
ジャッジサーバーID
（参考情報） 		judge3 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using ld=long double;
ll inf=10010010010010010;
ll mod=998244353;
int main(){
    ll n,a,b;
    cin >> n >> a >> b;
    vector<ll>x(n);
    set<pair<ll,ll>>s;
    s.insert({inf,inf});
    for (ll i = 0; i < n; i++)
    {
        cin >> x[i];
        s.insert({x[i],i});
    }
    vector<ll>ans(n);
    while (1)
    {
        auto v=s.lower_bound({-1,-1});
        pair<ll,ll>p=*(v);
        s.erase(p);
        if (p.first==inf)
        {
            break;
        }
        vector<pair<ll,ll>>memo;
        memo.push_back(p);
        for (ll i = 0; i < memo.size(); i++)
        {
            v=s.lower_bound({memo[i].first+a,-1});
            pair<ll,ll>q=*(v);
            while (a<=abs(q.first-memo[i].first)&&abs(q.first-memo[i].first)<=b)
            {
                s.erase(q);
                memo.push_back(q);
                v=s.lower_bound(q);
                q=*(v);
            }
            v=s.lower_bound({memo[i].first-b,-1});
            q=*(v);
            while (a<=abs(memo[i].first-q.first)&&abs(memo[i].first-q.first)<=b)
            {
                s.erase(q);
                memo.push_back(q);
                v=s.lower_bound(q);
                q=*(v);
            }
        }
        for (ll i = 0; i < memo.size(); i++)
        {
            ans[memo[i].second]=memo.size();
        }
    }
    for (ll i = 0; i < ans.size(); i++)
    {
        cout << ans[i] << endl;
    }
    
}
0