답안 #635939

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
635939 2022-08-27T12:24:45 Z PoonYaPat Roller Coaster Railroad (IOI16_railroad) C++14
100 / 100
751 ms 87540 KB
#include "railroad.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef tuple<int,int,int> tu;

ll ans,cross[400010],dis[400010];
int p[400010],lv[400010];
vector<int> adj[400010];
vector<tu> v;
set<int> ss;
map<int,int> c;

void dfs(int x, int val) {
    p[x]=val;
    for (auto s : adj[x]) {
        if (p[s]==-1) dfs(s,val);
    }
}

int find(int x) {
    while (x!=p[x]) x=p[x];
    return x;
}

bool same(int x, int y) {
    return find(x)==find(y);
}

void unite(int x, int y) {
    x=find(x); y=find(y);
    if (lv[x]>lv[y]) p[y]=x;
    else if (lv[y]>lv[x]) p[x]=y;
    else {
        p[x]=y; ++lv[y];
    }
}

long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) {
    int n = (int) s.size();
    memset(p,-1,sizeof(p));
    for (int i=0; i<n; ++i) {
        ss.insert(s[i]);
        ss.insert(t[i]);
    }
    ss.insert(1000000010);

    int cnt=0,pre=0;
    for (auto it=ss.begin(); it!=ss.end(); ++it) {
        if (cnt) v.push_back(tu(*it-pre,cnt,cnt-1)), dis[cnt]=*it-pre;
        c[*it]=cnt++;
        pre=*it;
    }
    --cnt;

    for (int i=0; i<n; ++i) {
        int st=c[s[i]], to=c[t[i]];
        cross[st+1]++;
        cross[to+1]--;
        adj[st].push_back(to);
    }
    adj[cnt].push_back(0);
    --cross[1];

    for (int i=1; i<=cnt; ++i) {
        cross[i]+=cross[i-1];
        if (cross[i]>0) {
            adj[i].push_back(i-1);
            ans+=dis[i]*cross[i];
        } else if (cross[i]<0) {
            adj[i-1].push_back(i);
        }
    }

    for (int i=0; i<=cnt; ++i) {
        if (p[i]==-1) dfs(i,i), lv[i]=1;
    }

    sort(v.begin(),v.end());
    for (auto s : v) {
        int a,b,w;
        tie(w,a,b)=s;

        if (!same(a,b)) {
            unite(a,b);
            ans+=w;
        }
    }

    return ans;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 11220 KB n = 2
2 Correct 6 ms 11252 KB n = 2
3 Correct 6 ms 11260 KB n = 2
4 Correct 6 ms 11220 KB n = 2
5 Correct 6 ms 11220 KB n = 2
6 Correct 6 ms 11220 KB n = 2
7 Correct 5 ms 11220 KB n = 3
8 Correct 6 ms 11220 KB n = 3
9 Correct 6 ms 11220 KB n = 3
10 Correct 5 ms 11220 KB n = 8
11 Correct 6 ms 11220 KB n = 8
12 Correct 6 ms 11240 KB n = 8
13 Correct 6 ms 11220 KB n = 8
14 Correct 6 ms 11220 KB n = 8
15 Correct 6 ms 11220 KB n = 8
16 Correct 6 ms 11220 KB n = 8
17 Correct 6 ms 11220 KB n = 8
18 Correct 6 ms 11220 KB n = 8
19 Correct 6 ms 11272 KB n = 3
20 Correct 6 ms 11220 KB n = 7
21 Correct 6 ms 11220 KB n = 8
22 Correct 5 ms 11176 KB n = 8
23 Correct 6 ms 11220 KB n = 8
24 Correct 6 ms 11220 KB n = 8
25 Correct 7 ms 11220 KB n = 8
# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 11220 KB n = 2
2 Correct 6 ms 11252 KB n = 2
3 Correct 6 ms 11260 KB n = 2
4 Correct 6 ms 11220 KB n = 2
5 Correct 6 ms 11220 KB n = 2
6 Correct 6 ms 11220 KB n = 2
7 Correct 5 ms 11220 KB n = 3
8 Correct 6 ms 11220 KB n = 3
9 Correct 6 ms 11220 KB n = 3
10 Correct 5 ms 11220 KB n = 8
11 Correct 6 ms 11220 KB n = 8
12 Correct 6 ms 11240 KB n = 8
13 Correct 6 ms 11220 KB n = 8
14 Correct 6 ms 11220 KB n = 8
15 Correct 6 ms 11220 KB n = 8
16 Correct 6 ms 11220 KB n = 8
17 Correct 6 ms 11220 KB n = 8
18 Correct 6 ms 11220 KB n = 8
19 Correct 6 ms 11272 KB n = 3
20 Correct 6 ms 11220 KB n = 7
21 Correct 6 ms 11220 KB n = 8
22 Correct 5 ms 11176 KB n = 8
23 Correct 6 ms 11220 KB n = 8
24 Correct 6 ms 11220 KB n = 8
25 Correct 7 ms 11220 KB n = 8
26 Correct 6 ms 11220 KB n = 8
27 Correct 5 ms 11220 KB n = 8
28 Correct 6 ms 11260 KB n = 8
29 Correct 6 ms 11220 KB n = 16
30 Correct 6 ms 11220 KB n = 16
31 Correct 5 ms 11220 KB n = 16
32 Correct 5 ms 11220 KB n = 16
33 Correct 6 ms 11196 KB n = 16
34 Correct 5 ms 11248 KB n = 16
35 Correct 5 ms 11220 KB n = 16
36 Correct 6 ms 11220 KB n = 15
37 Correct 6 ms 11220 KB n = 8
38 Correct 7 ms 11220 KB n = 16
39 Correct 6 ms 11220 KB n = 16
40 Correct 6 ms 11220 KB n = 9
41 Correct 6 ms 11220 KB n = 16
42 Correct 7 ms 11220 KB n = 16
43 Correct 6 ms 11220 KB n = 16
44 Correct 6 ms 11220 KB n = 9
45 Correct 6 ms 11220 KB n = 15
46 Correct 6 ms 11220 KB n = 16
47 Correct 6 ms 11272 KB n = 16
48 Correct 5 ms 11292 KB n = 16
# 결과 실행 시간 메모리 Grader output
1 Correct 722 ms 83508 KB n = 199999
2 Correct 741 ms 78916 KB n = 199991
3 Correct 708 ms 83256 KB n = 199993
4 Correct 506 ms 62444 KB n = 152076
5 Correct 273 ms 42728 KB n = 93249
6 Correct 598 ms 66356 KB n = 199910
7 Correct 671 ms 77952 KB n = 199999
8 Correct 562 ms 68000 KB n = 199997
9 Correct 604 ms 69056 KB n = 171294
10 Correct 475 ms 58660 KB n = 140872
11 Correct 600 ms 67184 KB n = 199886
12 Correct 695 ms 79264 KB n = 199996
13 Correct 568 ms 71412 KB n = 200000
14 Correct 575 ms 81208 KB n = 199998
15 Correct 581 ms 79048 KB n = 200000
16 Correct 575 ms 80812 KB n = 199998
17 Correct 626 ms 79464 KB n = 200000
18 Correct 614 ms 76376 KB n = 190000
19 Correct 546 ms 75492 KB n = 177777
20 Correct 257 ms 45160 KB n = 100000
21 Correct 660 ms 78892 KB n = 200000
22 Correct 644 ms 76980 KB n = 200000
23 Correct 726 ms 83624 KB n = 200000
# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 11220 KB n = 2
2 Correct 6 ms 11252 KB n = 2
3 Correct 6 ms 11260 KB n = 2
4 Correct 6 ms 11220 KB n = 2
5 Correct 6 ms 11220 KB n = 2
6 Correct 6 ms 11220 KB n = 2
7 Correct 5 ms 11220 KB n = 3
8 Correct 6 ms 11220 KB n = 3
9 Correct 6 ms 11220 KB n = 3
10 Correct 5 ms 11220 KB n = 8
11 Correct 6 ms 11220 KB n = 8
12 Correct 6 ms 11240 KB n = 8
13 Correct 6 ms 11220 KB n = 8
14 Correct 6 ms 11220 KB n = 8
15 Correct 6 ms 11220 KB n = 8
16 Correct 6 ms 11220 KB n = 8
17 Correct 6 ms 11220 KB n = 8
18 Correct 6 ms 11220 KB n = 8
19 Correct 6 ms 11272 KB n = 3
20 Correct 6 ms 11220 KB n = 7
21 Correct 6 ms 11220 KB n = 8
22 Correct 5 ms 11176 KB n = 8
23 Correct 6 ms 11220 KB n = 8
24 Correct 6 ms 11220 KB n = 8
25 Correct 7 ms 11220 KB n = 8
26 Correct 6 ms 11220 KB n = 8
27 Correct 5 ms 11220 KB n = 8
28 Correct 6 ms 11260 KB n = 8
29 Correct 6 ms 11220 KB n = 16
30 Correct 6 ms 11220 KB n = 16
31 Correct 5 ms 11220 KB n = 16
32 Correct 5 ms 11220 KB n = 16
33 Correct 6 ms 11196 KB n = 16
34 Correct 5 ms 11248 KB n = 16
35 Correct 5 ms 11220 KB n = 16
36 Correct 6 ms 11220 KB n = 15
37 Correct 6 ms 11220 KB n = 8
38 Correct 7 ms 11220 KB n = 16
39 Correct 6 ms 11220 KB n = 16
40 Correct 6 ms 11220 KB n = 9
41 Correct 6 ms 11220 KB n = 16
42 Correct 7 ms 11220 KB n = 16
43 Correct 6 ms 11220 KB n = 16
44 Correct 6 ms 11220 KB n = 9
45 Correct 6 ms 11220 KB n = 15
46 Correct 6 ms 11220 KB n = 16
47 Correct 6 ms 11272 KB n = 16
48 Correct 5 ms 11292 KB n = 16
49 Correct 722 ms 83508 KB n = 199999
50 Correct 741 ms 78916 KB n = 199991
51 Correct 708 ms 83256 KB n = 199993
52 Correct 506 ms 62444 KB n = 152076
53 Correct 273 ms 42728 KB n = 93249
54 Correct 598 ms 66356 KB n = 199910
55 Correct 671 ms 77952 KB n = 199999
56 Correct 562 ms 68000 KB n = 199997
57 Correct 604 ms 69056 KB n = 171294
58 Correct 475 ms 58660 KB n = 140872
59 Correct 600 ms 67184 KB n = 199886
60 Correct 695 ms 79264 KB n = 199996
61 Correct 568 ms 71412 KB n = 200000
62 Correct 575 ms 81208 KB n = 199998
63 Correct 581 ms 79048 KB n = 200000
64 Correct 575 ms 80812 KB n = 199998
65 Correct 626 ms 79464 KB n = 200000
66 Correct 614 ms 76376 KB n = 190000
67 Correct 546 ms 75492 KB n = 177777
68 Correct 257 ms 45160 KB n = 100000
69 Correct 660 ms 78892 KB n = 200000
70 Correct 644 ms 76980 KB n = 200000
71 Correct 726 ms 83624 KB n = 200000
72 Correct 736 ms 83444 KB n = 200000
73 Correct 751 ms 78940 KB n = 200000
74 Correct 724 ms 82412 KB n = 200000
75 Correct 652 ms 87468 KB n = 200000
76 Correct 650 ms 87432 KB n = 200000
77 Correct 366 ms 52772 KB n = 200000
78 Correct 387 ms 49568 KB n = 200000
79 Correct 667 ms 75996 KB n = 184307
80 Correct 219 ms 38472 KB n = 76040
81 Correct 635 ms 70048 KB n = 199981
82 Correct 603 ms 82480 KB n = 199994
83 Correct 589 ms 74284 KB n = 199996
84 Correct 615 ms 84396 KB n = 199998
85 Correct 566 ms 82072 KB n = 200000
86 Correct 631 ms 84200 KB n = 199998
87 Correct 699 ms 83460 KB n = 200000
88 Correct 709 ms 83484 KB n = 200000
89 Correct 679 ms 87376 KB n = 200000
90 Correct 662 ms 82764 KB n = 200000
91 Correct 698 ms 80868 KB n = 200000
92 Correct 746 ms 87540 KB n = 200000