#include "train.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define ss second
#define ff first
#define sz(a) int(a.size())
#define all(a) a.begin(),a.end()
typedef long double ld;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<ll> vll;
const ll inf = 1e18;
const int mod = 1e9+7; //998244353;
const int maxn = 5e3+5;
const int Xg[4] = {1,0,-1,0}, Yg[4] = {0,1,0,-1};
ll modpw(ll a,ll e) {if(e==0)return 1;ll x=modpw(a*a%mod,e>>1);return e&1?x*a%mod:x;}
int N,M,vis[maxn];
int con[maxn],jog,jog1;
vi who_wins(vi a, vi r, vi u, vi v) {
N = sz(r),M = sz(u);
for(int i = 0;i < M;i++) con[u[i]].pb(v[i]);
vi ans;
for(int i = 0;i < N;i++){
memset(vis,0,sizeof(vis));
int now = i; vis[i]=1;
bool tr=0;
while(1){
if(sz(con[now]) > 1){tr=1;break;}
if(vis[con[now]] == 1){jog=con[now],jog1=now;break;}
vis[now]=1;
now=con[now];
}
for(int j = jog;j != jog1;j=con[j]) if(r[j]==1) tr=1;
if(r[jog1]==1) tr=1;
ans.pb(tr);
}
return ans;
}
Compilation message
train.cpp: In function 'vi who_wins(vi, vi, vi, vi)':
train.cpp:4:12: error: request for member 'push_back' in 'con[u.std::vector<int>::operator[](((std::vector<int>::size_type)i))]', which is of non-class type 'int'
4 | #define pb push_back
| ^~~~~~~~~
train.cpp:28:37: note: in expansion of macro 'pb'
28 | for(int i = 0;i < M;i++) con[u[i]].pb(v[i]);
| ^~
train.cpp:7:21: error: request for member 'size' in 'con[now]', which is of non-class type 'int'
7 | #define sz(a) int(a.size())
| ^~~~
train.cpp:37:7: note: in expansion of macro 'sz'
37 | if(sz(con[now]) > 1){tr=1;break;}
| ^~
