제출 #613784

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
613784		drdilyor	고대 책들 (IOI17_books)	C++17	22 / 100	188 ms	43156 KiB

books

#include <bits/stdc++.h>
#include "books.h"
#ifdef ONPC
    #include "t_debug.cpp"
#else
    #define debug(...) 42
#endif
#define sz(a) ((int)(a).size())
using namespace std;
using ll = long long;
const int INF = 1e9;
const ll INFL = 1e18;
const int RANDOM = chrono::high_resolution_clock::now().time_since_epoch().count();
mt19937 rng(RANDOM);
template<typename T, typename U> istream& operator>>(istream& is, pair<T, U>& p) { return is >> p.first >> p.second; }
const int N = 2e5, LOGN = 17, MOD = 1e9+7;

struct DSU {
    int n;
    vector<int> parent;
    vector<int> size;

    DSU(int n) : n(n), parent(n), size(n, 1) {
        iota(parent.begin(), parent.end(), 0);
    }

    int get(int i) {
        return parent[i] == i ? i : parent[i] = get(parent[i]);
    }

    void merge(int v, int u) {
        v = get(v);
        u = get(u);
        if (u == v) return;
        if (size[v] < size[u]) swap(v, u);
        parent[v] = u;
        size[v] += size[u];
    }
};

long long minimum_walk(std::vector<int> p, int s) {
    int n = sz(p);
    vector<vector<int>> cycles;
    vector<pair<int,int>> range;
    vector<int> done(n);
    int distance = 0;
    for (int i = 0; i < n; i++) {
        distance += abs(p[i] - i);
        if (done[i]) continue;
        if (p[i] == i) continue;
        done[i] = true;
        cycles.push_back({i});
        for (int cur = p[i]; !done[cur]; cur = p[cur]) {
            cycles.back().push_back(cur);
            done[cur] = true;
        }
    }

    int k = sz(cycles);
    if (!k) return 0;

    for (int i = 0; i < k; i++) {
        int l = INF, r = 0;
        for (int i : cycles[i]) {
            l = min(l, i);
            r = max(r, i);
        }
        range.emplace_back(l, r);
    }

    int start = INF, end = 0;
    DSU groups(n);
    for (int i = 0; i < k; i++) {
        for (int j = range[i].first;j < range[i].second; j++) {
            groups.merge(j, j+1);
        }
        start = min(range[i].first, start);
        end = max(range[i].second, end);
    }
    int disjoint = 0;
    for (int i = start; i < end-1; i++) {
        if (groups.get(i) != groups.get(i+1)) {
            disjoint++;
        }
    }

    int waste = disjoint * 2;
    debug(cycles, range);
    debug(waste, disjoint);

    return distance + start * 2 + waste;
}

컴파일 시 표준 에러 (stderr) 메시지

books.cpp: In function 'long long int minimum_walk(std::vector<int>, int)':
books.cpp:6:24: warning: statement has no effect [-Wunused-value]
    6 |     #define debug(...) 42
      |                        ^~
books.cpp:88:5: note: in expansion of macro 'debug'
   88 |     debug(cycles, range);
      |     ^~~~~
books.cpp:6:24: warning: statement has no effect [-Wunused-value]
    6 |     #define debug(...) 42
      |                        ^~
books.cpp:89:5: note: in expansion of macro 'debug'
   89 |     debug(waste, disjoint);
      |     ^~~~~

• Subtask 1: 12 / 12
• Subtask 2: 10 / 10
• Subtask 3: 0 / 28
• Subtask 4: 0 / 20
• Subtask 5: 0 / 30