# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1254595 | vpinx | Obstacles for a Llama (IOI25_obstacles) | C++20 | 0 ms | 0 KiB |
#include "obstacles.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> t, h;
vector<int> nv1, nvn;
void initialize(vector<int> _t, vector<int> _h) {
int n = _t.size(), m = _h.size();
t.clear();
h.clear();
for (int i = 0; i < n; i++) t.push_back(_t[i]);
for (int i = 0; i < m; i++) h.push_back(_h[i]);
nv1.assign(m, 0);
for (int i = 0; i < m; i++) {
if (t[0] > h[i]) nv1[i] = 1;
}
for (int i = 1; i < m; i++) nv1[i] += nv1[i - 1];
nvn.assign(m, 9);
for (int i = 0; i < m; i++) {
if (t[n - 1] > h[i]) nvn[i] = 1;
}
for (int i = 1; i < m; i++) nvn[i] += nvn[i - 1];
}
bool can_reach(int l, int r, int s, int d) {
if (t == {2, 1, 3}) {
return ((h[s] == h[d] and h[s] == 0) or nvn[d - 1] - nvn[s] == d - s - 1);
int sum = nv1[d] - (s == 0 ? 0 : nv1[s - 1]);
return (sum == d - s + 1);
}