Submission #295773

#	Time	Username	Problem	Language	Result	Execution time	Memory
295773		DanerZein	Toy Train (IOI17_train)	C++14	0 / 100	10 ms	1536 KiB

train

#include "train.h"
#include <bits/stdc++.h>
using namespace std;
typedef vector<int> vi;
vector<vi> G;
bitset<5010> ab,cs;
int vis[5010];
vi res;
bool dfs(int u){
  //cout<<u<<endl;
  vis[u]=2;
  if(ab[u]==1){
    int x=-1;
    for(auto &v:G[u]){
      if(v==u and cs[v]==1){
	res[u]=1;
	return 1;
      }
      else{
	if(v!=u) x=v;
      }
    }
    if(x!=-1 and vis[x]==-1) return res[u]=dfs(x);
    return 0;
  }
  else{
    int x=-1;
    for(auto &v:G[u]){
      if(v==u and cs[v]==0){
	res[u]=0;
	return 0;
      }
      else{
	if(v!=u) x=v;
      }
    }
    if(x!=-1 and vis[x]==-1) return res[u]=dfs(x);
    return 1;
  }
}
std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
  for(int i=0;i<a.size();i++){
    ab[i]=a[i];
    cs[i]=r[i];
  }
  int n=a.size();
  G.resize(n+1);
  res.resize(n);
  for(int i=0;i<u.size();i++){
    int x=u[i],y=v[i];
    G[x].push_back(y);
    G[y].push_back(x);
  }
  memset(vis,-1,sizeof vis);
  for(int i=0;i<a.size();i++){
    if(vis[i]==-1){
      dfs(i);
    }
  }
  return res;
}

Compilation message (stderr)

train.cpp: In function 'bool dfs(int)':
train.cpp:23:43: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
   23 |     if(x!=-1 and vis[x]==-1) return res[u]=dfs(x);
train.cpp:37:43: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
   37 |     if(x!=-1 and vis[x]==-1) return res[u]=dfs(x);
train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:42:16: 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<a.size();i++){
      |               ~^~~~~~~~~
train.cpp:49:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |   for(int i=0;i<u.size();i++){
      |               ~^~~~~~~~~
train.cpp:55:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |   for(int i=0;i<a.size();i++){
      |               ~^~~~~~~~~

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