# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
660964 | 2022-11-23T18:17:01 Z | urosk | 장난감 기차 (IOI17_train) | C++14 | 0 ms | 0 KB |
#include "train.h" #define dbg(x) cerr<<#x<<": "<<x<<endl #define here cerr<<"===================================\n" #include <bits/stdc++.h> #define ll int #define sz(a) (ll)(a.size()) #define pb push_back #define popb pop_back #define all(a) a.begin(),a.end() #define llinf 1000000000LL #define fi first #define sc second #define pll pair<ll,ll> #define endl '\n' #define ceri(a,l,r) {cerr<<#a<<": ";for(ll i_ = l;i_<=r;i_++) cerr<<a[i_]<< " ";cerr<<endl;} using namespace std; #define maxn 5005 ll n,m,it; vector<ll> ans; vector<ll> g[maxn]; vector<ll> cur; bool c[maxn]; bool vis[maxn]; bool naso = 0; bool ok[maxn]; ll in[maxn],out[maxn]; void dfs(ll u,ll poc){ vis[u] = 1; for(ll s : g[u]){ if(c[s]) continue; if(s==poc){ naso = 1; return; } if(vis[s]) continue; dfs(s,poc); if(naso) return; } } void dfs2(ll u){ if(c[u]) cnt++; vis[u] = 1; for(ll s : g[u]){ if(s==poc&&cnt){ naso = 1; return; } if(vis[s]) continue; dfs(s,poc); if(naso) return; } if(c[u]) cnt--; } void dfs3(ll u){ vis[u] = 1; for(ll s : g[u]){ if(vis[s]) continue; dfs3(s); ok[u]|=ok[s]; } } vector<int> who_wins(vector<int> a,vector<int> r,vector<int> u,vector<int> v) { reverse(all(u)); u.pb(-1); reverse(all(u)); reverse(all(v)); v.pb(-1); reverse(all(v)); reverse(all(r)); r.pb(-1); reverse(all(r)); ll sum = 0; for(ll x : a) sum+=x; n = sz(a); m = sz(u)-1; for(ll i = 1;i<=n;i++) c[i] = r[i]; for(ll i = 1;i<=m;i++){ u[i]++; v[i]++; ll x = u[i]; ll y = v[i]; g[x].pb(y); } if(sum==0){ for(ll i = 1;i<=n;i++){ if(c[i]) continue; for(ll j = 1;j<=n;j++) vis[j] = 0; naso = 0; dfs(i,i); if(naso) ok[i] = 1; } for(ll i = 1;i<=n;i++){ for(ll j = 1;j<=n;j++) vis[j] = 0; dfs3(i); } for(ll i = 1;i<=n;i++) ans.pb(1^ok[i]); return ans; } if(sum==n){ for(ll i = 1;i<=n;i++){ for(ll j = 1;j<=n;j++) vis[j] = 0; naso = 0; dfs2(i,i); if(naso) ok[i] = 1; } for(ll i = 1;i<=n;i++) ans.pb(ok[i]); return ans; } return {}; } /* 4 6 0 0 0 0 0 1 0 0 0 1 0 2 0 3 1 2 1 3 2 3 */