#include<bits/stdc++.h>
#define MAXN 1000007
using namespace std;
int n;
bool is[MAXN],sp[MAXN];
vector<int> v[MAXN],ans;
int tim[MAXN];
bool brute(int x,int last,int t){
tim[x]=t;
if(sp[x])last=t;
for(int i=0;i<v[x].size();i++){
if(is[x]){
if(tim[v[x][i]]!=0 and tim[v[x][i]]<=last)return true;
}else{
if(tim[v[x][i]]!=0 and tim[v[x][i]]>last)return false;
}
}
for(int i=0;i<v[x].size();i++){
if(tim[v[x][i]]!=0)continue;
if(is[x]){
if(brute(v[x][i],last,t+1))return true;
}else{
if(!brute(v[x][i],last,t+1))return false;
}
}
tim[x]=0;
return !is[x];
}
vector<int> who_wins(vector<int> A,vector<int> R, vector<int> from,vector<int> to){
n=int(A.size());
for(int i=1;i<=n;i++){
is[i]=A[i-1];
sp[i]=R[i-1];
}
for(int i=1;i<=from.size();i++){
v[from[i-1]+1].push_back(to[i-1]+1);
}
if(n<=15){
for(int i=1;i<=n;i++){
for(int f=1;f<=n;f++)tim[f]=0;
if(brute(i,0,1))ans.push_back(1);
else ans.push_back(0);
}
}
return ans;
}
/*int main(){
ans=who_wins({0, 1}, {1, 0}, {0, 0, 1, 1}, {0, 1, 0, 1});
for(int i:ans)cout<<i<<" ";
return 0;
}*/
Compilation message
train.cpp: In function 'bool brute(int, int, int)':
train.cpp:14:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
14 | for(int i=0;i<v[x].size();i++){
| ~^~~~~~~~~~~~
train.cpp:22:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
22 | for(int i=0;i<v[x].size();i++){
| ~^~~~~~~~~~~~
train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:44:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
44 | for(int i=1;i<=from.size();i++){
| ~^~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
27216 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
5 ms |
26960 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 |
Incorrect |
9 ms |
27728 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
8 ms |
27472 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
10 ms |
27728 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
27216 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |