Submission #388212

#	Time	Username	Problem	Language	Result	Execution time	Memory
388212		alexxela12345	Comparing Plants (IOI20_plants)	C++17	0 / 100	4053 ms	7244 KiB

plants

#include <bits/stdc++.h>
#include "plants.h"

using namespace std;

int n;
int k;
vector<int> r;

vector<int> prefr;

void genSome();

void init(int k_, std::vector<int> r_) {
	k = k_;
	r = r_;
	n = r.size();
    prefr.resize(n + 1);
    for (int i = 1; i <= n; i++) {
        prefr[i] = prefr[i - 1] + r[i - 1];
    }
    genSome();
}

int get_sum(int l, int r) {
    if (l <= r) {
        return prefr[r] - prefr[l];
    }
    return prefr[n] - prefr[l] + prefr[r];
}

vector<int> h;

void genSome() {
    h.resize(n, -1);
    vector<int> r2 = r;
    for (int asdf = 0; asdf < n; asdf++) {
        for (int i = 0; i < n; i++) {
            if (r2[i] == 0 && h[i] == -1) {
                bool bad = 0;
                for (int j = 1; j < k; j++) {
                    int k = (i - j + n) % n;
                    if (r2[k] == 0) {
                        bad = 1;
                        break;
                    }
                }
                if (bad) {
                    continue; 
                }
                h[i] = n - asdf - 1;
                for (int j = 1; j < k; j++) {
                    int k = (i - j + n) % n;
                    r2[k]--;
                }
                break;
            }
        } 
    }
}

int compare_plants(int x, int y) {
    if (k == 2) {
        int a = get_sum(x, y);
        int b = get_sum(y, x); 
        int A = y - x;
        if (A < 0)
            A += n;
        int B = n - A;
        if (a == 0) {
            return 1;
        } else if (a == A) {
            return -1;
        }
        if (b == 0) {
            return -1;
        } else if (b == B) {
            return 1;
        }
        return 0;
    }
    if (2 * k > n) {
        if (h[x] > h[y]) {
            return 1;
        } else {
            return -1;
        }
    }
    assert(false);
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 14
• Subtask 3: 0 / 13
• Subtask 4: 0 / 17
• Subtask 5: 0 / 11
• Subtask 6: 0 / 15
• Subtask 7: 0 / 25