#include "Anthony.h"
#include <vector>
#include <queue>
using namespace std;
std::vector<int> Mark(int N, int M, int A, int B,std::vector<int> U, std::vector<int> V) {
vector<vector<pair<int,int>>> paths(N);
for(int i = 0;i<M;i++){
paths[U[i]].push_back({V[i],i});
paths[V[i]].push_back({U[i],i});
}
vector<int> dist(N,-1);
vector<int> col(M);
dist[0] = 0;
queue<int> q;
q.push(0);
while(!q.empty()){
auto now = q.front();
q.pop();
for(auto nxt:paths[now]){
if(dist[nxt.first] != -1)continue;
dist[nxt.first] = nxt.second+1;
col[nxt.second] = dist[nxt.first]%3;
q.push(nxt.first);
}
}
return col;
}
#include "Catherine.h"
#include <set>
#include <vector>
void Init(int A, int B) {
return;
}
using namespace std;
int Move(std::vector<int> y) {
set<int> st;
st.insert(1);
st.insert(2);
st.insert(0);
for(int i = 0;i<3;i++)if(y[i])st.erase(i);
assert(st.size()<3);
if(st.size() == 2){
for(int i = 0;i<3;i++)if(y[i])return i;
}
if(*st.begin() == 0)return 1;
else if(*st.begin() == 1)return 2;
else return 0;
return -1;
}
Compilation message
Catherine.cpp: In function 'int Move(std::vector<int>)':
Catherine.cpp:16:3: error: 'assert' was not declared in this scope
16 | assert(st.size()<3);
| ^~~~~~
Catherine.cpp:3:1: note: 'assert' is defined in header '<cassert>'; did you forget to '#include <cassert>'?
2 | #include <set>
+++ |+#include <cassert>
3 | #include <vector>
