結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー hiro71687k
提出日時 2023-02-27 05:01:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,652 bytes
コンパイル時間 3,476 ms
コンパイル使用メモリ 257,324 KB
最終ジャッジ日時 2025-02-10 23:50:19
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 27 WA * 10
権限があれば一括ダウンロードができます

ソースコード

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);
            if (q.first==inf)
            {
                continue;
            }
            while (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);
            if (q.first==inf)
            {
                continue;
            }
            while (a<=memo[i].first-q.first&&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