4423. 最近距离

给定一个长度为 n 的整数数组 a1,a2,…,an。

对于每个整数 i（1≤i≤n），请你找到一个整数 j，要求：

1. 1≤j≤n
2. aj=0
3. 在满足以上两个条件的情况下，|i−j| 应尽可能小。|i−j| 的最小可能值不妨用 bi 来表示。

请你计算并输出 b1,b2,…,bn。

保证给定数组中一定存在 0。

输入格式

第一行包含整数 n。

第二行包含 n 个整数 a1,a2,…,an。

输出格式

一行，n 个整数 b1,b2,…,bn。

数据范围

前 44 个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤2×105，−109≤ai≤109。

输入样例1：

9
2 1 0 3 0 0 3 2 4

输出样例1：

2 1 0 1 0 0 1 2 3 

输入样例2：

5
0 1 2 3 4

输出样例2：

0 1 2 3 4

输入样例3：

7
5 6 0 1 -2 3 4

输出样例3：

2 1 0 1 2 3 4

分析：正序和倒序遍历了每个i距离它最近的j，然后两个去局里更小的情况。

#include <iostream>

using namespace std;
const int N=2e5+10;
int a[N],b[N];
int n;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    int id=-4e5;
    for(int i=1;i<=n;i++)
    {
        if(!a[i])id=i;
        b[i]=i-id;
    }
    id=4e5;
    for(int i=n;i;i--)
    {
        if(!a[i])id=i;
        b[i]=min(b[i],id-i);
    }
    for(int i=1;i<=n;i++)cout<<b[i]<<" ";
    return 0;
}