답안 #982573

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
982573 2024-05-14T12:34:01 Z phoenix 즐거운 행로 (APIO20_fun) C++17
31 / 100
88 ms 15588 KB
#include <bits/stdc++.h>
#include "fun.h"

#define get_dist hoursRequired
#define get_size attractionsBehind

#ifdef LOCAL 
#define cerr cout
#else 
#define cerr if (false) cout
#endif


using namespace std;

vector<int> createFunTour(int n, int q) {
    vector<int> res;
    int C;
    vector<pair<int, int>> d[3];
    vector<int> dist(n);

    /*Step 1: Find Centroid */ {
        // a) pick a random vertex
        int v = 0;
        // b) for all i, find i's subtree's size if tree is rooted by v
        vector<int> sz(n);
        for (int i = 0; i < n; i++)
            sz[i] = get_size(v, i);
        // c) centroid is i s.t, 2 * sz[i] > n and sz[i] - min
        C = -1;
        for (int i = 0; i < n; i++)
            if (2 * sz[i] > n) 
                if (C == -1 || sz[i] < sz[C]) 
                    C = i;
        // d) Yay! We found centroid of tree
    }       

    /*Step 2: Find distance from C to any other vertices*/ {
        for (int i = 0; i < n; i++)
            dist[i] = get_dist(C, i);
    }

    /*Step 3: Divide all vertices to subrees of C*/ {
        vector<int> c;
        for (int i = 0; i < n; i++) 
            if (dist[i] == 1)
                c.push_back(i);
        
        for (int i = 0; i < n; i++) {
            if (i == C) continue;
            int j = 0;
            while (j < (int)c.size() - 1) {
                if (get_dist(i, c[j]) < dist[i]) break;
                j++;
            }
            d[j].push_back({dist[i], i});
        }
    }

    for (int i = 0; i < 3; i++) {
        sort(d[i].rbegin(), d[i].rend());
    }

    /*Step 4: Solve problem :)*/ {
        res.push_back(C);

        int lst = -1;
        while ((int)res.size() < n) {
            cerr << "oper: " << (int)res.size() << '\n';
            for (int i = 0; i < 3; i++) {
                cerr << "{";
                for (auto [x, y] : d[i]) cerr << '(' << x << ' ' << y << ") ";
                cerr << "}\n";
            }
            cerr << '\n';
            int s = 0;
            for (int i = 0; i < 3; i++) 
                s += (int)d[i].size();
            
            int nxt = -1;
            for (int i = 0; i < 3; i++) {
                if (d[i].empty() || i == lst) 
                    continue;
                if ((int)d[i].size() * 2 >= s) {
                    nxt = i;
                    break;
                } 
                if (nxt == -1 || d[i].back().first < d[nxt].back().first)
                    nxt = i;
            }
            assert(nxt != -1);
            res.push_back(d[nxt].back().second);
            d[nxt].pop_back();
            lst = nxt;
        }
    }
    for (int c : res)   
        cerr << c << ' ';
    cerr << '\n';
    reverse(res.begin(), res.end());

    return res;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 1 ms 344 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 0 ms 512 KB Output is correct
16 Correct 1 ms 440 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 436 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 1 ms 344 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 0 ms 512 KB Output is correct
16 Correct 1 ms 440 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 436 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 344 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 0 ms 348 KB Output is correct
24 Correct 1 ms 348 KB Output is correct
25 Correct 1 ms 348 KB Output is correct
26 Correct 0 ms 348 KB Output is correct
27 Correct 1 ms 348 KB Output is correct
28 Correct 0 ms 348 KB Output is correct
29 Correct 1 ms 348 KB Output is correct
30 Correct 0 ms 348 KB Output is correct
31 Incorrect 0 ms 436 KB Tour is not fun
32 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 344 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 0 ms 344 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 1 ms 348 KB Output is correct
13 Correct 1 ms 348 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 1 ms 348 KB Output is correct
16 Correct 1 ms 344 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 88 ms 15588 KB Output is correct
19 Correct 1 ms 344 KB Output is correct
20 Correct 1 ms 604 KB Output is correct
21 Correct 2 ms 848 KB Output is correct
22 Correct 4 ms 1372 KB Output is correct
23 Correct 9 ms 2140 KB Output is correct
24 Correct 11 ms 2908 KB Output is correct
25 Correct 48 ms 9076 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Incorrect 0 ms 436 KB Tour is not fun
11 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 1 ms 344 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 0 ms 512 KB Output is correct
16 Correct 1 ms 440 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 436 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 344 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 0 ms 348 KB Output is correct
24 Correct 1 ms 348 KB Output is correct
25 Correct 1 ms 348 KB Output is correct
26 Correct 0 ms 348 KB Output is correct
27 Correct 1 ms 348 KB Output is correct
28 Correct 0 ms 348 KB Output is correct
29 Correct 1 ms 348 KB Output is correct
30 Correct 0 ms 348 KB Output is correct
31 Incorrect 0 ms 436 KB Tour is not fun
32 Halted 0 ms 0 KB -