Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <iostream>
- #include <vector>
- #include <algorithm>
- using namespace std;
- int main() {
- int n;
- cin >> n;
- vector <int> a(n);
- vector <int> result(n, -1);
- int null = -1;
- for (int i = 0; i < n; i++) {
- cin >> a[i];
- if (a[i] == 0) {
- null = i;
- result[i] = 0;
- }
- else if (null != -1)
- result[i] = abs(i - null);
- }
- null = -1;
- for (int i = n - 1; i >= 0; i--)
- if (a[i] == 0)
- null = i;
- else if (null != -1)
- if (result[i] != -1)
- result[i] = min(abs(i - null), result[i]);
- else result[i] = abs(i - null);
- for (int i = 0; i < n; i++)
- cout << result[i] << " ";
- cout << endl;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement