結果
問題 |
No.1950 片道きゃっちぼーる
|
ユーザー |
|
提出日時 | 2023-07-16 23:40:41 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,796 bytes |
コンパイル時間 | 1,637 ms |
コンパイル使用メモリ | 133,044 KB |
最終ジャッジ日時 | 2025-02-15 15:21:54 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 17 WA * 6 |
ソースコード
#include<iostream> #include<set> #include<algorithm> #include<vector> #include<string> #include<set> #include<map> #include<numeric> #include<queue> #include<cmath> #include <unordered_set> using namespace std; typedef long long ll; const ll INF=1LL<<60; typedef pair<ll,ll> P; typedef pair<int,P> PP; const ll MOD=1e9+7; class UnionFind{ public: std::vector<int> par; std::vector<int> rank; std::vector<int> sz;//sz[i]で,頂点iが含まれているグループのサイズ std::vector<int> root; UnionFind(int size){ rank=std::vector<int>(size+1,0); par = std::vector<int>(size+1,0); std::iota(par.begin(),par.end(),0);//#include<numeric> sz = std::vector<int>(size+1,1); root=std::vector<int>(size+1); std::iota(root.begin(),root.end(),0); } ~UnionFind(){ std::vector<int>().swap(rank); std::vector<int>().swap(par); std::vector<int>().swap(sz); std::vector<int>().swap(root); } int find(int x){ if(par[x]==x)return x; else return par[x] = find(par[x]); } bool issame(int u,int v){ return find(u)==find(v); } void merge(int u,int v){ u = find(u); v = find(v); if(u==v) return; if(rank[u]<rank[v]) std::swap(u,v); //uの傘下へvが入る //rank[u]>=rank[v] par[v]=u; sz[u]+=sz[v]; if(rank[u]==rank[v])rank[u]++; } int size(int u){//頂点uが属すグループの大きさを表す. u=find(u); return sz[u]; } std::map<int,std::vector<int>> element(){ std::map<int,std::vector<int>> mp; for(int i=0;i<par.size();i++){ root[i]=find(i); mp[root[i]].push_back(i); } return mp; } }; int main(){ int N; cin>>N; map<int,P> mp; vector<int> X(N); vector<ll> A(N); for(int i=0;i<N;i++){ cin>>X[i]; } for(int i=0;i<N;i++){ cin>>A[i]; } for(int i=0;i<N;i++){ mp[X[i]]=make_pair(A[i],i); } UnionFind uf(N); for(int i=0;i<N;i++){ if(mp.count(X[i]+A[i])){ uf.merge(i,mp[X[i]+A[i]].second); } if(mp.count(X[i]-A[i])){ uf.merge(i,mp[X[i]-A[i]].second); } } map<int,vector<int>> mpid; for(int i=0;i<N;i++){ mpid[uf.find(i)].push_back(i); } map<ll,ll> dist; for(auto& [val,vec]:mpid){ ll cand=0; for(int v:vec){ cand=max(cand,X[v]+A[v]); } dist[val]=cand; } vector<ll> ans(N); for(int i=0;i<N;i++){ ans[i]=dist[uf.find(i)] - X[i]; } for(ll v:ans){ cout<<v<<endl; } }