Submission #596114

#	Time	Username	Problem	Language	Result	Execution time	Memory
596114		kshitij_sodani	Toy Train (IOI17_train)	C++14	12 / 100	65 ms	1240 KiB

train

#include <bits/stdc++.h>
using namespace std;
typedef long long llo;
#define a first
#define b second
#define pb push_back
#define endl '\n'


#include "train.h"
vector<int> adj[5001];
std::vector<int> who_wins(std::vector<int> aa, std::vector<int> bb, std::vector<int> u, std::vector<int> v) {
	int n=aa.size();
	for(int i=0;i<u.size();i++){
		adj[u[i]].pb(v[i]);
	}
	int ind=-1;
	for(int i=0;i<n;i++){
		if(bb[i]==1){
			ind=i;
		}
	}
	while(true){
		int st=1;
		for(int i=0;i<n;i++){
			if(bb[i]==0){
				int su=0;
				for(auto j:adj[i]){
					su+=bb[j];
				}
				if(aa[i]==1){
					if(su>0){
						bb[i]=1;
						st=0;
						break;
					}
				}
				else{
					if(su==adj[i].size()){
						bb[i]=1;
						st=0;
						break;
					}
				}
			}
		}
		if(st==1){
			break;
		}
	}
	vector<int> ans;
	for(int i=0;i<n;i++){
		ans.pb(0);
	}
	for(auto j:adj[ind]){
		if(bb[j]==1){
			for(int i=0;i<n;i++){
				if(bb[i]==1){
					ans[i]=1;
				}
			}
			break;
		}
	}
	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:14:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   14 |  for(int i=0;i<u.size();i++){
      |              ~^~~~~~~~~
train.cpp:39:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |      if(su==adj[i].size()){
      |         ~~^~~~~~~~~~~~~~~

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