結果

問題 No.1170 Never Want to Walk
ユーザー pes_magic
提出日時 2020-08-14 21:49:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,740 bytes
コンパイル時間 880 ms
コンパイル使用メモリ 79,744 KB
実行使用メモリ 7,552 KB
最終ジャッジ日時 2024-10-10 14:59:44
合計ジャッジ時間 7,011 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 26 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int main(){
    int N, A, B;
    while(cin >> N >> A >> B){
        vector<int> x(N);
        for(int i=0;i<N;i++) cin >> x[i];
        auto getIdx = [&](int val){
            int L = -1, R = N;
            while(R-L > 1){
                int mid = (L+R)/2;
                if(x[mid] < val) L = mid;
                else R = mid;
            }
            return R;
        };
        vector<int> res(N, 0);
        vector<int> visit(N, -1);
        for(int i=0;i<N;i++){
            if(visit[i] != -1) continue;
            queue<int> qu; qu.push(i);
            vector<int> set;
            visit[i] = i;
            while(!qu.empty()){
                int p = qu.front(); qu.pop();
                set.push_back(p);
                int ms = getIdx(x[p]-B), me = getIdx(x[p]-A+1);
                int ps = getIdx(x[p]+A), pe = getIdx(x[p]+B+1);
                for(int j=ms;j<me;j++){
                    if(visit[j] != -1) break;
                    visit[j] = i;
                    qu.push(j);
                }
                for(int j=me-1;j>=ms;j--){
                    if(visit[j] != -1) break;
                    visit[j] = i;
                    qu.push(j);
                }
                for(int j=ps;j<pe;j++){
                    if(visit[j] != -1) break;
                    visit[j] = i;
                    qu.push(j);
                }
                for(int j=pe-1;j>=ps;j--){
                    if(visit[j] != -1) break;
                    visit[j] = i;
                    qu.push(j);
                }
            }
            for(auto& t : set) res[t] = set.size();
        }
        for(auto& t : res) cout << t << endl;
    }
}
0