Submission #924165

#	Time	Username	Problem	Language	Result	Execution time	Memory
924165		Camillus	Floppy (RMI20_floppy)	C++17	28 / 100	39 ms	7064 KiB

floppy

#include "bits/stdc++.h"
using namespace std;

#ifndef LOCAL
#include "floppy.h"
#else
void read_array(int subtask_id, const vector<int> &v);
void save_to_floppy(const string &bits);
vector<int> solve_queries(int subtask_id, int N, const string &bits, const vector<int> &a, const vector<int> &b);
#endif





void read_array(int subtask_id, const vector<int> &v) {
    int n = (int)v.size();

    int len = __lg(n << 1);

    auto a = v;
    sort(a.begin(), a.end());

    map<int, int> b;
    for (int i = 0; i < n; i++) {
        b[a[i] + (int)1e9] = i;
    }

    string s;

    for (int i = 0; i < n; i++) {
        string c = bitset<32>(b[v[i] + (int)1e9]).to_string();
        s += string(c.end() - len, c.end());
    }
    // debug(s);
    save_to_floppy(s);
}



struct segment_tree {
    int n;
    vector<int> tree;

    segment_tree(int n) : n(n) {
        tree.resize(4 * n);
    }

    void set(int i, int v, int x = 0, int lx = 0, int rx = -1) {
        if (rx == -1) {
            rx = n;
        }

        if (rx - lx == 1) {
            tree[x] = v;
            return;
        }

        int mx = (lx + rx) / 2;

        if (i < mx) {
            set(i, v, x * 2 + 1, lx, mx);
        } else {
            set(i, v, x * 2 + 2, mx, rx);
        }

        tree[x] = max(tree[x * 2 + 1], tree[x * 2 + 2]);
    }

    int get(int l, int r, int x = 0, int lx = 0, int rx = -1) const {
        if (rx == -1) {
            rx = n;
        }

        if (l <= lx && rx <= r) {
            return tree[x];
        }

        if (l >= rx || lx >= r) {
            return 0;
        }

        return max(
            get(l, r, x * 2 + 1, lx, (lx + rx) / 2),
            get(l, r, x * 2 + 2, (lx + rx) / 2, rx)
        );
    }
};



vector<int> solve_queries(int subtask_id, int N, const string &bits, const vector<int> &a, const vector<int> &b) {
    int m = (int)a.size();
    int n = N;

    int len = __lg(n << 1);

    vector<int> A;

    for (int i = 0; i < n; i++) {
        A.push_back(bitset<32>(string(bits.begin() + i * len, bits.begin() + (i + 1) * len)).to_ulong());
    }

    // debug(A);

    segment_tree st(n);
    map<int, int> pos;

    for (int i = 0; i < n; i++) {
        pos[A[i]] = i;
        st.set(i, A[i]);
    }

    vector<int> result;

    for (int i = 0; i < m; i++) {
        result.push_back(pos[st.get(a[i], b[i] + 1)]);
    }

    // debug(result);

    return result;
}













#ifdef LOCAL
#define NMAX 100000
#define MMAX 100000

int subtask_id, N, M;
vector<int> v, sorted_v;
vector<int> a, b;
vector<int> correct_answers;

// Print score to stdout and exit.
void score_and_exit(const double pts, const char *verdict) {
    fprintf(stderr, "%s", verdict);
    fprintf(stdout, "%f", pts);
    exit(0);
}

// Contestant sent too many bits.
void too_many_bits() {
    score_and_exit(0, "Too many stored bits!");
}

// Contestant did not send any bits.
void misformatted_stored_bits() {
    score_and_exit(0, "Misformatted stored bits or save_to_floppy not called!");
}

// Contestant did not call the answer function.
void answer_not_provided() {
    score_and_exit(0, "Answer not provided!");
}

// Contestant sent a wrong answer.
void wrong_answer() {
    score_and_exit(0, "Wrong answer to query!");
}

// Contestant sent a correct answer.
void correct_answer() {
    score_and_exit(1, "OK!");
}

void read_test() {
    assert(scanf("%d", &subtask_id) == 1);
    assert(scanf("%d%d", &N, &M) == 2);
    assert(1 <= N && N <= NMAX);
    assert(0 <= M && M <= MMAX);
    v.resize(N);
    for (int i = 0; i < N; ++i) {
        assert(scanf("%d", &v[i]) == 1);
    }

    // Check all values are distinct.
    sorted_v.resize(N);
    for (int i = 0; i < N; ++i) {
        sorted_v[i] = v[i];
    }
    sort(sorted_v.begin(), sorted_v.end());
    for (int i = 0; i + 1 < N; ++i) {
        assert(sorted_v[i] < sorted_v[i + 1]);
    }

    a.resize(M);
    b.resize(M);
    correct_answers.resize(M);
    for (int i = 0; i < M; ++i) {
        assert(scanf("%d%d%d", &a[i], &b[i], &correct_answers[i]) == 3);
        assert(0 <= a[i] && a[i] <= correct_answers[i] &&
              correct_answers[i] <= b[i] && b[i] < N);
    }
}

void save_to_floppy(const string &bits) {
    vector<int> contestant_answers = solve_queries(subtask_id, N, bits, a, b);
    for (int i = 0; i < M; ++i) {
        if (contestant_answers[i] != correct_answers[i]) {
            wrong_answer();
        }
    }
    correct_answer();
    exit(0);
}

int main(int argc, char **argv) {
    // Read input data.
    read_test();

    // Send subtask_id, v.
    read_array(subtask_id, v);
    
    answer_not_provided();
    return 0;
}
#endif

Compilation message (stderr)

stub.cpp: In function 'void run2()':
stub.cpp:101:30: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  101 |     if (query_answers.size() != M) {
      |         ~~~~~~~~~~~~~~~~~~~~~^~~~

• Subtask 1: 7 / 7
• Subtask 2: 21 / 21
• Subtask 3: 0 / 72