이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 800010;
int n, q, ptr[maxn], pos[maxn], fa[maxn];
bool vis[maxn];
vector<int> G[maxn];
int main() {
ios::sync_with_stdio(0), cin.tie(0);
cin >> n >> q;
for (int i = 1, k; i <= n; i++) {
cin >> k;
while (k--) {
int c;
cin >> c, G[i].push_back(c);
}
}
auto dfs = [&](auto self, int u) -> void {
for (int v : G[u]) if (v ^ fa[u]) {
fa[v] = u, self(self, v);
}
};
dfs(dfs, 1);
ll step = 0;
for (int x = 1, cnt = 0; cnt < n || x > 1; step++) {
if (!vis[x]) vis[x] = 1, cnt++;
x = G[x][++pos[x] %= G[x].size()];
}
while (q--) {
ll k;
cin >> k;
int u = 1;
if (k < step) {
fill(ptr + 1, ptr + n + 1, 0);
while (k--) u = G[u][++ptr[u] %= G[u].size()];
cout << u << "\n"; continue;
}
k = (k - step) % (2 * n - 2);
copy(pos + 1, pos + n + 1, ptr + 1);
while (k--) u = G[u][++ptr[u] %= G[u].size()];
cout << u << "\n";
}
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |