Submission #794442

# Submission time Handle Problem Language Result Execution time Memory
794442 2023-07-26T14:25:53 Z firewater Toy Train (IOI17_train) C++14
11 / 100
8 ms 2084 KB
#include "train.h"
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include <cassert>
#define ll long long
#define MX 20230
using namespace std;
int n,m,tot,w,p[MX],a[MX],c[MX],cc[MX],h[MX],H[MX],fr[MX],cy[MX],pp[MX];
vector<int>ls,s[MX];
struct rec
{
	int to,nx;
}e[MX],E[MX];
void addl(int x,int y)
{
	e[++tot].to=y;
	e[tot].nx=h[x];
	h[x]=tot;
	return;
}
void addll(int x,int y)
{
	E[tot].to=y;
	E[tot].nx=H[x];
	H[x]=tot;
	return;
}
void dfs1(int x)
{
	if(c[x])return;
	p[x]=1;
	for(int i=h[x];i;i=e[i].nx){
		int y=e[i].to;
		if(p[y])continue;
		dfs1(y);
	}
	ls.push_back(x);
	return;
}
void dfs2(int x,int p)
{
	if(c[x])return;
	fr[x]=p;
	s[p].push_back(x);
	for(int i=H[x];i;i=E[i].nx){
		int y=E[i].to;
		if(fr[y])continue;
		dfs2(y,p);
	}
	return;
}
void dfs3(int x)
{
	pp[x]=1;
	for(int i=H[x];i;i=E[i].nx){
		int y=E[i].to;
		if(pp[y])continue;
		dfs3(y);
	}
	return;
}
std::vector<int> who_wins(std::vector<int> A, std::vector<int> C, std::vector<int> U, std::vector<int> V) {
	std::vector<int> ans(A.size());
	n=A.size();
	m=U.size();
	for(int i=1;i<=n;++i){
		a[i]=A[i-1];
		c[i]=C[i-1];
	}
	for(int i=0;i<m;++i){
		if(U[i]==V[i])cy[U[i]+1]=1;
		else{
			addl(U[i]+1,V[i]+1);
			addll(V[i]+1,U[i]+1);
		}
	}
	for(int i=1;i<=n;++i)
		p[i]=0;
	for(int i=1;i<=n;++i)
		if(!p[i])
			dfs1(i);
	reverse(ls.begin(),ls.end());
	for(int i=0;i<ls.size();++i)
		if(!fr[ls[i]])
			dfs2(ls[i],++w);
	for(int i=1;i<=n;++i)
		if(c[i]==0&&(s[fr[i]].size()>1||cy[i]==1))
			pp[i]=1;
	for(int i=1;i<=n;++i)
		if(pp[i])
			dfs3(i);
	for(int i=1;i<=n;++i)
		ans[i-1]=pp[i]^1;
	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:85:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   85 |  for(int i=0;i<ls.size();++i)
      |              ~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 1364 KB 3rd lines differ - on the 14th token, expected: '1', found: '0'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 792 KB 3rd lines differ - on the 2nd token, expected: '1', found: '0'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 5 ms 2004 KB Output is correct
2 Correct 5 ms 1928 KB Output is correct
3 Correct 6 ms 1900 KB Output is correct
4 Incorrect 6 ms 1948 KB 3rd lines differ - on the 1st token, expected: '1', found: '0'
5 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 4 ms 1588 KB Output is correct
2 Correct 5 ms 1876 KB Output is correct
3 Correct 6 ms 2004 KB Output is correct
4 Correct 5 ms 1876 KB Output is correct
5 Correct 7 ms 2000 KB Output is correct
6 Correct 5 ms 1976 KB Output is correct
7 Correct 6 ms 2004 KB Output is correct
8 Correct 6 ms 2004 KB Output is correct
9 Correct 6 ms 1620 KB Output is correct
10 Correct 5 ms 1876 KB Output is correct
11 Correct 7 ms 1828 KB Output is correct
12 Correct 5 ms 1828 KB Output is correct
13 Correct 5 ms 2084 KB Output is correct
14 Correct 5 ms 1924 KB Output is correct
# Verdict Execution time Memory Grader output
1 Incorrect 8 ms 2052 KB 3rd lines differ - on the 1st token, expected: '1', found: '0'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 1364 KB 3rd lines differ - on the 14th token, expected: '1', found: '0'
2 Halted 0 ms 0 KB -