题目
原题链接:B. Bear and Finding Criminals
题意
有n座城市,警察在第a座,每座城市有最多一个小偷,给出距离a城市不同距离的城市小偷的数量,问最多能抓几个小偷。
参考了其他作者的代码。有两种情况可以判断该城市有小偷,一是距离相同的两个城市都有,二是一个超出范围,另一个有小偷,则能确定。
代码
#include
using namespace std;
int main() {
int n,a,s[101]={0},t=0,ans=0;
scanf("%d%d",&n,&a);
for(int i=1; i<=n; i++) {
scanf("%d",&s[i]);
if(s[i]) t++;
}
for(int i=1;i<=n;i++){
if(s[i]){
int tmp=a-(i-a);//相同距离的另一个城市
if(tmp>n || tmp<1 || s[tmp]){
ans++;
}
}
}
printf("%d\n",ans);
return 0;
}