Submission #1037597

#	Time	Username	Problem	Language	Result	Execution time	Memory
1037597		12345678	Toy Train (IOI17_train)	C++17	11 / 100	376 ms	1736 KiB

train

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

using namespace std;

const int nx=5e3+5;

int n, cy[nx], pa[nx], vs[nx], f;
vector<int> d[nx], rv[nx];

void dfs(int u, int rt)
{
    vs[u]=1;
    if (f) return;
    for (auto v:d[u])
    {
        if (!f&&v==rt)
        {
            int tmp=u;
            while (tmp!=rt) cy[tmp]=1, tmp=pa[tmp];
            cy[rt]=1;
            f=1;
            return;
        }
        if (vs[v]) continue;
        pa[v]=u;
        dfs(v, rt);
    }
}
std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
    n=a.size();
    for (int i=0; i<u.size(); i++) d[u[i]].push_back(v[i]), rv[v[i]].push_back(u[i]);
    for (int i=0; i<n; i++) 
    {
        if (r[i]) 
        {
            f=0;
            for (int j=0; j<n;j ++) vs[j]=0;
            dfs(i, i);    
        }
    }
    vector<int> res(n);
    queue<int> q;
    for (int i=0; i<n;i ++) vs[i]=0;
    for (int i=0; i<n;i ++) if (cy[i]) q.push(i), vs[i]=1;
    while (!q.empty())
    {
        auto cur=q.front();
        q.pop();
        res[cur]=1;
        for (auto x:rv[cur]) if (!vs[x]) vs[x]=1,q.push(x);
    }
    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:32:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |     for (int i=0; i<u.size(); i++) d[u[i]].push_back(v[i]), rv[v[i]].push_back(u[i]);
      |                   ~^~~~~~~~~

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