4421. 信号(接近O(n))

有 n 个房子排成一排,从左到右依次编号为 1∼n。

其中一些房子内装有无线信号发射器。

这些信号发射器的有效覆盖半径为 r。

更准确地说,如果第 p 号房子内装有信号发射器,则所有房间编号在 [p−r+1,p+r−1][p−r+1,p+r−1] 范围内的房子均可被其发出的无线信号覆盖,而其余房子则不会被其发出的无线信号覆盖。

例如,假设 n=6,r=2,且第 2、5 号房子内装有信号发射器,则第 2 号房子内的发射器发出的信号可以覆盖第 1∼3 号房子,第 5 号房子内的发射器发出的信号可以覆盖第 4∼6 号房子,将两个发射器全部打开,则无线信号可以覆盖到所有房子。

初始时,所有无线信号发射器都是关闭的,请计算至少开启多少个无线信号发射器,才能保证所有房子都被无线信号覆盖到。

输入格式

第一行包含两个整数 n 和 r。

第二行包含 n 个整数 a1,a2,…,an,每个数要么是 1,要么是 0,ai=1 表示第 i 号房子内装有无线信号发射器,ai=0 表示第 i 号房子内未装无线信号发射器。

输出格式

一个整数,表示需要开启的无线信号发射器的最少数量。

如果无法使得所有房子都被无线信号覆盖到,则输出 −1。

数据范围

前 6 个测试点满足 1≤n≤10。
所有测试点满足 1≤n,r≤1000,0≤ai≤1。

输入样例1:

6 2
0 1 1 0 0 1

输出样例1:

3

输入样例2:

5 3
1 0 0 0 1

输出样例2:

2

输入样例3:

5 10
0 0 0 0 0

输出样例3:

-1

输入样例4:

10 3
0 0 1 1 0 1 0 0 0 1

输出样例4:

3

分析: 

(记忆化搜索)
这道题可以使用记忆化搜索,我的dfs(int id)只有一个参数id表示第id个信号发射器为打开状态,因此在dfs中遍历的的就是在id为打开的情况下,下一个需要到开的信号发射器是谁,因为id打开后,下一个信号发射器在最不浪费的情况下应该是id+r+r-1,所以我们倒着遍历,遍历下一个打开的发射器(i从id+r+r-1到id+1)当有一个可以成功的时候我们就停止遍历,直接返回该值,需要注意的是我们需要用dp数组存一下该数字,这样下次访问到时可以直接返回。

其实官解是线性dp,但每次都不太能想起来怎么dp,哎,那就每次都写记忆化搜索好了。

代码如下:

#include <iostream>

using namespace std;
const int N=2010;
int a[N],dp[N];
int n,r;

int dfs(int id)
{

    if(dp[id]) return dp[id];//遍历过的不需要在来一遍
    if(id+r>n) return 0;//表示该方法可行,可以在信号全部覆盖的情况下到达n
    int ans=-1;
    for(int i=min(id+2*r-1,n);i>id;i--)//在id打开的情况下,寻找下一个要打开的点,倒着遍历运用了贪心的思想
    {
        if(!a[i]) continue;
        int res=dfs(i);
        if(res==-1) continue;
        ans=res+1;
        break;//有一个符合结果便可以直接结束遍历
    }
    dp[id]=ans;//若都不符合结果,则证明该种方案无解,返回-1
    //cout<<id<<" "<<dp[id]<<endl;
    return ans;
}
int main()
{
    cin>>n>>r;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=min(r,n);i;i--)//寻找第一个需要打开的标号
    {
        if(!a[i])continue;
        int res=dfs(i);
        if(res==-1)continue;
        cout<<res+1;
        return 0;
    }
    cout<<-1;
    return 0;
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值