Submission #287149

#	Time	Username	Problem	Language	Result	Execution time	Memory
287149		peti1234	Toy Train (IOI17_train)	C++17	12 / 100	436 ms	1792 KiB

train

#include <bits/stdc++.h>

using namespace std;
const int c=5002;
int n, m, v[c], db[c], si[c];
bool sol[c];
vector<int> sz[c], inv[c], ans;
queue<int> q;
vector<int> who_wins(vector<int> ki, vector<int> t, vector<int> x, vector<int> y) {
    n=ki.size(), m=x.size();
    for (int i=0; i<m; i++) {
        int a=x[i], b=y[i];
        sz[a].push_back(b), inv[b].push_back(a);
    }
    for (int i=0; i<n; i++) si[i]=sz[i].size();
    for (int i=0; i<n; i++) {
        if (t[i]) {
            v[i]=-1;
            q.push(i);
            while(q.size()>0) {
                int id=q.front();
                q.pop();
                if (v[id]!=1) {
                    v[id]++;
                    for (int i=0; i<inv[id].size(); i++) {
                        int x=inv[id][i];
                        db[x]++;
                        if (!v[x] && (db[x]==si[x] || (db[x]==1 && ki[x]))) q.push(x);
                    }
                }
            }
            if (v[i]) for (int j=0; j<n; j++) sol[j]^=v[j];
            for (int j=0; j<n; j++) v[j]=0, db[j]=0;
        }
    }
    for (int i=0; i<n; i++) ans.push_back(sol[i]);
    return ans;
}

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:25:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   25 |                     for (int i=0; i<inv[id].size(); i++) {
      |                                   ~^~~~~~~~~~~~~~~

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