Submission #1021426

#	Time	Username	Problem	Language	Result	Execution time	Memory
1021426		MarwenElarbi	Toy Train (IOI17_train)	C++17	5 / 100	5 ms	860 KiB

train

#include <bits/stdc++.h>
#include "train.h"
 
#include <cstdio>
#include <vector>
#include <cassert>
 
using namespace std;
 
std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v){
    int n=a.size();
    int m=u.size();
    bool win[n];
    bool to[n];
    memset(to,0,sizeof to);
    memset(win,0,sizeof win);
    for (int i = 0; i < m; ++i)
    {
        if(u[i]==v[i]){
            win[u[i]]=true;
        }else to[u[i]]=true;
    }
    vector<int> ans(n,0);
    for (int i = n-1; i >=0; --i)
    {
        //cout <<win[i]<<" "<<a[i]<<endl;
        if(r[i]&&a[i]&&win[i]) ans[i]=1;
        else if(!r[i]&&!a[i]&&win[i]) ans[i]=0;
        else if(!r[i]&&!to[i]) ans[i]=0;
        else if(r[i]&&!to[i]) ans[i]=1;
        else ans[i]=ans[i+1];
    }
    return ans;
}

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