Submission #711551

#	Time	Username	Problem	Language	Result	Execution time	Memory
711551		penguinman	장난감 기차 (IOI17_train)	C++17	11 / 100	543 ms	1636 KiB

train

#include "train.h"
#include <bits/stdc++.h>

using std::cin;
using std::cout;
using std::vector;
using ll = int;
using vi = vector<ll>;
using vii = vector<vi>;
using std::string;
using pii = std::pair<ll,ll>;

#define rep(i,j,k) for(ll i=ll(j); i<ll(k); i++)
#define REP(i,j,k) for(ll i=ll(j); i<=ll(k); i++)
#define per(i,j,k) for(ll i=ll(j); i>=ll(k); i--)
#define ln "\n"
#define all(a) a.begin(), a.end()
#define mp std::make_pair
#define pb emplace_back

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
	ll N = a.size();
	vii edge(N);
	rep(i,0,u.size()) edge[u[i]].pb(v[i]);
	vii edge_mem = edge;
	vi dp = r;
	REP(t,0,N+10){
		rep(i,0,N){
			if(a[i] == 1){
				for(auto next: edge[i]){
					dp[i] |= dp[next];
				}
			}
			else{
				int idx = 0;
				while(idx < edge[i].size()){
					int next = edge[i][idx];
					if(dp[next]){
						int length = edge[i].size();
						std::swap(edge[i][idx], edge[i][length-1]);
						edge[i].pop_back();
					}
					else idx++;
				}
				if(edge[i].empty()) dp[i] = 1;
			}
		}
	}
	edge = edge_mem;
	vi dp2 = dp;
	rep(i,0,N) dp2[i] = 1-dp2[i];
	REP(t,0,N+10){
		rep(i,0,N){
			if(a[i] == 0){
				for(auto next: edge[i]){
					dp2[i] |= dp2[next];
				}
			}
			else{
				int idx = 0;
				while(idx < edge[i].size()){
					int next = edge[i][idx];
					if(dp2[next]){
						int length = edge[i].size();
						std::swap(edge[i][idx], edge[i][length-1]);
						edge[i].pop_back();
					}
					else idx++;
				}
				if(edge[i].empty()) dp2[i] = 1;
			}
		}
	}
	rep(i,0,N) dp2[i] = 1-dp2[i];
	return dp2;
}

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:36:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     while(idx < edge[i].size()){
      |           ~~~~^~~~~~~~~~~~~~~~
train.cpp:61:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |     while(idx < edge[i].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