Distances to Zero CodeForces - 803B (二分)
阿新 • • 發佈:2019-01-08
題目連結:https://vjudge.net/problem/CodeForces-803B#author=0
題意:
給你一個數組,其中至少包括一個0,求每一個元素距離最近一個0的距離是多少。
樣例:
Input9Output
2 1 0 3 0 0 3 2 4
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++) #defineView Coderepd(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 #definese 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'; } } }