oj.uz

Submission #44145

# Submission time Handle Problem Language Result Execution time Memory
44145 2018-03-30T08:29:56 Z wasyl Sightseeing (NOI14_sightseeing) C++11
15 / 25
 2650 ms 125920 KB 
#include <bits/stdc++.h>
#ifndef dbg
#define dbg(...)
#endif
#define all(x) begin(x), end(x)
#define rsz(...) resize(__VA_ARGS__)
#define psh(...) push_back(__VA_ARGS__)
#define emp(...) emplace_back(__VA_ARGS__)
#define prt(...) print(cout, __VA_ARGS__)
#define dmp(...) print(cerr, #__VA_ARGS__, '=', __VA_ARGS__)
#define dprt(...) dbg(print(cerr,__VA_ARGS__))
#define ddmp(...) dbg(dmp(__VA_ARGS__))
using namespace std;using ll=long long;
template<typename t>using V=vector<t>;
template<typename t>void print(ostream& os, const t& a){os<<a<<'\n';}
template<typename t, typename... A>void print
(ostream& os, const t& a, A&&... b){os<<a<<' ';print(os, b...);}

constexpr int MaxK = 1e5;

int n, m, q;
V< int > odp, siz, lid;
V< V< pair< int, int > > > ilo;
V< V< pair< int, int > > > gr;

inline int find (int v)
{
    return lid[v] == lid[lid[v]]? lid[v] : lid[v] = find(lid[v]);
}

inline void join (int p, int q)
{
    p = find(p);
    q = find(q);
    
    if (siz[p] < siz[q])
        swap(p, q);
    
    siz[p] += siz[q];
    lid[q] = p;
}

inline void dfs (int v, int o, int r = INT_MAX)
{
    odp[v] = r;
    for (auto& p : gr[v])
        if (p.first != o)
            dfs(p.first, v, min(r, p.second));
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    cin >> n >> m >> q;
    
    gr.rsz(n + 1);
    lid.rsz(n + 1);
    siz.rsz(n + 1, 1);
    odp.rsz(n + 1, 0);
    ilo.rsz(MaxK + 1);
    
    for (int i = 0; i < m; ++i)
    {
        int p, q, k;
        cin >> p >> q >> k;
        ilo[k].psh({p, q});
    }
    
    for (int i = 1; i <= n; ++i)
        lid[i] = i;
    
    for (int i = MaxK; i >= 0; --i)
        for (auto& v : ilo[i])
            if (find(v.first) != find(v.second))
            {
                join(v.first, v.second);
                gr[v.first].psh({v.second, i});
                gr[v.second].psh({v.first, i});
            } 
    
    dfs(1, 1);
    
    for (int i = 0; i < q; ++i)
    {
        int p; cin >> p;
        prt(odp[p]);    
    }
}

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 2680 KB Output is correct
2 Correct 4 ms 2792 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 3160 KB Output is correct
2 Correct 4 ms 3160 KB Output is correct

Subtask #3
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 45 ms 7844 KB Output is correct
2 Correct 43 ms 8148 KB Output is correct

Subtask #4
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2650 ms 125920 KB Output is correct
2 Runtime error 51 ms 125920 KB Execution killed with signal 11 (could be triggered by violating memory limits)