Submission #498511

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
498511	2021-12-25T11:12:53 Z	gratus907	None (KOI16_gas)	C++17	Compilation error	0 ms	0 KB

gas

#include <iostream>
#include <vector>
#include <queue>
#pragma GCC optimize("Ofast")
#define ll long long
#define eps 1e-7
#define all(x) ((x).begin()),((x).end())
#define usecppio ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
using pii = pair<int, int>;

const ll INF = 987654321987654321;
const int MX = 2625;

struct Edge {
    int src, dst, w;
    bool operator<(const Edge &p) const {
        return w > p.w;
    }
};
int fuel[MX];
ll cost[MX][MX];
vector <Edge> G[MX];

void dijkstra(int start, vector<Edge> graph[])
{
    memset(cost, 0x7f, sizeof(cost));
    cost[start][fuel[start]] = 0;
    priority_queue<Edge> pq;
    pq.push({0, start,0});
    while(!pq.empty()) {
        Edge x = pq.top();
        pq.pop();
        //printf("Process edge %lld -> %lld (len=%lld)\n",x.src, x.dst, x.w);
        for (auto ed : graph[x.dst]) {
            bool flag = false;
            int max_j = min(2500, fuel[ed.src]);
            for (int j = 1; j <= max_j; j++) {
                int fcost = min(fuel[ed.dst], j);
                int u = ed.src;
                int v = ed.dst;
                int w = ed.w * j;
                if (cost[v][fcost] > cost[u][j] + w) {
                    flag = true;
                    cost[v][fcost] = cost[u][j] + w;
                }
            }
            if (flag) pq.push(ed);
        }
    }
}

int32_t main() {
    usecppio
    int n, m; cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> fuel[i];
    for (int i = 0; i < m; i++) {
        int u, v, w; cin >> u >> v >> w;
        G[u].push_back({u, v, w});
        G[v].push_back({v, u, w});
    }
    dijkstra(1, G);
    ll ans = INF;
    for (int j = 1; j <= 2500; j++) ans = min(ans, cost[n][j]);
    cout << ans << '\n';
}

Compilation message

gas.cpp: In function 'void dijkstra(int, std::vector<Edge>*)':
gas.cpp:27:5: error: 'memset' was not declared in this scope
   27 |     memset(cost, 0x7f, sizeof(cost));
      |     ^~~~~~
gas.cpp:4:1: note: 'memset' is defined in header '<cstring>'; did you forget to '#include <cstring>'?
    3 | #include <queue>
  +++ |+#include <cstring>
    4 | #pragma GCC optimize("Ofast")