有 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;
}