1. 程式人生 > >Distances to Zero CodeForces - 803B (二分)

Distances to Zero CodeForces - 803B (二分)

題目連結:https://vjudge.net/problem/CodeForces-803B#author=0

題意:

給你一個數組,其中至少包括一個0,求每一個元素距離最近一個0的距離是多少。

樣例:

Input
9
2 1 0 3 0 0 3 2 4
Output
2 1 0 1 0 0 1 2 3 

思路:把每一個0的下標都放進一個新陣列中,然後列舉每一個數組pos,用二分查詢pos在新陣列中剛好大於等於pos的那個index,
然後再和前後的index比較下取最小值,然後即得出答案。

作者的部落格原連結:https://www.cnblogs.com/qieqiemin/
我的AC程式碼:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#define rep(i,x,n) for(int i=x;i<n;i++)
#define
repd(i,x,n) for(int i=x;i<=n;i++) #define pii pair<int,int> #define pll pair<long long ,long long> #define gbtb std::ios::sync_with_stdio(false) #define MS0(X) memset((X), 0, sizeof((X))) #define MSC0(X) memset((X), '\0', sizeof((X))) #define pb push_back #define mp make_pair #define fi first #define
se second #define gg(x) getInt(&x) using namespace std; typedef long long ll; inline void getInt(int* p); const int maxn=1000010; const int inf=0x3f3f3f3f; /*** TEMPLATE CODE * * STARTS HERE ***/ int n; int ans[maxn]; int a[maxn]; int cnt=0; int l[maxn]; int main() { gg(n); repd(i,1,n) { gg(a[i]); } repd(i,1,n) { if(a[i]==0) { l[cnt++]=i; } } repd(i,1,n) { if(a[i]!=0) { int j1=lower_bound(l,l+cnt,i)-l; // int j2=uppercase() int v=l[j1]; if(v==0) { ans[i]=i-l[cnt-1]; }else { int num=abs(v-i); if(j1>=cnt-1) { }else { num=min(num,abs(l[j1+1]-i)); } if(j1>0) { num=min(num,abs(l[j1-1]-i)); } ans[i]=num; } } } repd(i,1,n) { printf("%d ",ans[i]); } return 0; } inline void getInt(int* p) { char ch; do { ch = getchar(); } while (ch == ' ' || ch == '\n'); if (ch == '-') { *p = -(getchar() - '0'); while ((ch = getchar()) >= '0' && ch <= '9') { *p = *p * 10 - ch + '0'; } } else { *p = ch - '0'; while ((ch = getchar()) >= '0' && ch <= '9') { *p = *p * 10 + ch - '0'; } } }
View Code