Submission #660964

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
660964	2022-11-23T18:17:01 Z	urosk	Toy Train (IOI17_train)	C++14	Compilation error	0 ms	0 KB

train

#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
*/

Compilation message

train.cpp: In function 'void dfs2(int)':
train.cpp:41:14: error: 'cnt' was not declared in this scope; did you mean 'int'?
   41 |     if(c[u]) cnt++;
      |              ^~~
      |              int
train.cpp:44:15: error: 'poc' was not declared in this scope; did you mean 'pow'?
   44 |         if(s==poc&&cnt){
      |               ^~~
      |               pow
train.cpp:44:20: error: 'cnt' was not declared in this scope; did you mean 'int'?
   44 |         if(s==poc&&cnt){
      |                    ^~~
      |                    int
train.cpp:49:15: error: 'poc' was not declared in this scope; did you mean 'pow'?
   49 |         dfs(s,poc);
      |               ^~~
      |               pow
train.cpp:52:14: error: 'cnt' was not declared in this scope; did you mean 'int'?
   52 |     if(c[u]) cnt--;
      |              ^~~
      |              int
train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:94:21: error: too many arguments to function 'void dfs2(int)'
   94 |             dfs2(i,i);
      |                     ^
train.cpp:40:6: note: declared here
   40 | void dfs2(ll u){
      |      ^~~~