CodeFoeces-680B

题目

原题链接: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;
}

你可能感兴趣的:(CodeFoeces-680B)