Submission #1039203

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
1039203	2024-07-30T14:29:52 Z	Soumya1	Toy Train (IOI17_train)	C++17	0 / 100	10 ms	3420 KB

train

#include "train.h"
#include <bits/stdc++.h>
using namespace std;
const int mxN = 5005;
set<int> ad[mxN], rev[mxN];
vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v) {
	int n = a.size();
	for (int i = 0; i < u.size(); i++) {
		ad[u[i]].insert(v[i]);
		rev[v[i]].insert(u[i]);
	}
	auto get = [&](vector<int> s) {
		vector<int> out(n), vis(n);
		for (int i : s) vis[i] = true;
		for (int i = 0; i < s.size(); i++) {
			int u = s[i];
			for (int v : rev[u]) {
				out[v]++;
				if (!vis[v] && out[v] >= (a[v] ? 1 : ad[v].size())) {
					s.push_back(v);
					vis[v] = true;
				}
			}
		}
		return s;
	};
	vector<bool> removed(n);
	while (true) {
		vector<int> on;
		for (int i = 0; i < n; i++) {
			if (r[i] && !removed[i]) on.push_back(i);
		}
		vector<bool> can(n);
		on = get(on);
		for (int i : on) can[i] = true;
		vector<int> bad;
		for (int i = 0; i < n; i++) {
			if (!removed[i] && !can[i]) bad.push_back(i);
		}
		bad = get(bad);
		if (bad.empty()) break;
		for (int i : bad) {
			removed[i] = true;
			ad[i].clear();
			for (int j : rev[i]) ad[j].erase(i);
			rev[i].clear();
		}
	}
	vector<int> ans(n);
	for (int i = 0; i < n; i++) ans[i] = !removed[i];
	return ans;
}

Compilation message

train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:8:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    8 |  for (int i = 0; i < u.size(); i++) {
      |                  ~~^~~~~~~~~~
train.cpp: In lambda function:
train.cpp:15:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |   for (int i = 0; i < s.size(); i++) {
      |                   ~~^~~~~~~~~~
train.cpp:19:27: warning: comparison of integer expressions of different signedness: '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'} and 'long unsigned int' [-Wsign-compare]
   19 |     if (!vis[v] && out[v] >= (a[v] ? 1 : ad[v].size())) {

Subtask #1
0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	4 ms	1884 KB	3rd lines differ - on the 14th token, expected: '1', found: '0'
2	Halted	0 ms	0 KB	-

Subtask #2
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1 ms	856 KB	3rd lines differ - on the 8th token, expected: '0', found: '1'
2	Halted	0 ms	0 KB	-

Subtask #3
0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	9 ms	3420 KB	Output is correct
2	Correct	8 ms	3308 KB	Output is correct
3	Correct	8 ms	3420 KB	Output is correct
4	Incorrect	10 ms	3420 KB	3rd lines differ - on the 1st token, expected: '1', found: '0'
5	Halted	0 ms	0 KB	-