Submission #617742

#	Time	Username	Problem	Language	Result	Execution time	Memory
617742		Sergio_2357	Toy Train (IOI17_train)	C++17	15 / 100	2095 ms	1416 KiB

train

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

using namespace std;

#define int long long

typedef vector<int> vi;
typedef vector<vi> vii;

bool bk(int i, int d, int lp, bool obj, vi& set, vector<signed>& a, vector<signed>& r, vii& g)
{
    if (set[i] != -1) {
        if (set[i] <= lp)
            return obj;
        else
            return !obj;
    }
    set[i] = d;
    bool can = false;
    for (int ot : g[i]) {
        if (a[i] == a[ot]) {
            if (bk(ot, d + 1, (r[i] ? d : lp), a[ot], set, a, r, g))
                can = true;
        } else if (!bk(ot, d + 1, (r[i] ? d : lp), a[ot], set, a, r, g))
            can = true;
    }
    set[i] = -1;
    return can;
}

vector<signed> who_wins(vector<signed> a, vector<signed> r, vector<signed> u, vector<signed> v)
{
    vector<signed> res(a.size(), 0);
    vii g(a.size());
    for (int i = 0; i < u.size(); i++)
        g[u[i]].push_back(v[i]);
    for (int i = 0; i < a.size(); i++) {
        vi set(a.size(), -1);
        res[i] = ((bool)a[i]) == bk(i, 0, -1, a[i], set, a, r, g);
    }
    return res;
}

Compilation message (stderr)

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

• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 0 / 11
• Subtask 4: 0 / 11
• Subtask 5: 0 / 12
• Subtask 6: 0 / 51