思路:
第一眼-->"暴力"!,一看数据,不行,好好好,下播!(tao~)
后来认真一想其实我们只需要分别求出当前位置单单离左右两边的最近0位置,然后再从左右中一个min最小值即可啦~
复杂度:O(2*N)稳过!
细节请看下方代码:
ACcode:
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=2e5+10;
const int inf=0x3f3f3f3f;
int n,a[N],b[N];
void solve() {
cin>>n;
int j=-1;
for(int i=0; i<n; i++) cin>>a[i];
//从左跑一遍
for(int i=0; i<n; i++) {
if(a[i]==0)j=i;
if(j!=-1)b[i]=i-j;
else b[i]=inf;
}
j=-1;
for(int i=n-1;i>=0;i--){
if(a[i]==0)j=i;
if(j!=-1)b[i]=min(b[i],j-i);
}
for(int i=0;i<n;i++) cout<<b[i]<<" ";
}
signed main() {
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
int t=1;
// cin>>t;
while(t--) {
solve();
}
return 0;
}
okk