Submission #287179

#	Time	Username	Problem	Language	Result	Execution time	Memory
287179		peti1234	Toy Train (IOI17_train)	C++17	11 / 100	12 ms	1664 KiB

train

#include <bits/stdc++.h>

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

Compilation message (stderr)

train.cpp: In function 'void solve()':
train.cpp:15:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |             for (int i=0; i<inv[id].size(); i++) {
      |                           ~^~~~~~~~~~~~~~~

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