#include "train.h"
#include<bits/stdc++.h>
using namespace std;
const int maxn=5000+10;
vector<int>adj[maxn];
int n,shar[maxn],arez[maxn],m,dp[maxn],vas[maxn],dide[maxn];
int dfs(int u){
if(shar[u]){
return 0;
}
if(dide[u]){
return 1;
}
dide[u]=1;
int ret=0;
for(auto x:adj[u]){
ret|=dfs(x);
}
return ret;
}
bool ber(int u){
for(int i=0;i<n;i++){
vas[i]=dide[i]=0;
}
return 1^dfs(u);
}
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();
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++){
adj[u[i]].push_back(v[i]);
}
for(int i=0;i<n;i++){
dp[i]=ber(i);
}
for(int asd=0;asd<=n;asd++){
for(int i=0;i<n;i++){
int cnt=0;
int f=0;
for(auto x:adj[i]){
if(x==i){
f=1;
continue;
}
if(dp[x]==1){
cnt++;
}
}
if(arez[i]){
if(f){
if(shar[i]==0&&cnt==0){
dp[i]=0;
}
}else{
if(cnt==0){
dp[i]=0;
}
}
}else{
if(f){
if(shar[i]==0||cnt!=(int)adj[i].size()-1){
dp[i]=0;
}
}else{
if(cnt!=(int)adj[i].size()){
dp[i]=0;
}
}
}
}
}
vector<int>res(n);
for(int i=0;i<n;i++){
res[i]=dp[i];
}
return res;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
122 ms |
856 KB |
3rd lines differ - on the 26th token, expected: '1', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
348 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 |
247 ms |
1420 KB |
Output is correct |
2 |
Correct |
233 ms |
1364 KB |
Output is correct |
3 |
Correct |
204 ms |
1112 KB |
Output is correct |
4 |
Incorrect |
620 ms |
1364 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 |
198 ms |
1112 KB |
Output is correct |
2 |
Incorrect |
314 ms |
1160 KB |
3rd lines differ - on the 1st token, expected: '1', found: '0' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1055 ms |
1364 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 |
122 ms |
856 KB |
3rd lines differ - on the 26th token, expected: '1', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |