제출 #822937

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
822937		mindiyak	장난감 기차 (IOI17_train)	C++14	0 / 100	6 ms	1236 KiB

train

#include "train.h"
using namespace std;

vector<vector<int>> paths(1e4,vector<int> ());
vector<int> chargers(1e4);
vector<int> cost(1e4,-1);

int loop = 0;
int n;

void dfs(int pos,int start,int coins,int charging,vector<int> visited,int time){
	if(cost[pos] <= coins)loop=1;
	if(visited[pos]){
		if(chargers[pos] == 1){
			cost[pos] = min(cost[pos],time);
			loop = 1;
		}
		return;
	}
	if(loop==1)return;
	if(chargers[pos] == 1){
		if(time == -1){
			time = n-coins;
		}
		charging = 1;
		coins = n;
	}
	if(pos==start and charging==1){cost[pos] = min(cost[pos],time);loop = 1;}
	coins--;
	visited[pos]=1;
	if(coins <= 0)return;
	for(int a:paths[pos]){
		dfs(a,start,coins,charging,visited,time);
	}
}

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
	n=a.size();
	chargers = r;
	for (int i = 0; i < u.size(); i++){
		paths[u[i]].push_back(v[i]);
	}
	
	vector<int> ans(n,0);

	
	for (int i = 0; i < n; i++){
		loop = 0;
		dfs(i,i,n,chargers[i],vector<int>(n,0),-1);
		ans[i] = loop;
	}

	return ans;
}

컴파일 시 표준 에러 (stderr) 메시지

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