結果

問題 No.2630 Colorful Vertices and Cheapest Paths
ユーザー AD010AD010
提出日時 2024-02-16 23:20:15
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 1,252 ms / 2,500 ms
コード長 4,879 bytes
コンパイル時間 2,406 ms
コンパイル使用メモリ 144,432 KB
実行使用メモリ 83,712 KB
最終ジャッジ日時 2024-02-16 23:20:47
合計ジャッジ時間 19,787 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 310 ms
22,016 KB
testcase_01 AC 866 ms
43,776 KB
testcase_02 AC 1,124 ms
76,288 KB
testcase_03 AC 2 ms
6,548 KB
testcase_04 AC 2 ms
6,548 KB
testcase_05 AC 2 ms
6,548 KB
testcase_06 AC 298 ms
13,184 KB
testcase_07 AC 1,004 ms
82,432 KB
testcase_08 AC 935 ms
73,600 KB
testcase_09 AC 991 ms
83,712 KB
testcase_10 AC 1,252 ms
83,712 KB
testcase_11 AC 1,236 ms
83,712 KB
testcase_12 AC 1,248 ms
83,712 KB
testcase_13 AC 1,202 ms
83,712 KB
testcase_14 AC 1,239 ms
83,712 KB
testcase_15 AC 424 ms
6,548 KB
testcase_16 AC 417 ms
6,548 KB
testcase_17 AC 425 ms
6,548 KB
testcase_18 AC 543 ms
26,240 KB
testcase_19 AC 784 ms
64,896 KB
testcase_20 AC 709 ms
53,888 KB
testcase_21 AC 887 ms
83,712 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream> // cout, endl, cin
#include <string> // string, to_string, stoi
#include <vector> // vector
#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <utility> // pair, make_pair
#include <tuple> // tuple, make_tuple
#include <cstdint> // int64_t, int*_t
#include <cstdio> // printf
#include <map> // map
#include <queue> // queue, priority_queue
#include <set> // set
#include <stack> // stack
#include <deque> // deque
#include <unordered_map> // unordered_map
#include <unordered_set> // unordered_set
#include <bitset> // bitset
#include <cctype> // isupper, islower, isdigit, toupper, tolower
#include <math.h>
#include <cmath> 
#include <climits>
#include <functional>
#include <iomanip>
#include <limits>
#include <cassert>
#include <random>
#include <complex>
#include <numeric>
#include <regex>
#include <list>
#include <ctime> 

using namespace std;

#define ll long long
#define ull unsigned long long
#define ld  long double
typedef vector<long long> vll;

using LL = long long; using ULL = unsigned long long;
using VI = vector<int>; using VVI = vector<VI>; using VVVI = vector<VVI>;
using VL = vector<LL>; using VVL = vector<VL>; using VVVL = vector<VVL>;
using VB = vector<bool>; using VVB = vector<VB>; using VVVB = vector<VVB>;
using VD = vector<double>; using VVD = vector<VD>; using VVVD = vector<VVD>;
using VC = vector<char>; using VS = vector<string>; using VVC = vector<VC>;
using PII = pair<int,int>; using PLL = pair<LL,LL>; using PDD = pair<double,double>; using PIL = pair<int,LL>;
using MII = map<int,int>; using MLL = map<LL,LL>;
using SI = set<int>; using SL = set<LL>;
using MSI = multiset<int>; using MSL = multiset<LL>;
template<class T> using MAXPQ = priority_queue<T>;
template<class T> using MINPQ = priority_queue< T, vector<T>, greater<T> >;

const ll MOD = 1000000007;
const ll MOD2 = 998244353;
const ll INF = 1LL << 60;
#define PI  3.14159265358979323846

#define FOR(i, a, b) for(int i = (a); i < (b); ++i)
#define REP(i, n) FOR(i, 0, n)
#define EACH(e, v) for(auto &e : v)
#define RITR(it, v) for(auto it = (v).rbegin(); it != (v).rend(); ++it)
#define ALL(v) v.begin(),v.end()

vector<ll> x8={1,1,1,0,0,-1,-1,-1},y8={1,0,-1,1,-1,1,0,-1};
int dx4[4]={1,-1,0,0}, dy4[4]={0,0,1,-1};

/*
memo
-uf,RMQ(segtree),BIT,BIT2,SegTree,SegTreeLazy
-isprime,Eratosthenes,gcdlcm,factorize,divisors,modpow,moddiv(modpow必要)
nCr(+modnCr,inverse,extend_euclid.powmod),tobaseB,tobase10
-dijkstra,Floyd,bellmanford,sccd,topological,treediamiter
-compress1,compress2,rotate90

-co,ci,fo1,fo2,fo3,fo4
-bitsearch,binaryserach
-bfs(vis.assign忘れるな)
-SegTreedec,SegTreeLazydec
*/

struct UnionFind {
    // 親の要素とサイズを管理する変数
    vector<int> parent, size;

    // 変数の初期化
    //UnionFind(int n){
    //    init(n);
    //}

    void init(int n){
        parent.assign(n,-1);
        size.assign(n,1);
    }

    // x の根を求める
    int root(int x) {
        // x が根のとき
        if(parent[x] == -1) return x;

        // 経路圧縮
        return parent[x] = root(parent[x]);
    }

    // x と y の根が同じか
    bool isSame(int x, int y) {
        return root(x) == root(y);
    }

    // x と y のグループを併合する
    void unite(int x, int y) {
        // x と y の根を取得
        int rootX = root(x);
        int rootY = root(y);

        // x と y が同じグループのときは何もしない
        if (rootX == rootY) return;

        // union by size（ y のサイズが小さくなるように調整 ）
        if (size[rootX] < size[rootY]) swap(rootX, rootY);

        // y の親を x にする
        parent[rootY] = rootX;

        // x のサイズに y のサイズを足す
        size[rootX] += size[rootY]; 
    }
};

int main(){
    cin.tie(0);
    ios_base::sync_with_stdio(0);

    ll N,M;
    cin >> N >> M;

    VL a(M),b(M);
    for(ll i = 0; i < M; i++){
        cin >> a[i] >> b[i];
        a[i]--,b[i]--;
    }

    VL c(N);
    for(ll i = 0; i < N; i++){
        cin >> c[i];
        c[i]--;
    }

    VL w(10);
    for(ll i = 0; i < 10; i++) cin >> w[i];

    vector<UnionFind> v(1<<10);
    VL cost(1<<10,0);
    for(ll bit = 0; bit < 1<<10; bit++){
        v[bit].init(N);
        set<ll> st;

        for(ll i = 0; i < 10; i++){
            if(1<<i & bit){
                st.insert(i);
                cost[bit] += w[i];
            }
        }

        for(ll i = 0; i < M; i++){
            if(st.count(c[a[i]]) && st.count(c[b[i]])) v[bit].unite(a[i],b[i]);
        }
    }

    ll U,V,Q;
    cin >> Q;
    for(ll i = 0; i < Q; i++){
        cin >> U >> V;
        U--,V--;
        ll ans = INF;
        for(ll bit = 0; bit < 1<<10; bit++){
            if(v[bit].isSame(U,V)) ans = min(ans,cost[bit]);
        }

        cout << (ans == INF? -1 : ans) << endl;
    }
}
0