oj.uz

답안 #330104

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
330104 2020-11-23T23:17:11 Z ignaciocanta 관광 (NOI14_sightseeing) C++14
25 / 25
 2349 ms 202704 KB 
#include <bits/stdc++.h>
 
using namespace std;
 
using tint = long long;
using ld = long double;
 
#define forsn(i, s, n) for(int i = s; i < int(n); i++)
#define forn(i, n) forsn(i, 0, n)
#define trav(a, x) for(auto& a : x)
 
using vi = vector<int>;
using vl = vector<tint>;
using vb = vector<bool>;
#define pb push_back
#define pf push_front
#define rsz resize
#define all(x) begin(x), end(x)
#define rall(x) x.rbegin(), x.rend() 
#define sz(x) (int)(x).size()
#define ins insert
#define esta(x,c) ((c).find(x) != (c).end())
 
using pi = pair<int,int>;
using pl = pair<tint,tint>;
#define f first
#define s second
#define mp make_pair
 
#define DBG(x) cerr << #x << " = " << x << endl;
 
const int MOD = 1e9+7;
const tint mod = 998244353;
const int MX = 5e5+5; 
const tint INF = 1e18; 
const int inf = 2e9;
const ld PI = acos(ld(-1)); 
const ld eps = 1e-8;
 
const int dx[4] = {1, -1, 0, 0};
const int dy[4] = {0, 0, 1, -1};
 
template<class T> void remDup(vector<T> &v){ 
    sort(all(v)); v.erase(unique(all(v)),end(v));
}
 
template<class T> bool valid(T x, T y, T n, T m){
    return (0<=x && x<n && 0<=y && y<m);
}

tint cdiv(tint a, tint b) { return a/b+((a^b)>0&&a%b); } //redondea p arriba
int fdiv(int a, int b) { return a/b-((a^b)<0&&a%b); } //redonde p abajo
 
void NACHO(string name = "cbarn"){
    ios_base::sync_with_stdio(0); cin.tie(0);
    //freopen((name+".in").c_str(), "r", stdin);
    //freopen((name+".out").c_str(), "w", stdout);
}

struct edge{
	int u, v, w;
	bool operator<(const edge &a)const{ return w > a.w; }
};

vector<pi> tree[MX];

int pa[MX];
int tam[MX];
int ret[MX];

void dfs(int node, int pa){
	trav(u, tree[node]){
		if(u.f != pa){
			ret[u.f] = min(ret[node], u.s);
			dfs(u.f, node);
		}
	}
}

void init(int n){
	forn(i, n) pa[i] = i, tam[i] = 1;
}

int find(int a){
	if(a == pa[a]) return a;
	return pa[a] = find(pa[a]);
}

void unite(int a, int b){
	a = find(a), b = find(b);
	if(a != b){
		if(tam[a] < tam[b]) swap(a, b);
		pa[b] = a;
		tam[a]+=tam[b];
	}
}

int main(){
	NACHO();
	int n, m, q; cin >> n >> m >> q;
	vector<edge> edges (m);
	init(n);
	forn(i, m){
		int u, v, w; cin >> u >> v >> w;
		--u, --v;
		edges[i] = {u, v, w};
	}
	sort(all(edges));
	forn(i, m){
		int u = edges[i].u, v = edges[i].v, w = edges[i].w;
		if(find(u) != find(v)){
			unite(u, v);
			tree[u].pb(mp(v, w));
			tree[v].pb(mp(u, w));
		}
	}
	ret[0] = inf;
	dfs(0, -1);
	forn(assa, q){
		int x; cin >> x;
		--x;
		if(x == 0) cout << 0 << "\n";
		else cout << ret[x] << "\n";
	}
}

Subtask #1
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 12140 KB Output is correct
2 Correct 8 ms 12140 KB Output is correct

Subtask #2
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 10 ms 12268 KB Output is correct
2 Correct 8 ms 12140 KB Output is correct

Subtask #3
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 38 ms 15340 KB Output is correct
2 Correct 34 ms 14956 KB Output is correct

Subtask #4
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1675 ms 125116 KB Output is correct
2 Correct 2349 ms 202704 KB Output is correct