You are given the array of integer numbers a0, a1, ..., an - 1. For each element find the distance to the nearest zero (to the element which equals to zero). There is at least one zero element in the given array.
InputThe first line contains integer n (1 ≤ n ≤ 2·105) — length of the array a. The second line contains integer elements of the array separated by single spaces ( - 109 ≤ ai ≤ 109).
Print the sequence d0, d1, ..., dn - 1, where di is the difference of indices between i and nearest j such that aj = 0. It is possible that i = j.
Input
9 2 1 0 3 0 0 3 2 4
Output
2 1 0 1 0 0 1 2 3
Input
5 0 1 2 3 4
Output
0 1 2 3 4
Input
7 5 6 0 1 -2 3 4
Output
2 1 0 1 2 3 4
这题很好理解就是求这个数与 离它最近的0的 距离,
题解:不断比较离它近的0的距离,更新距离就行
#include <bits/stdc++.h>
using namespace std;
vector<int>v;
int main()
{
int n;
int x;
scanf("%d",&n);
for(int i = 0;i < n; i++) {
scanf("%d",&x);
if(x==0) v.push_back(i);
}
int k = 0;
for(int i = 0;i < n; i++) {
int ans = abs(i-v[k]);//求距离
if(k>0) {
if(ans > 0) ans = min(ans,abs(i-v[k-1])); // 比较和上一个零点哪个离零点小
}
if(ans == 0) k++;
if(i)
printf(" %d",ans);
else printf("%d",ans);
}
printf("\n");
return 0;
}