Submission #775926

#	Time	Username	Problem	Language	Result	Execution time	Memory
775926		ttamx	Toy Train (IOI17_train)	C++14	22 / 100	2063 ms	1564 KiB

train

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

using namespace std;

vector<int> who_wins(vector<int> a, vector<int> r, vector<int> U, vector<int> V) {
	int n=a.size();
    if(a==vector<int>(n,1)){
    vector<int> ans(n);
    vector<vector<int>> adj(n),radj(n);
    for(int i=0;i<U.size();i++){
        int u=U[i],v=V[i];
        adj[u].emplace_back(v);
        radj[v].emplace_back(u);
    }
    for(int i=0;i<n;i++){
        if(!r[i])continue;
        vector<bool> vis(n);
        function<void(int)> dfs=[&](int u){
            for(auto v:adj[u]){
                if(!vis[v]){
                    vis[v]=true;
                    dfs(v);
                }
            }
        };
        dfs(i);
        if(!vis[i])continue;
        vector<bool> rvis(n);
        function<void(int)> rdfs=[&](int u){
            if(rvis[u])return;
            rvis[u]=true;
            ans[u]=1;
            for(auto v:radj[u])rdfs(v);
        };
        rdfs(i);
    }
	return ans;
    }
    vector<int> ans(n,1);
    vector<vector<int>> adj(n),radj(n);
    for(int i=0;i<U.size();i++){
        int u=U[i],v=V[i];
        adj[u].emplace_back(v);
        radj[v].emplace_back(u);
    }
    for(int i=0;i<n;i++){
        if(r[i])continue;
        vector<bool> vis(n);
        function<void(int)> dfs=[&](int u){
            for(auto v:adj[u]){
                if(!vis[v]&&!r[v]){
                    vis[v]=true;
                    dfs(v);
                }
            }
        };
        dfs(i);
        if(!vis[i])continue;
        vector<bool> rvis(n);
        function<void(int)> rdfs=[&](int u){
            if(rvis[u])return;
            rvis[u]=true;
            ans[u]=0;
            for(auto v:radj[u])rdfs(v);
        };
        rdfs(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:11:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 |     for(int i=0;i<U.size();i++){
      |                 ~^~~~~~~~~
train.cpp:42:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   42 |     for(int i=0;i<U.size();i++){
      |                 ~^~~~~~~~~

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