結果

問題 No.2630 Colorful Vertices and Cheapest Paths
ユーザー cho435
提出日時 2024-02-16 22:00:19
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 862 ms / 2,500 ms
コード長 1,101 bytes
コンパイル時間 24,645 ms
コンパイル使用メモリ 359,148 KB
最終ジャッジ日時 2025-02-19 13:59:03
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(int i=0;i<(int)(n);i++)
using mint = atcoder::modint998244353;

using atcoder::dsu;

int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n,m;
	cin>>n>>m;
	vector<vector<int>> g(n);
	rep(i,m){
		int a,b;
		cin>>a>>b;
		a--; b--;
		g.at(a).push_back(b);
		g.at(b).push_back(a);
	}
	vector<int> c(n);
	rep(i,n) cin>>c.at(i);
	rep(i,n) c.at(i)--;
	vector<int> w(10);
	rep(i,10) cin>>w.at(i);
	vector<dsu> d(1<<10,dsu(n));
	vector<ll> sc(1<<10,0);
	rep(bit,1<<10){
		dsu &uf=d.at(bit);
		rep(x,n){
			if((bit>>c.at(x))&1){
				for(int y:g.at(x)){
					if((bit>>c.at(y))&1){
						uf.merge(x,y);
					}
				}
			}
		}
		rep(i,10){
			if((bit>>i)&1) sc.at(bit)+=w.at(i);
		}
	}
	int q;
	cin>>q;
	rep(Qi,q){
		int u,v;
		cin>>u>>v;
		u--; v--;
		ll ans=1e18;
		rep(bit,1<<10){
			if(d.at(bit).same(u,v)){
				ans=min(ans,sc.at(bit));
			}
		}
		if(ans==1e18) ans=-1;
		cout<<ans<<'\n';
	}
}
0