제출 #495129

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
495129		pragmatist	Evacuation plan (IZhO18_plan)	C++17	23 / 100	4046 ms	102052 KiB

plan

#include <bits/stdc++.h>                            
 
#define pb push_back
#define sz(v) (int)v.size()
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define x first
#define y second
#define nl "\n"
 
using namespace std;

typedef long long ll;
typedef pair<long long, long long> pll;
typedef pair <ll, ll> pii;

const int N = (int)3e5 + 7;
const int M = (int)7e6 + 7;
const ll MOD = (ll)1e9 + 7;                    
const int inf = (int)1e9 + 7;
const ll INF = (ll)3e18 + 7;

pii dir[] = {{-1, -1}, {1, 1}, {-1, 1}, {1, -1}};

int n, m, k, s, t, ans, Q, d[N];
vector<pii> g[N];
bool used[N];

void dfs(int v, int mn) {
	used[v] = 1;
	mn = min(mn, d[v]);
	if(v == t) {                            
		ans = max(ans, mn);
	}
	for(auto [to, w] : g[v]) {
		if(!used[to]) {  
			dfs(to, mn);
		}	
	}
	used[v] = 0;
}

void solve() {          	             
	cin >> n >> m;
	map<pii, bool> was;
	for(int i = 1; i <= m; ++i) {
		int u, v, w;
		cin >> u >> v >> w;
		was[{u, v}] = was[{v, u}] = 1;
		g[u].pb({v, w});
		g[v].pb({u, w});
	}
	cin >> k;
	for(int i = 1; i <= n; ++i) d[i] = inf;
	set<pii> q;
	for(int i = 1; i <= k; ++i) {
		int x;
		cin >> x;
		d[x] = 0;
		q.insert({0, x});		
	}
	while(!q.empty()) {
		int v = (*q.begin()).y;
		q.erase(q.begin());
		for(auto [to, w] : g[v]) {
			if(d[v] + w < d[to]) {
				q.erase({d[to], to});
				d[to] = d[v] + w;
				q.insert({d[to], to});
			}
		}
	}
	cin >> Q;
	while(Q--) {
		cin >> s >> t;
		
		if(was[{s, t}]) cout << min(d[s], d[t]) << nl;
		else {
			fill(used + 1, used + 1 + n, 0);
			ans = -inf;
			dfs(s, d[s]);   
			cout << ans << nl;	
		}
	}
}           

signed main() {                   
	ios_base::sync_with_stdio(NULL);
    cin.tie(0);
    cout.tie(0);
   	int test = 1;
	//cin >> test;
	for(int i = 1; i <= test; ++i) {
        //cout << "Case " << i << ": ";
        solve();
    }
    return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 13 / 13
• Subtask 3: 0 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 46