結果

問題 No.1950 片道きゃっちぼーる
ユーザー HIcoderHIcoder
提出日時 2023-07-17 15:52:32
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,281 bytes
コンパイル時間 2,014 ms
コンパイル使用メモリ 145,672 KB
実行使用メモリ 75,996 KB
最終ジャッジ日時 2023-10-18 00:25:31
合計ジャッジ時間 27,470 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
4,348 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 AC 2 ms
4,348 KB
testcase_06 WA -
testcase_07 AC 706 ms
44,824 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
to do

有効グラフ


*/
#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,int> mp;
    vector<ll> 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]]=i;//X[i]に人iがいる
    }

    vector<vector<int>> G(N),rG(N);


    UnionFind uf(N);

    set<pair<int,int>> edges;

    for(int i=0;i<N;i++){
        //i番目の人から辺を張る

        if(mp.count(X[i]+A[i])){
            int j=mp[X[i]+A[i]];
            G[i].push_back(j);
            edges.insert( make_pair(i,j));
        } 
        if(mp.count(X[i]-A[i])){
            int j=mp[X[i]-A[i]];
            G[i].push_back(j);
            edges.insert( make_pair(i,j));
            
        } 
    }

    for(auto [i,j]:edges){
        cout<<"i="<<i<<".j="<<j<<endl;
    }



    for(int i=0;i<N;i++){
        //i番目の人から辺を張る

        if(mp.count(X[i]+A[i])){
            int j=mp[X[i]+A[i]];
            G[i].push_back(j);

            if(edges.count(make_pair(j,i))){
                uf.merge(j,i);

            }
        } 
        if(mp.count(X[i]-A[i])){
            int j=mp[X[i]-A[i]];
            G[i].push_back(j);
            if(edges.count(make_pair(j,i))){
                uf.merge(j,i);

            }
            
        } 
    }



    vector<bool> visit(N,false);
    vector<bool> seen(N,false);
    vector<bool> finish(N,false);
    vector<ll> dist(N,0);

    /*
    for(int i=0;i<N;i++){
        dist[i]=X[i]+A[i];
    }
    */

    map<int,ll> mpid;
    for(int i=0;i<N;i++){
       // cout<<"i="<<i<<",uf.find="<<uf.find(i)<<endl;

        mpid[uf.find(i)]=max(mpid[uf.find(i)],X[i]+A[i]);
    }

    vector<int> distid(N);
    for(int i=0;i<N;i++){
        dist[i]=max(dist[i],mpid[uf.find(i)]);
    }

    /*
    for(int i=0;i<N;i++){
        //i番目の人間が直接投げるか
        cout<<"dist["<<i<<"]="<<dist[i]<<endl;
        
    }

    for(auto [id,dis]:mpid){
        cout<<"id="<<id<<",dis="<<dis<<endl;
    }
    */



    /*
    auto dfs=[&](auto f,int now)->ll{
        //cout<<"now="<<now<<endl;

        if(seen[now] && !finish[now]){
            finish[now]=true;
            return dist[now];
        }
        seen[now]=true;

        for(int to:G[now]){
            if(finish[to]) continue;
            cout<<"now="<<now<<",to="<<to<<endl;
            
            dist[now]=max(dist[now],f(f,to));
        }

        finish[now]=true;
        return dist[now];
    };
    */
    auto dfs=[&](auto f,int now)->ll{
        //cout<<"now="<<now<<endl;
        /*
        if(seen[now]){
            finish[now]=true;
            return dist[now];
        }
        */
        if(seen[now]){
            return dist[now];
        }
        seen[now]=true;

        for(int to:G[now]){
            //if(seen[to]) continue;
            //cout<<"now="<<now<<",to="<<to<<endl;
            
            dist[now]=max(dist[now],f(f,to));
        }

        //finish[now]=true;
        return dist[now];
    };


    for(int i=0;i<N;i++){
        if(!seen[i]){
            //cout<<"i="<<i<<endl;
            dfs(dfs,i);
        }
    }

   

    vector<ll> ans(N);
    for(int i=0;i<N;i++){
        //i番目の人間が直接投げるか
        //cout<<"dist["<<i<<"]="<<dist[i]<<endl;
        ans[i]=dist[i]-X[i];
    }

    for(ll v:ans){
        cout<<v<<endl;
    }


}
0