QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#277806 | #5540. City Hall | NYCU_template# | WA | 2ms | 6108kb | C++20 | 1.8kb | 2023-12-06 22:58:21 | 2023-12-06 22:58:24 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
using ll = long long int;
using ld = long double;
const int N = 100005;
vector<int> adj[N];
ll h[N];
int n, m;
const ll INF = 1e18;
vector<ll> Dijkstra(int src) {
vector<ll> dis(n, INF);
vector<ll> vis(n, false);
using pll = pair<ll, int>;
priority_queue<pll, vector<pll>, greater<pll>> pq;
dis[src] = 0;
pq.push({0, src});
while(!pq.empty()) {
int x = pq.top().second;
pq.pop();
if(vis[x]) continue;
vis[x] = true;
for(auto v : adj[x]) {
ll w = (h[v] - h[x]) * (h[v] - h[x]);
if(dis[v] > dis[x] + w) {
dis[v] = dis[x] + w;
pq.push({dis[v], v});
}
}
}
return dis;
}
int main() {
cin.tie(0);
ios_base::sync_with_stdio(false);
int s, t;
cin >> n >> m >> s >> t;
s--, t--;
for(int i = 0; i < n; i++) {
cin >> h[i];
}
for(int i = 0; i < m; i++) {
int a, b;
cin >> a >> b;
a--, b--;
adj[a].push_back(b);
adj[b].push_back(a);
}
vector<ll> from_src = Dijkstra(s), to_dst = Dijkstra(t);
ld ans = from_src[t];
for(int i = 0; i < n; i++) {
sort(adj[i].begin(), adj[i].end(), [&](int a, int b) {
return h[a] < h[b];
});
for(int j = 0; j < (int)adj[i].size(); j++) {
for(int k = 0; k < (int)adj[i].size(); k++) {
if(j == k) continue;
int v = adj[i][j], u = adj[i][k];
ans = min(ans, from_src[v] + to_dst[u] + (h[v] - h[u]) * (h[v] - h[u]) / (ld)2.0);
}
}
}
cout << fixed << setprecision(10) << ans << "\n";
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 5980kb
input:
5 6 1 3 5 100 8 2 10 1 2 2 3 2 5 1 4 4 5 3 5
output:
4.5000000000
result:
ok found '4.5000000', expected '4.5000000', error '0.0000000'
Test #2:
score: 0
Accepted
time: 2ms
memory: 6108kb
input:
5 5 1 5 1 2 10 10 4 1 2 2 3 2 4 3 5 4 5
output:
3.0000000000
result:
ok found '3.0000000', expected '3.0000000', error '0.0000000'
Test #3:
score: 0
Accepted
time: 0ms
memory: 5924kb
input:
5 4 1 4 8 8 8 8 100 1 2 2 3 3 4 4 5
output:
0.0000000000
result:
ok found '0.0000000', expected '0.0000000', error '-0.0000000'
Test #4:
score: -100
Wrong Answer
time: 1ms
memory: 5984kb
input:
2 1 1 2 0 100000 1 2
output:
10000000000.0000000000
result:
wrong answer 1st numbers differ - expected: '0.0000000', found: '10000000000.0000000', error = '10000000000.0000000'