Submission #280361

#	Time	Username	Problem	Language	Result	Execution time	Memory
280361		Noam13	Toy Train (IOI17_train)	C++14	5 / 100	12 ms	1280 KiB

train

#include <bits/stdc++.h>
#define vi vector<int>
#define vvi vector<vi>
#define vb vector<bool>
#define vvb vector<vb>
#define ii pair<int, int>
#define x first
#define y second
#define vii vector<ii>
#define pb push_back
#define all(x) x.begin(), x.end()
#define loop(i,s,e) for(int i=s;i<e;++i)
#define loopr(i,s,e) for(int i=e-1;i>=s;--i)
#define chkmin(a,b) a = min(a,b)
#define chkmax(a,b) a = max(a,b)
using namespace std;
const int INF = 1e9;


int n;
vi who_wins(vi a, vi r, vi u, vi v) {
    n = a.size();
    vvi g(n);
    loop(i,0,u.size()){
        int a = u[i], b = v[i];
        g[a].pb(b);
    }
	vi res(n); res[n-1] = r[n-1];
    loopr(i,0,n-1){
        int tmp = -1;
        res[i] = r[i];
        for(auto nei:g[i]){
            if (nei==i){
                if ((a[i] ^ r[i]) == 0) tmp = r[i];
            }
            else res[i] = res[nei];
        }
        if (tmp!=-1) res[i] = tmp;
    }
	return res;
}

/*
clear
g++ c.cpp grader.cpp -o c ; ./c
2 4
1 0
0 1
0 0
0 1
1 0
1 1
*/

Compilation message (stderr)

train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:12:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 | #define loop(i,s,e) for(int i=s;i<e;++i)
......
   24 |     loop(i,0,u.size()){
      |          ~~~~~~~~~~~~             
train.cpp:24:5: note: in expansion of macro 'loop'
   24 |     loop(i,0,u.size()){
      |     ^~~~

• Subtask 1: 5 / 5
• Subtask 2: 0 / 10
• Subtask 3: 0 / 11
• Subtask 4: 0 / 11
• Subtask 5: 0 / 12
• Subtask 6: 0 / 51