CSES - Datatähti Open 2021

Submission details
Task:	Programmers
Sender:	endagorion
Submission time:	2021-01-31 08:32:49 +0200
Language:	C++ (C++17)
Status:	READY
Result:	100

Feedback
group	verdict	score
#1	ACCEPTED	34
#2	ACCEPTED	21
#3	ACCEPTED	45

Test results
test	verdict	time	group
#1	ACCEPTED	0.01 s	1, 2, 3	details
#2	ACCEPTED	0.01 s	1, 2, 3	details
#3	ACCEPTED	0.01 s	1, 2, 3	details
#4	ACCEPTED	0.01 s	1, 2, 3	details
#5	ACCEPTED	0.01 s	1, 2, 3	details
#6	ACCEPTED	0.01 s	1, 2, 3	details
#7	ACCEPTED	0.01 s	1, 2, 3	details
#8	ACCEPTED	0.01 s	1, 2, 3	details
#9	ACCEPTED	0.01 s	1, 2, 3	details
#10	ACCEPTED	0.01 s	1, 2, 3	details
#11	ACCEPTED	0.01 s	1, 2, 3	details
#12	ACCEPTED	0.01 s	1, 3	details
#13	ACCEPTED	0.01 s	1, 3	details
#14	ACCEPTED	0.01 s	1, 3	details
#15	ACCEPTED	0.01 s	1, 3	details
#16	ACCEPTED	0.01 s	1, 3	details
#17	ACCEPTED	0.04 s	2, 3	details
#18	ACCEPTED	0.11 s	2, 3	details
#19	ACCEPTED	0.10 s	2, 3	details
#20	ACCEPTED	0.12 s	2, 3	details
#21	ACCEPTED	0.12 s	2, 3	details
#22	ACCEPTED	0.12 s	2, 3	details
#23	ACCEPTED	0.12 s	2, 3	details
#24	ACCEPTED	0.35 s	3	details
#25	ACCEPTED	0.20 s	3	details
#26	ACCEPTED	0.38 s	3	details
#27	ACCEPTED	0.26 s	3	details
#28	ACCEPTED	0.37 s	3	details
#29	ACCEPTED	0.39 s	3	details
#30	ACCEPTED	0.39 s	3	details
#31	ACCEPTED	0.41 s	3	details

Code

#include <bits/stdc++.h>

#define mp make_pair
#define mt make_tuple
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define forn(i, n) for (int i = 0; i < (int)(n); ++i)
#define for1(i, n) for (int i = 1; i <= (int)(n); ++i)
#define ford(i, n) for (int i = (int)(n) - 1; i >= 0; --i)
#define fore(i, a, b) for (int i = (int)(a); i <= (int)(b); ++i)

using namespace std;

typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<pii> vpi;
typedef vector<vi> vvi;
typedef long long i64;
typedef vector<i64> vi64;
typedef vector<vi64> vvi64;
typedef pair<i64, i64> pi64;
typedef double ld;

template<class T> bool uin(T &a, T b) { return a > b ? (a = b, true) : false; }
template<class T> bool uax(T &a, T b) { return a < b ? (a = b, true) : false; }

const int maxn = 210000;
int par[maxn], l[maxn], r[maxn];
i64 a[maxn], b[maxn], ps[maxn];

int root(int x) {
    return x == par[x] ? x : par[x] = root(par[x]);
}

i64 segs(int l, int r) {
    ++r;
    return ps[r] + ps[l] * ((r - l) % 2 == 0 ? -1 : 1);
}

void unite(int i, int j) {
    i = root(i);
    j = root(j);
    uin(l[i], l[j]);
    uax(r[i], r[j]);
    par[j] = i;
}

int L[maxn], R[maxn];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.precision(10);
    cout << fixed;
#ifdef LOCAL_DEFINE
    freopen("input.txt", "rt", stdin);
#endif
    
    int n, k;
    cin >> n >> k;
    forn(i, n) cin >> a[i];
    sort(a, a + n);

    b[0] = 1e18;
    forn(i, n - 1) b[i + 1] = a[i + 1] - a[i];
    b[n] = 1e18;
    set<pi64> q;
    forn(i, n + 1) {
        par[i] = i;
        l[i] = r[i] = i;
        q.insert(mp(b[i], i));
    }

    i64 ans = 0;
    forn(i, k) {
        auto ww = *q.begin();
        ans += ww.fi;
        q.erase(ww);
        int v = ww.se;
        assert(l[v] && r[v] < n);
        int u = root(l[v] - 1), w = root(r[v] + 1);
        q.erase(mp(b[u], u));
        q.erase(mp(b[w], w));
        b[v] = b[u] + b[w] - b[v];
        unite(v, u);
        unite(v, w);
        b[root(v)] = b[v];
        v = root(v);
        q.insert(mp(b[v], v));
    }

    cout << ans << '\n';

#ifdef LOCAL_DEFINE
    cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
    return 0;
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
`8 3 3 1 2 7 9 3 4 7` view save

correct output
`1` view save

user output
`1` view save

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2 1 2 13` view save

correct output
`11` view save

user output
`11` view save

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
`20 10 16 20 6 15 19 12 11 17 20 6 15...` view save

correct output
`6` view save

user output
`6` view save

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

input
`14 5 11 3 8 3 14 8 10 13 11 10 17 1...` view save

correct output
`0` view save

user output
`0` view save

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
`15 1 8 5 1 8 18 15 6 20 14 9 10 9 1...` view save

correct output
`0` view save

user output
`0` view save

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

input
`10 3 10 19 16 15 12 5 14 8 3 15` view save

correct output
`4` view save

user output
`4` view save

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

input
`202 90 177 187 183 647 616 580 499 78...` view save

correct output
`213` view save

user output
`213` view save

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2000 512 141 583 135 833 900 308 248 58...` view save

correct output
`0` view save

user output
`0` view save

Test 9

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2000 972 685 4 289 865 93 159 48 866 56...` view save

correct output
`276` view save

user output
`276` view save

Test 10

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2000 1000 698 153 298 118 631 341 238 7 ...` view save

correct output
`517` view save

user output
`517` view save

Test 11

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2000 1 983 144 449 584 839 166 77 885...` view save

correct output
`0` view save

user output
`0` view save

Test 12

Group: 1, 3

Verdict: ACCEPTED

input
`1464 320 846762124 954854396 12767390 7...` view save

correct output
`35809369` view save

user output
`35809369` view save

Test 13

Group: 1, 3

Verdict: ACCEPTED

input
`2000 231 801945178 924940258 369188694 ...` view save

correct output
`7831421` view save

user output
`7831421` view save

Test 14

Group: 1, 3

Verdict: ACCEPTED

input
`2000 461 464790475 932031556 838378103 ...` view save

correct output
`37272564` view save

user output
`37272564` view save

Test 15

Group: 1, 3

Verdict: ACCEPTED

input
`2000 100 484046702 267135814 995006323 ...` view save

correct output
`1268400` view save

user output
`1268400` view save

Test 16

Group: 1, 3

Verdict: ACCEPTED

input
`2000 996 98352148 438929491 242618159 1...` view save

correct output
`445965905` view save

user output
`445965905` view save

Test 17

Group: 2, 3

Verdict: ACCEPTED

input
`65879 19675 896 316 972 476 636 227 716 78...` view save

correct output
`0` view save

user output
`0` view save

Test 18

Group: 2, 3

Verdict: ACCEPTED

input
`200000 53820 995 720 135 767 943 742 191 26...` view save

correct output
`0` view save

user output
`0` view save

Test 19

Group: 2, 3

Verdict: ACCEPTED

input
`200000 32297 527 947 84 851 908 833 339 112...` view save

correct output
`0` view save

user output
`0` view save

Test 20

Group: 2, 3

Verdict: ACCEPTED

input
`200000 99982 561 174 242 275 460 109 664 68...` view save

correct output
`322` view save

user output
`322` view save

Test 21

Group: 2, 3

Verdict: ACCEPTED

input
`200000 99955 911 33 314 861 298 117 972 982...` view save

correct output
`245` view save

user output
`245` view save

Test 22

Group: 2, 3

Verdict: ACCEPTED

input
`200000 99972 783 1000 673 611 87 452 702 92...` view save

correct output
`290` view save

user output
`290` view save

Test 23

Group: 2, 3

Verdict: ACCEPTED

input
`200000 99961 795 136 128 643 60 422 371 839...` view save

correct output
`252` view save

user output
`252` view save

Test 24

Group: 3

Verdict: ACCEPTED

input
`195426 76599 442872072 619088799 118541378 ...` view save

correct output
`143376538` view save

user output
`143376538` view save

Test 25

Group: 3

Verdict: ACCEPTED

input
`200000 1661 894106972 620084612 931442312 ...` view save

correct output
`33089` view save

user output
`33089` view save

Test 26

Group: 3

Verdict: ACCEPTED

input
`200000 86032 211444153 846442677 297198384 ...` view save

correct output
`196001810` view save

user output
`196001810` view save

Test 27

Group: 3

Verdict: ACCEPTED

input
`200000 28275 28280312 349705372 96535649 84...` view save

correct output
`11627219` view save

user output
`11627219` view save

Test 28

Group: 3

Verdict: ACCEPTED

input
`200000 81473 178022892 112774306 250584651 ...` view save

correct output
`162430841` view save

user output
`162430841` view save

Test 29

Group: 3

Verdict: ACCEPTED

input
`200000 99987 297598052 494409138 182268523 ...` view save

correct output
`489497036` view save

user output
`489497036` view save

Test 30

Group: 3

Verdict: ACCEPTED

input
`200000 99971 316462272 843156468 434342923 ...` view save

correct output
`483167476` view save

user output
`483167476` view save

Test 31

Group: 3

Verdict: ACCEPTED

input
`200000 99964 811543559 465033274 620180191 ...` view save

correct output
`481497328` view save

user output
`481497328` view save

Datatähti Open 2021

Programmers

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Test 31

Tasks

Submissions (endagorion)