Submission #596120

#	Time	Username	Problem	Language	Result	Execution time	Memory
596120		kshitij_sodani	Toy Train (IOI17_train)	C++14	11 / 100	75 ms	1268 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;
	vector<int> ss;
	for(int i=0;i<n;i++){
		if(bb[i]==1){
			ind=i;
			ss.pb(ind);
		}
	}
	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);
	}
	while(true){
		int st=1;
		for(int j=0;j<n;j++){
			if(bb[j]==1){
				int su=0;
				for(auto i:adj[j]){
					su+=bb[i];
				}
				if(aa[j]==1){
					if(su==0){
						bb[j]=0;
						st=0;
						break;
					}
				}
				else{
					if(su<adj[j].size()){
						bb[j]=0;
						st=0;
						break;
					}
				}
			}
		}
		if(st==1){
			break;
		}
	}
	return bb;
	/*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:41:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |      if(su==adj[i].size()){
      |         ~~^~~~~~~~~~~~~~~
train.cpp:73:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   73 |      if(su<adj[j].size()){
      |         ~~^~~~~~~~~~~~~~

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