Submission #65472495

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include <bits/stdc++.h>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin >> N;
    vector<int> C(N), A(N);
    for (int i = 1; i < N; i++) cin >> C[i];
    for (int i = 1; i < N; i++) cin >> A[i];
    vector<int> dist(N, 1e9);
    vector<int> par(N, -1);
    dist[0] = 0;
    queue<int> q;
    q.push(0);
    while (!q.empty()) {
        int i = q.front(); q.pop();
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin >> N;
    vector<int> C(N), A(N);
    for (int i = 1; i < N; i++) cin >> C[i];
    for (int i = 1; i < N; i++) cin >> A[i];

    vector<int> dist(N, 1e9);
    vector<int> par(N, -1);
    dist[0] = 0;

    queue<int> q;
    q.push(0);

    while (!q.empty()) {
        int i = q.front(); q.pop();
        for (int j = i + 1; j < N; j++) {
            if (j - C[j] <= i && i < j) {
                if (dist[j] > dist[i] + 1) {
                    dist[j] = dist[i] + 1;
                    par[j] = i;
                    q.push(j);
                }
            }
        }
    }
    vector<bool> used(N, false);
    for (int i = 1; i < N; i++) {
        if (A[i] == 1) {
            int x = i;
            while (x != -1 && !used[x]) {
                used[x] = true;
                x = par[x];
            }
        }
    }
    int ans = 0;
    for (int i = 1; i < N; i++) {
        if (used[i]) ans++;
    }
    cout << ans << "\n";
    return 0;
}

Submission Info

Judge Result