oj.uz

Submission #920993

# Submission time Handle Problem Language Result Execution time Memory
920993 2024-02-03T08:43:32 Z danikoynov Roller Coaster Railroad (IOI16_railroad) C++14
64 / 100
 126 ms 22680 KB 
#include "railroad.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const ll inf = 1e18;

const int maxn = 2e5 + 10;

ll dp[17][(1 << 17)];

pair < int, int > sec[maxn];

int sp[maxn];

int find_leader(int v)
{
    if (v == sp[v])
        return v;
    return (sp[v] = find_leader(sp[v]));
}
long long plan_roller_coaster(std::vector<int> s, std::vector<int> t)
{
    int n = (int) s.size();
    if (n <= 17)
    {
        for (int i = 0; i < n; i ++)
            for (int mask = 0; mask < (1 << n); mask ++)
                dp[i][mask] = inf;

        for (int i = 0; i < n; i ++)
            dp[i][(1 << i)] = 0;

        for (ll mask = 0; mask < (1 << n); mask ++)
        {
            for (int i = 0; i < n; i ++)
            {
                if ((mask & (1 << i)) == 0)
                    continue;
                //cout << "state " << i << " " << mask << " " << dp[i][mask] << endl;
                for (int j = 0; j < n; j ++)
                {
                    if ((mask & (1 << j)) != 0)
                        continue;

                    int new_mask = (mask | (1 << j));
                    //cout << "transition " << j << " " << max(0, t[i] - s[j]) << " " << t[i] << " " << s[j] << endl;
                    dp[j][new_mask] = min(dp[j][new_mask], dp[i][mask] + (ll)max(0, t[i] - s[j]));
                }
            }
        }
        ll ans = inf;
        for (int i = 0; i < n; i ++)
            ans = min(ans, dp[i][(1 << n) - 1]);
        return ans;
    }
    else
    {

        for (int i = 0; i < n; i ++)
        {
            ///assert(t[i] != 0 && s[i] != 0);
            sec[i + 1] = {t[i], s[i]};
        }

        sort(sec + 1, sec + n + 1);

        vector < pair < int, int > > ord;
        for (int i = 1; i <= n; i ++)
        {
            //assert(sec[i].first != 0 && sec[i].second != 0);
            ord.push_back({sec[i].second, i});
        }


        sort(ord.begin(), ord.end());
        set < int > act;
        for (int i = 1; i <= n; i ++)
            act.insert(i);

        for (int i = 1; i <= n; i ++)
            sp[i] = i;

        bool start = false;
        int cnt = 0;
        for (pair < int, int > cur : ord)
        {
            //cnt ++;
            //if (cnt % 10000 == 0 || cnt > 91000)
            //cout << cnt << " " << cur.second << " " << cur.first << " "  << endl;
            int lf = 1, rf = n;
            while(lf <= rf)
            {
                int mf = (lf + rf) / 2;
                if (sec[mf].first <= cur.first)
                    lf = mf + 1;
                else
                    rf = mf - 1;
            }

            set < int > :: iterator it = act.lower_bound(lf);
            if (it == act.begin())
            {

                if (start == true)
                    return 1;
                start = true;
            }
            else
            {
                it = prev(it);
                //if (cnt % 10000 == 0 || cnt > 91000)
                ///cout << "fine " << " " << *it << " " << find_leader(sp[cur.second]) << endl;
                if (*it == find_leader(sp[cur.second]))
                {
                    if (it == act.begin())
                    {
                        if (start == true)
                            return 1;
                        ///cout << "fuck" << endl;
                        start = true;
                        continue;
                    }
                    else
                    {
                        it = prev(it);
                    }
                }
                sp[*it] = sp[cur.second];
                act.erase(it);
            }
        }
        return 0;
    }
}

Compilation message

railroad.cpp: In function 'long long int plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:85:13: warning: unused variable 'cnt' [-Wunused-variable]
   85 |         int cnt = 0;
      |             ^~~

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2396 KB n = 2
2 Correct 0 ms 2396 KB n = 2
3 Correct 1 ms 2396 KB n = 2
4 Correct 0 ms 2396 KB n = 2
5 Correct 0 ms 2396 KB n = 2
6 Correct 1 ms 2396 KB n = 2
7 Correct 1 ms 4440 KB n = 3
8 Correct 1 ms 4540 KB n = 3
9 Correct 1 ms 4444 KB n = 3
10 Correct 2 ms 8540 KB n = 8
11 Correct 1 ms 8540 KB n = 8
12 Correct 2 ms 8540 KB n = 8
13 Correct 2 ms 8540 KB n = 8
14 Correct 2 ms 8584 KB n = 8
15 Correct 2 ms 8540 KB n = 8
16 Correct 1 ms 8540 KB n = 8
17 Correct 1 ms 8540 KB n = 8
18 Correct 1 ms 8540 KB n = 8
19 Correct 1 ms 4440 KB n = 3
20 Correct 2 ms 8540 KB n = 7
21 Correct 2 ms 8540 KB n = 8
22 Correct 2 ms 8540 KB n = 8
23 Correct 1 ms 8540 KB n = 8
24 Correct 1 ms 8540 KB n = 8
25 Correct 1 ms 8540 KB n = 8

Subtask #2
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2396 KB n = 2
2 Correct 0 ms 2396 KB n = 2
3 Correct 1 ms 2396 KB n = 2
4 Correct 0 ms 2396 KB n = 2
5 Correct 0 ms 2396 KB n = 2
6 Correct 1 ms 2396 KB n = 2
7 Correct 1 ms 4440 KB n = 3
8 Correct 1 ms 4540 KB n = 3
9 Correct 1 ms 4444 KB n = 3
10 Correct 2 ms 8540 KB n = 8
11 Correct 1 ms 8540 KB n = 8
12 Correct 2 ms 8540 KB n = 8
13 Correct 2 ms 8540 KB n = 8
14 Correct 2 ms 8584 KB n = 8
15 Correct 2 ms 8540 KB n = 8
16 Correct 1 ms 8540 KB n = 8
17 Correct 1 ms 8540 KB n = 8
18 Correct 1 ms 8540 KB n = 8
19 Correct 1 ms 4440 KB n = 3
20 Correct 2 ms 8540 KB n = 7
21 Correct 2 ms 8540 KB n = 8
22 Correct 2 ms 8540 KB n = 8
23 Correct 1 ms 8540 KB n = 8
24 Correct 1 ms 8540 KB n = 8
25 Correct 1 ms 8540 KB n = 8
26 Correct 1 ms 8540 KB n = 8
27 Correct 1 ms 8540 KB n = 8
28 Correct 2 ms 8540 KB n = 8
29 Correct 33 ms 16732 KB n = 16
30 Correct 33 ms 16732 KB n = 16
31 Correct 33 ms 16728 KB n = 16
32 Correct 32 ms 16728 KB n = 16
33 Correct 32 ms 16728 KB n = 16
34 Correct 32 ms 16732 KB n = 16
35 Correct 33 ms 16732 KB n = 16
36 Correct 16 ms 16732 KB n = 15
37 Correct 1 ms 8540 KB n = 8
38 Correct 33 ms 16728 KB n = 16
39 Correct 32 ms 16732 KB n = 16
40 Correct 2 ms 10588 KB n = 9
41 Correct 32 ms 16728 KB n = 16
42 Correct 33 ms 16728 KB n = 16
43 Correct 32 ms 16732 KB n = 16
44 Correct 2 ms 10588 KB n = 9
45 Correct 16 ms 16728 KB n = 15
46 Correct 32 ms 16728 KB n = 16
47 Correct 33 ms 16832 KB n = 16
48 Correct 33 ms 16732 KB n = 16

Subtask #3
30.0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 124 ms 18624 KB n = 199999
2 Correct 96 ms 19000 KB n = 199991
3 Correct 96 ms 18624 KB n = 199993
4 Correct 91 ms 13248 KB n = 152076
5 Correct 55 ms 9168 KB n = 93249
6 Correct 113 ms 18628 KB n = 199910
7 Correct 117 ms 18576 KB n = 199999
8 Correct 111 ms 18628 KB n = 199997
9 Correct 106 ms 14596 KB n = 171294
10 Correct 88 ms 12480 KB n = 140872
11 Correct 87 ms 18628 KB n = 199886
12 Correct 86 ms 18628 KB n = 199996
13 Correct 84 ms 18624 KB n = 200000
14 Correct 85 ms 18668 KB n = 199998
15 Correct 84 ms 18624 KB n = 200000
16 Correct 87 ms 18628 KB n = 199998
17 Correct 85 ms 18792 KB n = 200000
18 Correct 83 ms 17912 KB n = 190000
19 Correct 87 ms 15040 KB n = 177777
20 Correct 55 ms 9668 KB n = 100000
21 Correct 122 ms 18588 KB n = 200000
22 Correct 101 ms 18596 KB n = 200000
23 Correct 121 ms 18624 KB n = 200000

Subtask #4
0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2396 KB n = 2
2 Correct 0 ms 2396 KB n = 2
3 Correct 1 ms 2396 KB n = 2
4 Correct 0 ms 2396 KB n = 2
5 Correct 0 ms 2396 KB n = 2
6 Correct 1 ms 2396 KB n = 2
7 Correct 1 ms 4440 KB n = 3
8 Correct 1 ms 4540 KB n = 3
9 Correct 1 ms 4444 KB n = 3
10 Correct 2 ms 8540 KB n = 8
11 Correct 1 ms 8540 KB n = 8
12 Correct 2 ms 8540 KB n = 8
13 Correct 2 ms 8540 KB n = 8
14 Correct 2 ms 8584 KB n = 8
15 Correct 2 ms 8540 KB n = 8
16 Correct 1 ms 8540 KB n = 8
17 Correct 1 ms 8540 KB n = 8
18 Correct 1 ms 8540 KB n = 8
19 Correct 1 ms 4440 KB n = 3
20 Correct 2 ms 8540 KB n = 7
21 Correct 2 ms 8540 KB n = 8
22 Correct 2 ms 8540 KB n = 8
23 Correct 1 ms 8540 KB n = 8
24 Correct 1 ms 8540 KB n = 8
25 Correct 1 ms 8540 KB n = 8
26 Correct 1 ms 8540 KB n = 8
27 Correct 1 ms 8540 KB n = 8
28 Correct 2 ms 8540 KB n = 8
29 Correct 33 ms 16732 KB n = 16
30 Correct 33 ms 16732 KB n = 16
31 Correct 33 ms 16728 KB n = 16
32 Correct 32 ms 16728 KB n = 16
33 Correct 32 ms 16728 KB n = 16
34 Correct 32 ms 16732 KB n = 16
35 Correct 33 ms 16732 KB n = 16
36 Correct 16 ms 16732 KB n = 15
37 Correct 1 ms 8540 KB n = 8
38 Correct 33 ms 16728 KB n = 16
39 Correct 32 ms 16732 KB n = 16
40 Correct 2 ms 10588 KB n = 9
41 Correct 32 ms 16728 KB n = 16
42 Correct 33 ms 16728 KB n = 16
43 Correct 32 ms 16732 KB n = 16
44 Correct 2 ms 10588 KB n = 9
45 Correct 16 ms 16728 KB n = 15
46 Correct 32 ms 16728 KB n = 16
47 Correct 33 ms 16832 KB n = 16
48 Correct 33 ms 16732 KB n = 16
49 Correct 124 ms 18624 KB n = 199999
50 Correct 96 ms 19000 KB n = 199991
51 Correct 96 ms 18624 KB n = 199993
52 Correct 91 ms 13248 KB n = 152076
53 Correct 55 ms 9168 KB n = 93249
54 Correct 113 ms 18628 KB n = 199910
55 Correct 117 ms 18576 KB n = 199999
56 Correct 111 ms 18628 KB n = 199997
57 Correct 106 ms 14596 KB n = 171294
58 Correct 88 ms 12480 KB n = 140872
59 Correct 87 ms 18628 KB n = 199886
60 Correct 86 ms 18628 KB n = 199996
61 Correct 84 ms 18624 KB n = 200000
62 Correct 85 ms 18668 KB n = 199998
63 Correct 84 ms 18624 KB n = 200000
64 Correct 87 ms 18628 KB n = 199998
65 Correct 85 ms 18792 KB n = 200000
66 Correct 83 ms 17912 KB n = 190000
67 Correct 87 ms 15040 KB n = 177777
68 Correct 55 ms 9668 KB n = 100000
69 Correct 122 ms 18588 KB n = 200000
70 Correct 101 ms 18596 KB n = 200000
71 Correct 121 ms 18624 KB n = 200000
72 Correct 126 ms 22428 KB n = 200000
73 Incorrect 98 ms 22680 KB answer is not correct: 1 instead of 34382661743
74 Halted 0 ms 0 KB -