QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#405286#437. Fun TourMax_s_xaM100 ✓220ms50744kbC++204.1kb2024-05-05 16:28:202024-05-05 16:28:21

Judging History

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

  • [2024-05-05 16:28:21]
  • 评测
  • 测评结果:100
  • 用时:220ms
  • 内存:50744kb
  • [2024-05-05 16:28:20]
  • 提交

answer

#include "fun.h"
#include <iostream>
#include <vector>
#include <algorithm>
#include <map>

typedef long long ll;
typedef double lf;

using namespace std;

const int MAXN = 1e5 + 10;

int n, q;
vector <int> res, ans;

map <pair <int, int>, int> dist, tsz;
inline int GetDist(int u, int v)
{
    if (dist[make_pair(u, v)]) return dist[make_pair(u, v)];
    return dist[make_pair(u, v)] = dist[make_pair(v, u)] = hoursRequired(u - 1, v - 1);
}
inline int GetSize(int u, int v)
{
    if (tsz[make_pair(u, v)]) return tsz[make_pair(u, v)];
    return tsz[make_pair(u, v)] = attractionsBehind(u - 1, v - 1);
}

int rt, sz[MAXN], dep[MAXN], bel[MAXN];
vector <int> son, vec[3];

inline void Solve2()
{
    if (vec[0].size() < vec[1].size()) swap(vec[0], vec[1]);
    for (int i = vec[0].size() - 1, j = vec[1].size() - 1; ~i; i--)
    {
        res.push_back(vec[0][i]);
        if (j == -1) res.push_back(rt);
        else res.push_back(vec[1][j--]);
    }
    if (res.size() != n) res.push_back(rt);
}

vector <int> createFunTour(int N, int Q)
{
    n = N, q = Q;
    if (n == 2)
    {
        res.push_back(0), res.push_back(1);
        return res;
    }
    sz[1] = n;
    for (int i = 2; i <= n; i++) sz[i] = GetSize(1, i);
    int mn = n; rt = 1;
    for (int i = 1; i <= n; i++)
        if (n - sz[i] <= n / 2 && mn > sz[i]) mn = sz[i], rt = i;
    for (int i = 1; i <= n; i++)
        if (i != rt)
        {
            dep[i] = GetDist(rt, i);
            if (dep[i] == 1) son.push_back(i);
        }
    bel[rt] = -1;
    for (int i = 1; i <= n; i++)
        if (dep[i] == 1)
        {
            for (int j = 0; j < son.size(); j++)
                if (son[j] == i) bel[i] = j;
        }
        else if (dep[i] > 1)
        {
            bel[i] = -1;
            for (int j = 0; j < son.size() - 1; j++)
                if (GetDist(i, son[j]) == dep[i] - 1) {bel[i] = j; break;}
            if (bel[i] == -1) bel[i] = son.size() - 1;
        }
    for (int i = 1; i <= n; i++)
        if (dep[i]) vec[bel[i]].push_back(i);
    for (int i = 0; i < son.size(); i++) sort(vec[i].begin(), vec[i].end(), [&](int x, int y) {return dep[x] < dep[y];});

    // cout << rt << '\n';
    // for (int i = 1; i <= n; i++) cout << bel[i] << ' '; cout << '\n';
    // for (int i = 1; i <= n; i++) cout << dep[i] << ' '; cout << '\n';

    if (son.size() == 2)
    {
        Solve2();
        for (auto x : res) ans.push_back(x - 1);
        // for (auto x : res) cout << x << " "; cout << '\n';
        return ans;
    }
    int num = n - 1, mxs; int pre = -1;
    while ((mxs = max(vec[0].size(), max(vec[1].size(), vec[2].size()))) < num - mxs)
    {
        int fr = 0, cur = 0;
        for (int i = 0; i < 3; i++)
            if (i != pre && vec[i].size() && fr < dep[vec[i].back()]) fr = dep[vec[i].back()], cur = i;
        res.push_back(vec[cur].back()), vec[cur].pop_back(), num--;
        pre = cur;
    }
    // cout << res.size() << "\n";
    // for (auto x : res) cout << x << ' '; cout << '\n';
    if (vec[0].size() < vec[1].size()) swap(vec[0], vec[1]);
    if (vec[0].size() < vec[2].size()) swap(vec[0], vec[2]);
    for (auto x : vec[2]) vec[1].push_back(x);
    sort(vec[1].begin(), vec[1].end(), [&](int x, int y) {return dep[x] < dep[y];});
    // for (auto x : vec[0]) cout << x << ' '; cout << '\n';
    // for (auto x : vec[1]) cout << x << ' '; cout << '\n';
    if (vec[1].size() == 0)
    {
        res.push_back(vec[0].back()), res.push_back(rt);
        for (auto x : res) ans.push_back(x - 1);
        // for (auto x : res) cout << x << " "; cout << '\n';
        return ans;
    }
    int fst1 = vec[0].back(), fst2 = vec[1].back();
    if (res.size() && (bel[fst1] == bel[res.back()] || (res.size() >= 2 && dep[res[res.size() - 2]] < dep[fst1])))
    {
        swap(vec[0], vec[1]);
        Solve2();
        if (res.size() != n) res.push_back(vec[1].front());
    }
    else Solve2();
    for (auto x : res) ans.push_back(x - 1);
    // for (auto x : res) cout << x << " "; cout << '\n';
    return ans;
}

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 0
Accepted

Test #1:

score: 0
Accepted
time: 1ms
memory: 3996kb

input:

2 400000
1 0

output:

0 1

result:

ok correct

Subtask #2:

score: 0
Accepted

Test #3:

score: 0
Accepted
time: 1ms
memory: 3764kb

input:

4 400000
1 0
2 0
3 1

output:

2 3 0 1

result:

ok correct

Subtask #3:

score: 0
Accepted

Test #9:

score: 0
Accepted
time: 0ms
memory: 3768kb

input:

4 400000
1 0
2 1
3 2

output:

0 3 1 2

result:

ok correct

Subtask #4:

score: 0
Accepted

Test #13:

score: 0
Accepted
time: 0ms
memory: 3800kb

input:

4 400000
0 1
0 2
0 3

output:

1 2 3 0

result:

ok correct

Subtask #5:

score: 0
Accepted

Test #20:

score: 0
Accepted
time: 1ms
memory: 4004kb

input:

18 400000
1 0
2 0
3 1
4 1
5 2
6 2
7 3
8 3
9 4
10 4
11 5
12 5
13 6
14 6
15 7
16 7
17 8

output:

14 17 13 16 12 15 11 8 6 7 5 10 2 9 0 4 3 1

result:

ok correct

Subtask #6:

score: 0
Accepted

Test #27:

score: 0
Accepted
time: 1ms
memory: 3728kb

input:

18 400000
7 6
8 7
5 8
4 5
3 4
9 3
10 11
12 10
13 12
14 13
9 14
2 9
15 2
16 15
17 16
1 17
0 1

output:

0 6 1 7 11 17 8 10 16 5 12 15 4 13 2 14 3 9

result:

ok correct

Subtask #7:

score: 0
Accepted

Test #33:

score: 0
Accepted
time: 1ms
memory: 4124kb

input:

500 400000
33 414
398 33
254 398
400 254
376 400
302 376
466 302
362 466
301 362
267 301
417 267
413 417
14 413
305 14
353 305
83 353
227 83
57 227
446 57
406 446
154 406
336 154
195 336
405 195
264 405
76 264
203 76
459 203
178 459
72 178
296 72
488 296
404 488
60 404
156 60
393 156
108 393
216 108...

output:

212 414 88 33 124 398 399 254 190 400 10 376 366 302 234 466 432 362 28 301 188 267 237 417 64 413 268 14 39 305 349 353 345 83 125 227 70 57 276 446 420 406 369 154 48 336 211 195 482 405 8 264 22 76 487 203 9 459 230 178 339 72 95 296 340 488 50 404 102 60 174 156 47 393 461 108 424 216 181 106 13...

result:

ok correct

Subtask #8:

score: 0
Accepted

Test #41:

score: 0
Accepted
time: 1ms
memory: 4080kb

input:

501 400000
1 0
2 0
3 1
4 1
5 2
6 2
7 3
8 3
9 4
10 4
11 5
12 5
13 6
14 6
15 7
16 7
17 8
18 8
19 9
20 9
21 10
22 10
23 11
24 11
25 12
26 12
27 13
28 13
29 14
30 14
31 15
32 15
33 16
34 16
35 17
36 17
37 18
38 18
39 19
40 19
41 20
42 20
43 21
44 21
45 22
46 22
47 23
48 23
49 24
50 24
51 25
52 25
53 26
...

output:

413 285 414 284 415 283 416 282 417 281 418 280 419 279 420 278 421 277 422 276 423 275 424 274 425 273 441 272 427 271 428 286 429 269 430 268 431 267 432 266 433 265 434 264 435 263 436 262 437 261 438 260 439 259 440 258 426 257 383 256 384 255 385 270 386 317 387 316 388 315 389 314 390 313 391 ...

result:

ok correct

Subtask #9:

score: 0
Accepted

Test #51:

score: 0
Accepted
time: 2ms
memory: 4572kb

input:

1000 400000
6 5
7 8
6 7
9 6
11 10
11 12
13 11
14 15
13 14
9 13
16 9
18 17
18 19
20 18
22 21
22 23
20 22
24 20
26 25
26 27
28 26
30 29
30 31
28 30
24 28
16 24
32 16
33 32
35 34
35 36
37 35
39 38
39 40
37 39
41 37
43 42
43 44
45 43
47 46
47 48
45 47
41 45
49 41
50 51
52 50
54 53
54 55
52 54
56 52
58 5...

output:

139 684 141 682 323 680 321 510 143 678 319 676 145 526 147 674 149 672 151 559 312 502 153 668 114 666 155 664 307 512 305 662 158 660 303 561 160 658 301 656 299 654 162 652 297 670 164 722 166 550 310 504 116 717 357 715 118 552 355 713 353 711 121 506 351 709 349 707 123 686 347 703 125 554 345 ...

result:

ok correct

Subtask #10:

score: 0
Accepted

Test #59:

score: 0
Accepted
time: 220ms
memory: 50744kb

input:

99999 400000
11681 58292
11681 63929
49752 11681
30596 74400
30596 39261
49752 30596
19390 49752
89694 31923
19390 89694
54297 19390
42389 12902
42389 60328
72803 42389
69881 43761
69881 95741
72803 69881
96271 72803
63872 20658
63872 93588
35833 63872
48418 44153
35833 48418
96271 35833
54297 96271...

output:

62996 54185 26009 13388 62998 54193 26002 54198 26001 54199 63008 54200 25999 54204 63016 54210 63017 54212 25997 13385 25996 54217 73666 54218 63031 54226 63033 69843 63036 54232 25989 54238 63040 54243 25986 54244 25985 54247 63062 54250 63064 54252 63065 54253 63066 54255 25968 54258 63030 13383 ...

result:

ok correct

Subtask #11:

score: 10
Accepted

Dependency #1:

100%
Accepted

Dependency #2:

100%
Accepted

Dependency #3:

100%
Accepted

Dependency #4:

100%
Accepted

Subtask #12:

score: 16
Accepted

Dependency #1:

100%
Accepted

Dependency #2:

100%
Accepted

Dependency #3:

100%
Accepted

Dependency #4:

100%
Accepted

Dependency #5:

100%
Accepted

Dependency #6:

100%
Accepted

Dependency #7:

100%
Accepted

Subtask #13:

score: 21
Accepted

Dependency #1:

100%
Accepted

Dependency #2:

100%
Accepted

Dependency #5:

100%
Accepted

Dependency #8:

100%
Accepted

Subtask #14:

score: 19
Accepted

Dependency #1:

100%
Accepted

Dependency #3:

100%
Accepted

Dependency #6:

100%
Accepted

Dependency #9:

100%
Accepted

Subtask #15:

score: 34
Accepted

Dependency #1:

100%
Accepted

Dependency #2:

100%
Accepted

Dependency #3:

100%
Accepted

Dependency #4:

100%
Accepted

Dependency #5:

100%
Accepted

Dependency #6:

100%
Accepted

Dependency #7:

100%
Accepted

Dependency #8:

100%
Accepted

Dependency #9:

100%
Accepted

Dependency #10:

100%
Accepted

Dependency #11:

100%
Accepted

Dependency #12:

100%
Accepted

Dependency #13:

100%
Accepted

Dependency #14:

100%
Accepted

Extra Test:

score: 0
Extra Test Passed