#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]]){
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
train.cpp: In function 'void bfs()':
train.cpp:19:8: warning: unused variable 'f' [-Wunused-variable]
19 | int f=0;
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
272 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
498 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
247 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
41 ms |
2652 KB |
3rd lines differ - on the 1st token, expected: '1', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
1860 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
272 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |