Submission #1033131

#	Time	Username	Problem	Language	Result	Execution time	Memory
1033131		Zicrus	Comparing Plants (IOI20_plants)	C++17	0 / 100	4059 ms	6648 KiB

plants

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

typedef long long ll;

vector<ll> lnk;
vector<int> a;

ll find(ll a) {
    if (lnk[a] != a) lnk[a] = find(lnk[a]);
    return lnk[a];
}

void unite(ll a, ll b) {
    a = find(a); b = find(b);
    lnk[b] = a;
}

bool same(ll a, ll b) {
    return find(a) == find(b);
}

void init(int k, vector<int> r) {
    ll n = r.size();
    lnk = vector<ll>(n);
    for (int i = 0; i < n; i++) lnk[i] = i;
    for (int i = n-1; i >= 0; i--) {
        int o = (i+1) % n;
        if (r[i] == r[o]) unite(i, o);
    }
    a = vector<int>(n);
    for (int i = 0; i < n; i++) a[i] = r[i];
}

int compare_plants(int x, int y) {
    ll n = a.size();

    ll dir = a[x];
    for (int i = x; i != y; i++) {
        if (dir != a[i]) break;
        if (i == y-1) {
            //return dir == 1 ? -1 : 1;
            break;
        }
    }
    dir = a[y];
    for (int i = y; i != x; i = (i+1)%n) {
        if (dir != a[i]) break;
        if (i == (x-1+n)%n) {
            return dir == 0 ? -1 : 1;
        }
    }

    if (same(x, (y-1+n)%n)) {
         return (a[x] ? -1 : 1);
    }
    if (same((x-1+n)%n, y)) {
        return (!a[y] ? -1 : 1);
    }
    return 0;
}

• 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