Codeforces803B Distances to Zero

最新推荐文章于 2019-11-11 21:00:41 发布

原创最新推荐文章于 2019-11-11 21:00:41 发布 · 603 阅读

1 ·

CC 4.0 BY-SA版权

一般动态规划专栏收录该内容

113 篇文章

订阅专栏

本文介绍了一个序列问题的解决方法，使用动态规划算法找到每个元素到最近0的距离，并通过两个数组记录左右两侧最近的0，最终得出每个位置上的最短距离。

链接

　　http://codeforces.com/problemset/problem/803/B

题目大意

　　给你一个序列，对每个数求出离它最近的 $0$ 跟它的距离。

题解

　　 $f_1[i]$ 表示左边最近的 $0$ 和这个数的距离， $f_2[i]$ 表示右边最近的 $0$ 和这个数的距离。
　　如果 $num_i=0$ ， $f_1[i]=f_2[i]=0$ ，否则 $f_1[i]=f_1[i-1]+1\ f_2[i]=f_2[i+1]+1$
　　最后 $ans_i=min(f_1[i],f_2[i])$

代码

//dp
#include <cstdio>
#include <algorithm>
#define maxn 200005
#define inf 0x3f3f3f3f
using namespace std;
int a[maxn], f[maxn], g[maxn], N;
int main()
{
    int i;
    scanf("%d",&N);
    for(i=1;i<=N;i++)scanf("%d",a+i);
    f[0]=inf, g[N+1]=inf;
    for(i=1;i<=N;i++)
    {
        if(a[i]==0)f[i]=0;
        else f[i]=f[i-1]+1;
    }
    for(i=N;i;i--)
    {
        if(a[i]==0)g[i]=0;
        else g[i]=g[i+1]+1;
    }
    for(i=1;i<=N;i++)printf("%d ",min(f[i],g[i]));
    return 0;
}