Submission #1025866

#	Time	Username	Problem	Language	Result	Execution time	Memory
1025866		amirhoseinfar1385	Toy Train (IOI17_train)	C++17	26 / 100	2100 ms	2816 KiB

train

#include "train.h"
#include<bits/stdc++.h>
using namespace std;
const int maxn=30000+10;
vector<int>adj[maxn],revadj[maxn];
int n,shar[maxn],arez[maxn],m,dp[maxn],vas[maxn],dide[maxn];

void bfs(){
	vector<int>bf;
	for(int i=0;i<n;i++){
		vas[i]=0;
		if(dp[i]==1){
			bf.push_back(i);
		}
	}
	for(int u=0;u<(int)bf.size();u++){
		for(auto i:revadj[bf[u]]){
			if(dp[i]){
				continue;
			}
			int cnt=0;
			int f=0;
			for(auto x:adj[i]){
				if(dp[x]==1){
					cnt++;
				}
			}
			if(arez[i]){
				if(cnt>0){
					dp[i]=1;
					bf.push_back(i);
				}else{
					dp[i]=0;
				}
			}else{
				if(cnt!=(int)adj[i].size()){
					dp[i]=0;
				}else{
					dp[i]=1;
					bf.push_back(i);
				}
			}
		}
	}
}

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
	n=(int)a.size();
	m=(int)u.size();
	for(int i=0;i<=n;i++){
		adj[i].clear();
		revadj[i].clear();
		shar[i]=vas[i]=dide[i]=dp[i]=arez[i]=0;
	}
	for(int i=0;i<n;i++){
		if(a[i]){
			arez[i]=1;
		}
		if(r[i]){
			shar[i]=1;
		}
	}
	for(int i=0;i<m;i++){
		revadj[v[i]].push_back(u[i]);
		adj[u[i]].push_back(v[i]);
	}
	for(int nago=0;nago<=n;nago++){
		for(int i=0;i<n;i++){
			dp[i]=shar[i];
		}
		bfs();
	/*for(int asd=0;asd<=n;asd++){
		for(int i=0;i<n;i++){
			if(dp[i]){
				continue;
			}
			int cnt=0;
			int f=0;
			for(auto x:adj[i]){
				if(dp[x]==1){
					cnt++;
				}
			}
			if(arez[i]){
				if(cnt>0){
					dp[i]=1;
				}else{
					dp[i]=0;
				}
			}else{
				if(cnt!=(int)adj[i].size()){
					dp[i]=0;
				}else{
					dp[i]=1;
				}
			}
		}
	}*/
		if(nago!=n){
			for(int i=0;i<n;i++){
				if(shar[i]==1){
					int cnt=0;
					for(auto x:adj[i]){
						if(dp[x]==1){
							cnt++;
						}
					}
					if(arez[i]){
						shar[i]=(cnt>=1);
					}else{
						shar[i]=(cnt==(int)adj[i].size());
					}
				}
			}
		}
	}
	vector<int>res(n);
	for(int i=0;i<n;i++){
		res[i]=dp[i];
	}
	return res;
}

Compilation message (stderr)

train.cpp: In function 'void bfs()':
train.cpp:22:8: warning: unused variable 'f' [-Wunused-variable]
   22 |    int f=0;
      |        ^

• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 11 / 11
• Subtask 4: 0 / 11
• Subtask 5: 0 / 12
• Subtask 6: 0 / 51