CodeForces - 803B 二分 模拟

Description

给定一个数列 a0, a1, …, an - 1。对于数列中的每一项都要求出与该项最近的0与该项的距离。保证数列中有至少一个0。

Input

第一行包含一个整数 n (1 ≤ n ≤ 2·105) — 表示数列 a 的长度。第二行包含这个数列的每一项。 ( - 109 ≤ ai ≤ 109).

Output

输出数列 d0, d1, …, dn - 1，其中数列 di 表示两侧距离 i 最近的 j 满足 aj = 0， i = j 是合法的。

Sample Input

9
2 1 0 3 0 0 3 2 4

5
0 1 2 3 4

7
5 6 0 1 -2 3 4

Sample Output

2 1 0 1 0 0 1 2 3

0 1 2 3 4

2 1 0 1 2 3 4

Hint

题意

题解：

AC代码

对每个距离 计算它与最近的0点的距离 二分查找零点时更新距离 只有两个0点的时候二分只会偏向一边所以要在查找时更新

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 200000+100;
const int INF = 0x3f3f3f3f;
int a[MAXN];
int b[MAXN];
inline int read(){
    int x = 0,f = 1; char ch = getchar();
    while (ch<'0'||ch>'9'){if (ch == '-') f = -1; ch = getchar();}
    while (ch>='0'&&ch<='9'){x = x*10+ch-'0';ch = getchar();}
    return x*f;
}
int main(){
    int n;
    cin>>n;

    int k = 0;
    for (int i = 0; i < n; ++i){
        a[i] = read();
        if (a[i] == 0) b[k++] = i;
    }
    int le,ri,mid,ans;
    int sp = 0;
    for (int i = 0; i < n; ++i){
        if (sp) putchar(' ');
        sp = 1;
        if (a[i] == 0) printf("0");
        else {
            le = 0,ri = k-1;
            ans = INF;
            while (le <= ri){
                 mid = (le+ri)>>1;
                 if (i<b[mid]) ri = mid-1;
                 else le = mid+1;
                 ans = min(ans,abs(i-b[mid]));
            }
            printf("%d",ans);
        }
    }
    printf("\n");
    return 0;
}