Submission #777310

#	Time	Username	Problem	Language	Result	Execution time	Memory
777310		yeyso	Cyberland (APIO23_cyberland)	C++17	0 / 100	30 ms	9796 KiB

cyberland

#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;
#include <vector>


double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {

    vector<vector<long long>> adj(N, vector<long long>());

    vector<vector<long long>> dma(N, vector<long long>());
    for(long long i = 0; i < M; i ++){
        adj[x[i]].push_back(y[i]);
        adj[y[i]].push_back(x[i]);

        dma[x[i]].push_back(c[i]);
        dma[y[i]].push_back(c[i]);
    }
    // parent, node
    //queue<pair<long long, long long>> q;

    queue<pair<long long, long long>> q;
    q.push({0, H});
    vector<long long> v(N, 0);
    vector<long long> d(N, LONG_LONG_MAX / 2);
    long long node, dist;
    while(!q.empty()){
        node = q.front().second;
        dist = q.front().first;
        q.pop();
        if(!v[node]){
            v[node] = 1;
            for(long long i = 0; i < adj[node].size(); i ++){
                d[adj[node][i]] = min(d[adj[node][i]], d[node] + dma[node][i]);
                q.push({dist + dma[node][i], adj[node][i]});
            }
        }
    }
    q.push({0, 0});
    vector<long long> v2(N, 0);
    v2[H] = 1;
    while(!q.empty()){
        node = q.front().second;
        q.pop();
        if(!v2[node]){
            v2[node] = 1;
            for(long long i = 0; i < adj[node].size(); i ++){
                q.push({0, adj[node][i]});
            }
        }
    }
    long long lowest = d[0];
    for(long long i = 0; i < d.size(); i ++){
        if(d[i] < lowest and v2[i] == 1 and arr[i] == 0){
            lowest = d[i];
        }
    }
    if(lowest >= LONG_LONG_MAX / 2){
        lowest = -1;
    }
    return lowest;


}
/*
g++ -std=gnu++17 -O2 -Wall -pipe -static -o cyberland stub.cpp cyberland.cpp
1
10 9 10
6
1 1 0 1 2 2 1 0 2 1
0 1 3
1 2 1
2 3 1
3 4 1
4 5 1
5 6 1
6 7 1
7 8 1
8 9 1

1
3 2 30
2
1 0 1
1 2 12
2 0 4
*/

Compilation message (stderr)

cyberland.cpp: In function 'double solve(int, int, int, int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
cyberland.cpp:33:36: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |             for(long long i = 0; i < adj[node].size(); i ++){
      |                                  ~~^~~~~~~~~~~~~~~~~~
cyberland.cpp:47:36: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   47 |             for(long long i = 0; i < adj[node].size(); i ++){
      |                                  ~~^~~~~~~~~~~~~~~~~~
cyberland.cpp:53:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |     for(long long i = 0; i < d.size(); i ++){
      |                          ~~^~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 8
• Subtask 3: 0 / 13
• Subtask 4: 0 / 19
• Subtask 5: 0 / 7
• Subtask 6: 0 / 16
• Subtask 7: 0 / 29
• Subtask 8: 0 / 3