QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#277781#5540. City HallNYCU_template#WA 1ms6416kbC++203.2kb2023-12-06 22:45:082023-12-06 22:45:17

Judging History

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

  • [2023-12-06 22:45:17]
  • 评测
  • 测评结果:WA
  • 用时:1ms
  • 内存:6416kb
  • [2023-12-06 22:45:08]
  • 提交

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;
}

struct Line {
    ld a, b; // ax + b
    ld val(ld x) { return (a - x) * (a - x) / 2.0 + b; }
    Line() {}
    Line(ld _a, ld _b) : a(_a), b(_b) {}
};
struct Node {
    int L, R;
    Line line;
    Node *lc, *rc;
};

int cur;

void build(Node *r, int L, int R) {
    r->L = L, r->R = R;
    r->line = Line(2e5, INF);
    if(L == R) return;
    int M = (L + R) / 2;
    build(r->lc = new Node(), L, M);
    build(r->rc = new Node(), M + 1, R);
}
void update(Node *r, Line line) {
    Line left_line, right_line;
    if(line.a > r->line.a) {
        left_line = line;
        right_line = r->line;
    } else {
        left_line = r->line;
        right_line = line;
    }

    int M = (r->L + r->R) / 2;
    ld Mx = h[adj[cur][M]];
    if(left_line.val(Mx) < right_line.val(Mx)) {
        r->line = left_line;
        if(r->L != r->R) update(r->rc, right_line);
    } else {
        r->line = right_line;
        if(r->L != r->R) update(r->lc, left_line);
    }
}
ld qry(Node *r, ld x) {
    if(r->L == r->R)
        return r->line.val(x);
    int M = (r->L + r->R) / 2;
    ld Mx = h[adj[cur][M]];
    if(x <= Mx) return min(r->line.val(x), qry(r->lc, x));
    else return min(r->line.val(x), qry(r->rc, x));
}

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];
        });
        
        cur = i;
        Node *root;
        build(root = new Node(), 0, (int)adj[i].size() - 1);
        for(auto v : adj[i]) {
            ans = min(qry(root, h[v]) + to_dst[v], ans);
            update(root, Line(h[v], from_src[v]));
        }

        reverse(adj[i].begin(), adj[i].end());

        Node *root2;
        build(root2 = new Node(), 0, (int)adj[i].size() - 1);
        for(auto v : adj[i]) {
            ans = min(qry(root2, h[v]) + to_dst[v], ans);
            update(root2, Line(h[v], from_src[v]));
        }
    }

    cout << fixed << setprecision(10) << ans << "\n";
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 5988kb

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: 1ms
memory: 5980kb

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: 1ms
memory: 6416kb

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: 6048kb

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'