Submission #618007

#	Time	Username	Problem	Language	Result	Execution time	Memory
618007		chirathnirodha	장난감 기차 (IOI17_train)	C++17	0 / 100	1084 ms	1316 KiB

train

#include "train.h"
#include<bits/stdc++.h>
using namespace std;
#define PB push_back
#define MP make_pair
#define F first
#define S second
#define P push

int n,m;
vector<int> ed[5000];
bool own[5000];
bool charge[5000];
bool self[5000];
vector<int> are,bor;

vector<int> SUB_1() {
	vector<int> ans(n);
	for(int i=0;i<n;i++){
		for(int j=i;j<n;j++){
			if(self[j] && ed[j].size()==1){
				if(charge[j])ans[i]=1;
				else ans[i]=0;
				break;
			}
			if(own[j] && charge[j] && self[j]){ans[i]=1;break;}
			if(!own[j] && !charge[j] && self[j]){ans[i]=0;break;}
		}
	}
	return ans;
}
bool visited[5000];
bool ok;
void dfs(int x,int ch){
	if(charge[x])ch++;
	if(visited[x]){
		if(ch>0)ok=true;
		return;
	}
	visited[x]=true;
	for(int i=0;i<ed[x].size();i++){
		int y=ed[x][i];
		dfs(y,ch);
	}
	if(charge[x])ch--;
    return;
}

vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v) {
	n=a.size();m=u.size();
	memset(self,false,sizeof(self));
	for(int i=0;i<m;i++){
		ed[u[i]].PB(v[i]);
		if(u[i]==v[i])self[u[i]]=true;
	}
	for(int i=0;i<n;i++){
		own[i]=a[i];
		charge[i]=r[i];
		if(a[i])are.PB(i);
		else bor.PB(i);
	}
	vector<int> ans(n);
	for(int i=0;i<n;i++){
		memset(visited,false,sizeof(visited));
		ok=false;
		dfs(i,0);
		ans[i]=ok;
	}
	return ans;
	return SUB_1();
}

Compilation message (stderr)

train.cpp: In function 'void dfs(int, int)':
train.cpp:41:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |  for(int i=0;i<ed[x].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