oj.uz

Submission #296559

# Submission time Handle Problem Language Result Execution time Memory
296559 2020-09-10T16:14:44 Z Aldas25 Ancient Books (IOI17_books) C++14
0 / 100
 13 ms 9596 KB 
#include "books.h"
#include <bits/stdc++.h>

using namespace std;

#define FAST_IO ios_base::sync_with_stdio(0); cin.tie(nullptr)
#define FOR(i, a, b) for(int i = (a); i <= (b); i++)
#define REP(n) FOR(O, 1, (n))
#define pb push_back
#define f first
#define s second
typedef long double ld;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<int, pii> piii;
typedef vector<int> vi;
typedef vector<pii> vii;
typedef vector<ll> vl;
typedef vector<piii> viii;

const int MAXN = 1000100;

int n;
ll d[1100][1100];

int par[MAXN], sz[MAXN];
int find (int a) {return par[a] = par[a]==a ? a : find(par[a]);}
bool same (int a, int b) {return find(a) == find(b);}
void unite (int a, int b) {
    a = find(a), b = find (b);
    if (a == b) return;
    par[b] = a;
    sz[a] += sz[b];
}

vector<pair<ll, pii>> edges;

long long minimum_walk(std::vector<int> p, int s) {
    n = (int)p.size();
    ll ret = 0ll;
    FOR(i,0, n-1) ret += abs(i - p[i]);

    FOR(i, 0, n-1) par[i] = i, sz[i] = 1;
    FOR(i, 0, n-1) unite (i, p[i]);

    FOR(i, 0, n-1) FOR(j, 0, n-1) d[i][j] = n + 5;
    FOR(i, 0, n-1) FOR(j, i+1, n-1) {
        int pi = find(i), pj = find(j);
        d[pi][pj] = min(d[pi][pj], (ll)abs(j-i));
        //d[pj][pi] = min(d[pj][pi], abs(j-i));
    }

    FOR(i, 0, n-1) FOR(j, i+1, n-1) {
        if (find(i) != i || find(j) != j) continue;
        if (i != s && j != s && (sz[i] == 1 || sz[j] == 1)) continue;
        edges.pb({d[i][j], {i,j}});
    }

    sort(edges.begin(), edges.end());
    ll sum = 0ll;
    FOR(i, 0, n-1) par[i] = i;
    for (auto e : edges) {
        ll w = e.f;
        int u = e.s.f, v = e.s.s;
        if (same(u,v)) continue;
        unite(u,v);
        sum += w;
    }

    ret += 2*sum;

	return ret;
}

Subtask #1
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Incorrect 1 ms 384 KB 3rd lines differ - on the 1st token, expected: '6', found: '10'
3 Halted 0 ms 0 KB -

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Incorrect 1 ms 384 KB 3rd lines differ - on the 1st token, expected: '6', found: '10'
3 Halted 0 ms 0 KB -

Subtask #3
0 / 28.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Incorrect 1 ms 384 KB 3rd lines differ - on the 1st token, expected: '6', found: '10'
3 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 13 ms 9596 KB 3rd lines differ - on the 1st token, expected: '3304', found: '275314'
2 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Incorrect 1 ms 384 KB 3rd lines differ - on the 1st token, expected: '6', found: '10'
3 Halted 0 ms 0 KB -