結果

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

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N;
  cin>>N;
  vector<int>X(N),A(N);
  map<int,int>iru;
  for(int i=0;i<N;i++){
    cin>>X[i];
    iru[X[i]]=i;
  }
  for(int &i:A)cin>>i;
  vector<vector<int>>G(N);
  for(int i=0;i<N;i++){
    if(iru.find(X[i]+A[i])!=iru.end()){
      G[iru[X[i]+A[i]]].push_back(i);
    }
    if(iru.find(X[i]-A[i])!=iru.end()){
      G[iru[X[i]-A[i]]].push_back(i);
    }
  }
  vector<array<int,2>>D(N);
  for(int i=0;i<N;i++){
    D[i]={X[i]+A[i],i};
  }
  sort(D.begin(),D.end(),greater{});
  vector<bool>vst(N);
  vector<int>ans(N);
  for(int i=0;i<N;i++){
    if(vst[D[i][1]])continue;
    vst[D[i][1]]=true;
    queue<int>bfs;
    bfs.push(D[i][1]);
    ans[D[i][1]]=A[D[i][1]];
    while(bfs.size()){
      int v=bfs.front();
      bfs.pop();
      for(int j:G[v]){
        if(!vst[j]){
          vst[j]=true;
          bfs.push(j);
          ans[j]=D[i][0]-X[j];
        }
      }
    }
  }
  for(int i:ans)cout<<i<<'\n';
}
0