#include <bits/stdc++.h>
#include "crocodile.h"
using namespace std;
int n, m, k; vector<pair<int, long long>>adj[100000];
vector<int>ex; vector<bool>e; long long memo[1024][1024];
long long dfs(int v, int p)
{
if(e[v]) return 0;
if(memo[v][p]!=-1) return memo[v][p];
long long best=1e9, ans=1e9; int node=-1;
for(auto u:adj[v]) {
int to=u.first; long long w=u.second;
if(to==p) continue;
long long cnt=w+dfs(to, v);
if(cnt<best) node=to, best=cnt;
}
for(auto u:adj[v]) {
int to=u.first; long long w=u.second;
if(to==p||to==node) continue;
ans=min(ans, w+dfs(to, v));
}
return memo[v][p]=ans;
}
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
n=N, m=M, k=K; e.assign(n, false);
memset(memo, -1, sizeof memo);
for(int l=0;l<m;l++) {
int a=R[l][0], b=R[l][1]; long long w=L[l];
adj[a].push_back({b, w}); adj[b].push_back({a, w});
}
for(int l=0;l<k;l++) {
int i=P[l]; e[i]=true; ex.push_back(i);
}
N=dfs(0, -1);
return N;
}
Compilation message
crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:11:14: warning: array subscript -1 is below array bounds of 'long long int [1024]' [-Warray-bounds]
11 | if(memo[v][p]!=-1) return memo[v][p];
| ~~~~~~~~~^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
6 ms |
10752 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
6 ms |
10752 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
6 ms |
10752 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |