oj.uz

Submission #772774

# Submission time Handle Problem Language Result Execution time Memory
772774 2023-07-04T11:12:07 Z boris_mihov Cyberland (APIO23_cyberland) C++17
49 / 100
 31 ms 8824 KB 
#include "cyberland.h"
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <vector>
#include <queue>

typedef long long llong;
const int MAXN = 100000 + 10;
const llong INF = 1e14;
const int INTINF = 1e9;

int n, m, k, h;
bool vis[MAXN];
llong dist[MAXN];
int ability[MAXN];
std::vector <std::pair <int,int>> g[MAXN];
std::priority_queue <std::pair <llong,int>> pq;
double toReach[MAXN];

void reset()
{
    for (int i = 1 ; i <= n ; ++i)
    {
        g[i].clear();
        dist[i] = ability[i] = 0;
        toReach[i] = 0.0;
        vis[i] = false;
    }

    while (!pq.empty())
    {
        pq.pop();
    }
}

void dfs(int node)
{
    vis[node] = true;
    if (node == h)
    {
        return;
    }

    for (const auto &[u, e] : g[node])
    {
        if (vis[u])
        {
            continue;
        }

        dfs(u);
    }
}

double solve(int N, int M, int K, int H, std::vector <int> x, std::vector <int> y, std::vector <int> c, std::vector <int> arr) 
{
    reset();
    n = N;
    m = M;
    k = K;
    h = H + 1;

    for (int i = 0 ; i < m ; ++i)
    {
        g[x[i] + 1].emplace_back(y[i] + 1, c[i]);
        g[y[i] + 1].emplace_back(x[i] + 1, c[i]);
    }

    dfs(1);
    if (!vis[h])
    {
        return -1;
    }

    for (int i = 2 ; i <= n ; ++i)
    {
        ability[i] = arr[i - 1];
    }

    std::fill(dist + 1, dist + 1 + n, INF);
    for (int i = 1 ; i <= n ; ++i)
    {
        if (!vis[i]) continue;
        if (ability[i] == 0)
        {
            dist[i] = 0;
            pq.push({0, i});
        }
    }

    std::fill(vis + 1, vis + 1 + n, false);
    while (!pq.empty())
    {
        auto [currDist, node] = pq.top();
        pq.pop();

        if (vis[node])
        {
            continue;
        }
        
        vis[node] = true;
        if (node != h)
        {
            for (const auto &[u, e] : g[node])
            {
                if (dist[u] > dist[node] + e)
                {
                    dist[u] = dist[node] + e;
                    pq.push({-dist[u], u});
                }
            }
        }
    }

    for (int i = 1 ; i <= n ; ++i)
    {
        toReach[i] = dist[i];
        if (ability[i] == 2)
        {
            int minEdge = INTINF;
            for (const auto &[u, e] : g[i])
            {
                if (u == h)
                {
                    continue;
                }

                minEdge = std::min(minEdge, e);
            }

            toReach[i] /= 2.0;
            if (k > 60)
            {
                toReach[i] = std::min(toReach[i], (double)2*minEdge);
                break;
            }

            for (int j = 1 ; j < k && toReach[i] >= 2*minEdge ; ++j)
            {
                toReach[i] /= 2.0;
                toReach[i] += minEdge;
            }
        }
    }

    std::fill(vis + 1, vis + 1 + n, false);
    std::fill(dist + 1, dist + 1 + n, INF);
    pq.push({0, h});
    dist[h] = 0;

    while (!pq.empty())
    {
        auto [currDist, node] = pq.top();
        pq.pop();

        if (vis[node])
        {
            continue;
        }
        
        vis[node] = true;
        for (const auto &[u, e] : g[node])
        {
            if (dist[u] > dist[node] + e)
            {
                dist[u] = dist[node] + e;
                pq.push({-dist[u], u});
            }
        }
    }

    double ans = INF;
    for (int i = 1 ; i <= n ; ++i)
    {
        ans = std::min(ans, (double)dist[i] + toReach[i]);
    }

    return ans;
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 2856 KB Correct.
2 Correct 15 ms 3072 KB Correct.

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 3168 KB Correct.
2 Correct 24 ms 3380 KB Correct.
3 Correct 23 ms 3472 KB Correct.
4 Correct 23 ms 3412 KB Correct.
5 Correct 30 ms 3468 KB Correct.
6 Correct 27 ms 4132 KB Correct.
7 Correct 28 ms 4180 KB Correct.
8 Correct 13 ms 4888 KB Correct.
9 Correct 21 ms 3252 KB Correct.
10 Correct 23 ms 3332 KB Correct.

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 3476 KB Correct.
2 Correct 24 ms 3164 KB Correct.
3 Correct 23 ms 3240 KB Correct.
4 Correct 24 ms 3176 KB Correct.
5 Correct 24 ms 3104 KB Correct.
6 Correct 6 ms 3580 KB Correct.

Subtask #4
0.0 / 19.0

# Verdict Execution time Memory Grader output
1 Incorrect 27 ms 8824 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #5
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 3532 KB Correct.
2 Correct 21 ms 3464 KB Correct.
3 Correct 21 ms 3492 KB Correct.
4 Correct 23 ms 4424 KB Correct.
5 Correct 19 ms 3312 KB Correct.

Subtask #6
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 22 ms 3520 KB Correct.
2 Correct 18 ms 3260 KB Correct.
3 Correct 31 ms 8184 KB Correct.
4 Correct 17 ms 4040 KB Correct.
5 Correct 20 ms 3084 KB Correct.
6 Correct 20 ms 3272 KB Correct.

Subtask #7
0.0 / 29.0

# Verdict Execution time Memory Grader output
1 Incorrect 21 ms 3508 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #8
0.0 / 3.0

# Verdict Execution time Memory Grader output
1 Incorrect 23 ms 3436 KB Wrong Answer.
2 Halted 0 ms 0 KB -