답안 #793394

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
793394 2023-07-25T19:05:08 Z Valaki2 007 (CEOI14_007) C++14
69.6667 / 100
166 ms 17584 KB
#include <bits/stdc++.h>
using namespace std;

#define pb push_back

const int maxn = 2e5;

int n, m;
vector<int> graph[1 + maxn];
int g, s, a, b;
vector<int> dg, ds, da, db;

vector<int> bfs(int start) {
    vector<int> res(1 + n, -1);
    res[start] = 0;
    queue<int> q;
    q.push(start);
    while(!q.empty()) {
        int cur = q.front();
        q.pop();
        for(int nei : graph[cur]) {
            if(res[nei] == -1) {
                res[nei] = res[cur] + 1;
                q.push(nei);
            }
        }
    }
    return res;
}

void solve() {
    cin >> n >> m;
    cin >> g >> s >> a >> b;
    for(int i = 0; i < m; i++) {
        int x, y;
        cin >> x >> y;
        graph[x].pb(y);
        graph[y].pb(x);
    }
    dg = bfs(g), ds = bfs(s), da = bfs(a), db = bfs(b);
    if((da[s] < da[g]) || (db[s] < db[g])) {
        cout << "-1\n";
        return;
    }
    int add_a = da[s] - 1 - da[g], add_b = db[s] - 1 - db[g];
    bool extra_case = true;
    if(add_a != add_b) {
        add_a++, add_b++;
        extra_case = false;
    }
    int to_add = min(add_a, add_b);
    if(!extra_case) {
        cout << to_add << "\n";
        return;
    }
    to_add++;
    da[g] += to_add;
    db[g] += to_add;
    for(int i = 1; i <= n; i++) {
        if(i != g) {
            dg[i] += to_add;
        }
    }
    int closest_g = n + 1, closest_s = n + 1;
    if(da[s] != db[s] || da[g] != db[g]) {
        //exit(1);
    }
    if(da[s] != da[g] || db[s] != db[g]) {
        exit(1);
    }
    for(int cur = 1; cur <= n; cur++) {
        /*int max_dab = max(da[cur], db[cur]), min_dab = min(da[cur], db[cur]);
        if((max_dab + closest_g == max(da[g], db[g])) && (min_dab + closest_g == min(da[g], db[g]))) {
            closest_g = min(closest_g, max_dab);
        }
        if((max_dab + closest_s == max(da[s], db[s]))) {
            closest_s = min(closest_s, max_dab);
        }*/
        int max_dab = max(da[cur], db[cur]);
        int min_dab = min(da[cur], db[cur]);
        if(max_dab + dg[cur] == max(da[g], db[g])) {
            if(min_dab + dg[cur] == min(da[g], db[g])) {
                closest_g = min(closest_g, max_dab);
            }
        }
        if(max_dab + ds[cur] == max(da[s], db[s])) {
            if(min_dab + ds[cur] == min(da[s], db[s])) {
                closest_s = min(closest_s, max_dab);
            }
        }
    }
    if(closest_s < closest_g) {
        to_add--;
    }
    cout << to_add << "\n";
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    solve();
    return 0;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 3 ms 4948 KB Output is correct
4 Correct 2 ms 4948 KB Output is correct
5 Correct 2 ms 4948 KB Output is correct
6 Partially correct 2 ms 5020 KB Partially correct
7 Correct 2 ms 4948 KB Output is correct
8 Correct 2 ms 4948 KB Output is correct
9 Correct 2 ms 4948 KB Output is correct
10 Correct 2 ms 4948 KB Output is correct
11 Correct 2 ms 4948 KB Output is correct
12 Correct 3 ms 4948 KB Output is correct
13 Correct 2 ms 4948 KB Output is correct
14 Correct 2 ms 4948 KB Output is correct
15 Correct 2 ms 4948 KB Output is correct
16 Correct 2 ms 4948 KB Output is correct
17 Correct 2 ms 4948 KB Output is correct
18 Correct 2 ms 4948 KB Output is correct
19 Correct 2 ms 4948 KB Output is correct
20 Correct 2 ms 4948 KB Output is correct
21 Correct 2 ms 5004 KB Output is correct
22 Partially correct 2 ms 4948 KB Partially correct
23 Partially correct 2 ms 4948 KB Partially correct
24 Correct 3 ms 4948 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 14 ms 6932 KB Output is correct
2 Correct 17 ms 7764 KB Output is correct
3 Correct 22 ms 6996 KB Output is correct
4 Correct 18 ms 7840 KB Output is correct
5 Correct 15 ms 6820 KB Output is correct
6 Correct 13 ms 7124 KB Output is correct
7 Partially correct 16 ms 7236 KB Partially correct
8 Correct 19 ms 7228 KB Output is correct
9 Correct 25 ms 7724 KB Output is correct
10 Correct 102 ms 12060 KB Output is correct
11 Correct 29 ms 9304 KB Output is correct
12 Correct 40 ms 10324 KB Output is correct
13 Correct 45 ms 9500 KB Output is correct
14 Correct 25 ms 8820 KB Output is correct
15 Correct 36 ms 10344 KB Output is correct
16 Correct 37 ms 10712 KB Output is correct
17 Correct 37 ms 10068 KB Output is correct
18 Correct 42 ms 10064 KB Output is correct
19 Correct 65 ms 11276 KB Output is correct
20 Correct 161 ms 14200 KB Output is correct
21 Correct 48 ms 12492 KB Output is correct
22 Correct 59 ms 11452 KB Output is correct
23 Correct 51 ms 12340 KB Output is correct
24 Correct 46 ms 12320 KB Output is correct
25 Correct 47 ms 11980 KB Output is correct
26 Correct 44 ms 11600 KB Output is correct
27 Correct 49 ms 12480 KB Output is correct
28 Correct 63 ms 12504 KB Output is correct
29 Correct 77 ms 12976 KB Output is correct
30 Correct 121 ms 14844 KB Output is correct
31 Correct 67 ms 13784 KB Output is correct
32 Correct 62 ms 12348 KB Output is correct
33 Correct 60 ms 12700 KB Output is correct
34 Correct 60 ms 13036 KB Output is correct
35 Correct 50 ms 12772 KB Output is correct
36 Correct 51 ms 13004 KB Output is correct
37 Correct 61 ms 14052 KB Output is correct
38 Correct 86 ms 13944 KB Output is correct
39 Correct 80 ms 13856 KB Output is correct
40 Correct 110 ms 15400 KB Output is correct
41 Correct 166 ms 17584 KB Output is correct