Submission #51269

#	Time	Username	Problem	Language	Result	Execution time	Memory
51269		Kubalionzzale	Sightseeing (NOI14_sightseeing)	C++14	15 / 25	3553 ms	113036 KiB

sightseeing

#include <iostream>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <functional>

//std::vector< std::vector< std::pair<int, int> > > g(500010);
int ans[1000010] = { 0 }, start[1000010] = { 0 };
std::pair<int, std::pair<int, int> > edge[10000010];
int n, m, q;

void bfs()
{
    std::set<std::pair<int, int> , std::greater<std::pair<int, int> > > set;
    set.insert(std::make_pair(2e8, 0));
    bool visited[1000010] = { 0 };

    int curNode, curVal, nextVal, nextNode;
    while (set.empty() == false)
    {
        std::pair<int, int> top = *set.begin();
        curNode = top.second;
        curVal = top.first;
        set.erase(set.begin());

  //      std::cout << curNode << " " << curVal << "\n";

        if (visited[curNode])
            continue;
        else
            visited[curNode] = 1;

        for (int i = start[curNode]; i < m; ++i)
        {
            if (edge[i].first != curNode)
                break;
            nextVal = edge[i].second.first;
            nextNode = edge[i].second.second;
            if (std::min(nextVal, curVal) > ans[nextNode])
            {
                ans[nextNode] = std::max(ans[nextNode], std::min(nextVal, curVal));
                set.insert(std::make_pair(std::min(nextVal, curVal), nextNode));
            }
        }
    }
}

int main()
{
    std::ios_base::sync_with_stdio(0);
    std::cin.tie(0);
//    scanf("%d%d%d", &n, &m, &q);
    std::cin >> n >> m >> q;

    int x, y, quality;
    for (int i = 0; i < m; ++i)
    {
//        scanf("%d%d%d", &x, &y, &quality);
        std::cin >> x >> y >> quality;
        --x,
        --y;


        edge[i] = std::make_pair(x, std::make_pair(quality, y));
        edge[m + i] = std::make_pair(y, std::make_pair(quality, x));
    }
    m *= 2;

    std::sort(edge, edge + m);

    int node[1000010];
    for (int i = 0; i < q; ++i)
    {
//        scanf("%d", &node[i]);
        std::cin >> node[i];
    }

    for (int i = 0; i < n; ++i)
    {
        start[i] = std::lower_bound(edge, edge + m, std::make_pair(i, std::make_pair(0, 0))) - edge;
    }

    bfs();

    for (int i = 0; i < q; ++i)
    {
//        printf("%d\n", ans[node[i] - 1]);
        std::cout << ans[node[i] - 1] << "\n";
    }
}

• Subtask 1: 4 / 4
• Subtask 2: 5 / 5
• Subtask 3: 6 / 6
• Subtask 4: 0 / 10