【Acw周赛补题】暂存 暑假再整理

#52  4423. 最近距离 

给定一个长度为 nn 的整数数组 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。

数据范围

前 4 个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤2×10^5，−10^9≤ai≤10^9。

输入样例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

#include <iostream>
#include <cmath>
using namespace std;

const int N=2*1e5;
int a[N],ans[N];

int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++) cin>>a[i];
    
    //计算左边
    for(int i=0,last=-n;i<n;i++)  //last用于记录0的位置
    {
        if(!a[i]) last=i;
        ans[i]=i-last;
    }
    //计算右边
    for(int i=n-1,last=n*2;i>=0;i--)
    {
        if(!a[i]) last=i;
        ans[i]=min(ans[i],last-i);
    }
    
    for(int i=0;i<n;i++) cout<<ans[i]<<' ';
}