#752248 (C++14) No.1382 Travel in Mitaru city

提出ソース

結果

問題	No.1382 Travel in Mitaru city
ユーザー	t98slider
提出日時	2022-04-08 04:07:50
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	111 ms / 2,000 ms
コード長	7,177 bytes
コンパイル時間	2,483 ms
コンパイル使用メモリ	190,084 KB
実行使用メモリ	16,640 KB
最終ジャッジ日時	2024-05-06 01:12:36
合計ジャッジ時間	9,557 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	1 ms 5,248 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	33 ms 5,760 KB
testcase_07	AC	26 ms 5,504 KB
testcase_08	AC	17 ms 5,376 KB
testcase_09	AC	35 ms 5,888 KB
testcase_10	AC	35 ms 5,760 KB
testcase_11	AC	72 ms 11,392 KB
testcase_12	AC	94 ms 11,264 KB
testcase_13	AC	70 ms 11,520 KB
testcase_14	AC	62 ms 11,136 KB
testcase_15	AC	64 ms 11,264 KB
testcase_16	AC	105 ms 16,256 KB
testcase_17	AC	107 ms 16,256 KB
testcase_18	AC	107 ms 16,256 KB
testcase_19	AC	110 ms 16,256 KB
testcase_20	AC	106 ms 16,256 KB
testcase_21	AC	104 ms 16,256 KB
testcase_22	AC	111 ms 16,256 KB
testcase_23	AC	108 ms 16,256 KB
testcase_24	AC	100 ms 16,256 KB
testcase_25	AC	103 ms 16,384 KB
testcase_26	AC	74 ms 11,904 KB
testcase_27	AC	73 ms 12,032 KB
testcase_28	AC	73 ms 12,028 KB
testcase_29	AC	71 ms 12,032 KB
testcase_30	AC	59 ms 11,384 KB
testcase_31	AC	70 ms 10,368 KB
testcase_32	AC	90 ms 11,776 KB
testcase_33	AC	81 ms 11,136 KB
testcase_34	AC	56 ms 8,832 KB
testcase_35	AC	37 ms 7,040 KB
testcase_36	AC	45 ms 10,368 KB
testcase_37	AC	8 ms 5,376 KB
testcase_38	AC	48 ms 11,008 KB
testcase_39	AC	13 ms 5,504 KB
testcase_40	AC	36 ms 9,216 KB
testcase_41	AC	38 ms 9,984 KB
testcase_42	AC	47 ms 11,136 KB
testcase_43	AC	40 ms 10,240 KB
testcase_44	AC	18 ms 6,144 KB
testcase_45	AC	36 ms 9,600 KB
testcase_46	AC	21 ms 6,528 KB
testcase_47	AC	83 ms 14,308 KB
testcase_48	AC	45 ms 9,596 KB
testcase_49	AC	86 ms 15,004 KB
testcase_50	AC	32 ms 8,064 KB
testcase_51	AC	86 ms 14,592 KB
testcase_52	AC	104 ms 16,640 KB
testcase_53	AC	23 ms 6,528 KB
testcase_54	AC	44 ms 9,472 KB
testcase_55	AC	98 ms 16,128 KB
testcase_56	AC	31 ms 7,808 KB
testcase_57	AC	61 ms 11,904 KB
testcase_58	AC	13 ms 5,376 KB
testcase_59	AC	29 ms 7,552 KB
testcase_60	AC	93 ms 15,744 KB
testcase_61	AC	3 ms 5,376 KB
testcase_62	AC	2 ms 5,376 KB
testcase_63	AC	9 ms 5,376 KB
testcase_64	AC	3 ms 5,376 KB
testcase_65	AC	2 ms 5,376 KB
testcase_66	AC	3 ms 5,376 KB
testcase_67	AC	3 ms 5,376 KB
testcase_68	AC	4 ms 5,376 KB
testcase_69	AC	2 ms 5,376 KB
testcase_70	AC	5 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define drep(i,j,n) for(int i=0;i<(int)(n-1);i++)for(int j=i+1;j<(int)(n);j++)
#define trep(i,j,k,n) for(int i=0;i<(int)(n-2);i++)for(int j=i+1;j<(int)(n-1);j++)for(int k=j+1;k<(int)(n);k++)
#define codefor int test;scanf("%d",&test);while(test--)
#define INT(...) int __VA_ARGS__;in(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__;in(__VA_ARGS__)
#define yes(ans) if(ans)printf("yes\n");else printf("no\n")
#define Yes(ans) if(ans)printf("Yes\n");else printf("No\n")
#define YES(ans) if(ans)printf("YES\n");else printf("NO\n")
#define popcount(v) __builtin_popcountll(v)
#define vector2d(type,name,h,...) vector<vector<type>>name(h,vector<type>(__VA_ARGS__))
#define vector3d(type,name,h,w,...) vector<vector<vector<type>>>name(h,vector<vector<type>>(w,vector<type>(__VA_ARGS__)))
#define vector4d(type,name,h,w,d,...) vector<vector<vector<vector<type>>>>name(h,vector<vector<vector<type>>>(w,vector<vector<type>>(d,vector<type>(__VA_ARGS__))))
using namespace std;
using ll = long long;
template<class T> using rpriority_queue = priority_queue<T, vector<T>, greater<T>>;
const int MOD=1000000007;
const int MOD2=998244353;
const int INF=1<<30;
const ll INF2=1LL<<60;
void scan(int& a){scanf("%d",&a);}
void scan(long long& a){scanf("%lld",&a);}
template<class T,class L>void scan(pair<T, L>& p){scan(p.first);scan(p.second);}
template<class T,class U,class V>void scan(tuple<T,U,V>& p){scan(get<0>(p));scan(get<1>(p));scan(get<2>(p));}
template<class T, size_t size> void scan(array<T, size>& a){ for(auto&& i : a) scan(i);}
template<class T> void scan(T& a){cin>>a;}
template<class T> void scan(vector<T>& vec){for(auto&& it:vec)scan(it);}
void in(){}
template <class Head, class... Tail> void in(Head& head, Tail&... tail){scan(head);in(tail...);}
void print(const int& a){printf("%d",a);}
void print(const long long& a){printf("%lld",a);}
void print(const double& a){printf("%.15lf",a);}
template<class T,class L>void print(const pair<T, L>& p){print(p.first);putchar(' ');print(p.second);}
template<class T> void print(const T& a){cout<<a;}
template<class T> void print(const vector<T>& vec){if(vec.empty())return;print(vec[0]);for(auto it=vec.begin();++it!= vec.end();){putchar(' ');print(*it);}}
void out(){putchar('\n');}
template<class T> void out(const T& t){print(t);putchar('\n');}
template <class Head, class... Tail> void out(const Head& head,const Tail&... tail){print(head);putchar(' ');out(tail...);}
template<class T> void dprint(const T& a){cerr<<a;}
template<class T> void dprint(const vector<T>& vec){if(vec.empty())return;cerr<<vec[0];for(auto it=vec.begin();++it!= vec.end();){cerr<<" "<<*it;}}
void debug(){cerr<<'\n';}
template<class T> void debug(const T& t){dprint(t);cerr<<endl;}
template <class Head, class... Tail> void debug(const Head& head, const Tail&... tail){dprint(head);cerr<<" ";debug(tail...);}
ll intpow(ll a, ll b){ ll ans = 1; while(b){ if(b & 1) ans *= a; a *= a; b /= 2; } return ans; }
ll modpow(ll a, ll b, ll p){ ll ans = 1; while(b){ if(b & 1) (ans *= a) %= p; (a *= a) %= p; b /= 2; } return ans; }
ll modinv(ll a, ll m) {ll b = m, u = 1, v = 0;while (b) {ll t = a / b;a -= t * b; swap(a, b);u -= t * v; swap(u, v);}u %= m;if (u < 0) u += m;return u;}
ll updivide(ll a,ll b){return (a+b-1)/b;}
template<class T> void chmax(T &a,const T b){if(b>a)a=b;}
template<class T> void chmin(T &a,const T b){if(b<a)a=b;}

template<class T> struct compressed_array{
    int N;
    vector<T> dat;
    const T operator[](int i) const { return dat[i]; }
    T operator[](int i) { return dat[i]; }
    compressed_array(){}
    compressed_array(vector<int> &a){
        dat=a;
    }
    void build(){
        sort(dat.begin(), dat.end());
        dat.erase(unique(dat.begin(), dat.end()), dat.end());
        N = dat.size();
    }
    int size(){return N;}
    void insert(const T v){
        dat.push_back(v);
    }
    void insert(vector<T> &a){
        for(int i = 0; i < a.size() ; i++)dat.push_back(a[i]);
    }
    //type1でfirst,type2でsecond
    void insert(vector<pair<T, T>> &a, int type){
        if(type==1){
            for(int i = 0; i < a.size(); i++)dat.push_back(a[i].first);
        }else{
            for(int i = 0; i < a.size(); i++)dat.push_back(a[i].second);
        }
    }
    template<size_t size> void insert(vector<array<T,size>> &a, int pos){
        assert(0 <= pos && pos < size);
        for(int i = 0; i < a.size() ; i++){
            dat.push_back(a[i][pos]);
        }
    }
    int load(T &v){
        return lower_bound(dat.begin(), dat.end(), v) - dat.begin();
    }
    void load(vector<T> &a){
        for(int i = 0; i < a.size() ; i++)a[i] = load(a[i]);
    }
    void load(vector<pair<T, T>> &a, int type){
        if(type==1){
            for(int i = 0; i < a.size(); i++)a[i].first = load(a[i].first);
        }else{
            for(int i = 0; i < a.size(); i++)a[i].second = load(a[i].second);
        }
    }
    template<size_t size> void load(vector<array<T,size>> &a, int pos){
        assert(0 <= pos && pos < size);
        for(int i = 0; i < a.size() ; i++){
            a[i][pos] = load(a[i][pos]);
        }
    }
};

//Union Add Query
struct dsu{
    using U = long long;
    int n;
    vector<pair<int,U>> dat;
    dsu() : n(0) {}
    dsu(int _n) : n(_n), dat(_n, {-1, 0}) {}
    int merge(int a, int b) {
        assert(0 <= a && a < n);
        assert(0 <= b && b < n);
        int x = leader(a).first, y = leader(b).first;
        if (x == y) return x;
        if (-dat[x].first < -dat[y].first) std::swap(x, y);
        dat[x].first += dat[y].first;
        dat[y].first = x;
        dat[y].second -= dat[x].second;
        return x;
    }
    pair<int,U> leader(int a) {
        assert(0 <= a && a < n);
        if (dat[a].first < 0 ) return {a, 0};
        auto p = leader(dat[a].first);
        dat[a].first = p.first;
        dat[a].second += p.second;
        return dat[a];
    }
    void add(int a, U v){
        assert(0 <= a && a < n);
        dat[leader(a).first].second += v;
    }
    U get(int a){
        assert(0 <= a && a < n);
        auto p = leader(a);
        return p.second + dat[p.first].second;
    }
};

int main(){
    INT(n,m,s,t);
    s--,t--;
    dsu uf(n);
    vector<int> p(n);
    in(p);
    compressed_array<int> ca(p);
    ca.build();
    ca.load(p);
    vector<vector<int>> tb(ca.size());
    for(int i=0;i<n;i++){
        tb[p[i]].push_back(i);
    }
    vector<vector<int>> g(n);
    int u,v;
    for(int i=0;i<m;i++){
        in(u,v);
        u--,v--;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    vector<bool> used(n);
    for(int i=ca.size()-1;i>=0;i--){
        set<int> S;
        for(auto &&v:tb[i]){
            for(auto &&u:g[v]){
                if(!used[u])continue;
                S.insert(uf.leader(u).first);
            }
        }
        for(auto &&v:S)uf.add(v,1);
        for(auto &&v:tb[i]){
            for(auto &&u:g[v]){
                if(!used[u])continue;
                uf.merge(u,v);
            }
            used[v]=true;
        }
    }
    out(uf.get(s));
}

yukicoder

結果

テストケース

ソースコード