oj.uz

Submission #44153

# Submission time Handle Problem Language Result Execution time Memory
44153 2018-03-30T08:48:02 Z wasyl Sightseeing (NOI14_sightseeing) C++11
15 / 25
 3500 ms 70704 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;
constexpr int MaxN = 5e5;

int n, m, q; bool fl = false;
int odp[MaxN + 1], siz[MaxN + 1], lid[MaxN + 1];
V< pair< int, int > > ilo[MaxK + 1];
V< pair< int, int > > gr[MaxN + 1];

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;
    
    if (siz[p] == n)    
        fl = true;
}

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;
    scanf("%d%d%d", &n, &m, &q);
    
    for (int i = 0; i < m; ++i)
    {
        int p, q, k;
        scanf("%d%d%d", &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});
            } 
        ilo[i].clear(); ilo[i].shrink_to_fit();
        if (fl)
            break;
    }
    
    dfs(1, 1);
    
    for (int i = 0; i < q; ++i)
    {
        int p; //cin >> p;
        scanf("%d", &p);
        printf("%d\n", odp[p]);    
    }
}

Compilation message

sightseeing.cpp: In function 'int main()':
sightseeing.cpp:61:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d%d", &n, &m, &q);
     ~~~~~^~~~~~~~~~~~~~~~~~~~~~
sightseeing.cpp:66:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d%d", &p, &q, &k);
         ~~~~~^~~~~~~~~~~~~~~~~~~~~~
sightseeing.cpp:93:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &p);
         ~~~~~^~~~~~~~~~

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 14456 KB Output is correct
2 Correct 14 ms 14520 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 22 ms 14620 KB Output is correct
2 Correct 14 ms 14620 KB Output is correct

Subtask #3
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 61 ms 16956 KB Output is correct
2 Correct 52 ms 16956 KB Output is correct

Subtask #4
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2786 ms 62680 KB Output is correct
2 Execution timed out 3560 ms 70704 KB Time limit exceeded