Submission #772920

#	Time	Username	Problem	Language	Result	Execution time	Memory
772920		fve5	Comparing Plants (IOI20_plants)	C++17	0 / 100	4077 ms	18112 KiB

plants

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

#define MAXN 200'000

vector<int> adj[MAXN];

void init(int k, vector<int> r) {
    for (int i = 0; i < r.size(); i++) {
        if (r[i] == k - 1) {
            for (int j = 1; j < k; j++) {
                adj[(i + j) % r.size()].push_back(i);
            }
        }
        if (r[i] == 0) {
            for (int j = 1; j < k; j++) {
                adj[i].push_back((i + j) % r.size());
            }
        }
    }
}

int compare_plants(int x, int y) {
    {
        set<int> visited;
        stack<int> q;
        q.push(x);
        while (!q.empty()) {
            auto curr = q.top(); q.pop();
            if (visited.count(curr)) continue;
            if (curr == y) return 1;
            visited.insert(curr);

            for (auto n: adj[curr]) {
                q.push(n);
            }
        }
    }
    {
        set<int> visited;
        stack<int> q;
        q.push(y);
        while (!q.empty()) {
            auto curr = q.top(); q.pop();
            if (visited.count(curr)) continue;
            if (curr == x) return -1;
            visited.insert(curr);

            for (auto n: adj[curr]) {
                q.push(n);
            }
        }
    }
    return 0;
}

Compilation message (stderr)

plants.cpp: In function 'void init(int, std::vector<int>)':
plants.cpp:9:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |     for (int i = 0; i < r.size(); i++) {
      |                     ~~^~~~~~~~~~

• 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