QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#277806#5540. City HallNYCU_template#WA 2ms6108kbC++201.8kb2023-12-06 22:58:212023-12-06 22:58:24

Judging History

你现在查看的是最新测评结果

  • [2023-12-06 22:58:24]
  • 评测
  • 测评结果:WA
  • 用时:2ms
  • 内存:6108kb
  • [2023-12-06 22:58:21]
  • 提交

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'