1333: 找罪犯

题目

Description

有n个城市，编号为1到n。城市i到j的距离等于|i-j|。
马克是个警察。他住在一个城市里，他的工作是抓罪犯。对于他说这是很困难的，因为他不知道哪个城市的罪犯。虽然，他知道每个城市最多一个罪犯。
马克使用一种先进仪器BCD（刑事检测器）。BCD会告诉马克，距离马克为0,1,2…的城市分别有多少罪犯，马克可以赶到每个城市，他肯定要有一个刑事犯罪。
当然你知道哪座城市里有罪犯。计算马克使用BCD能抓捕到的罪犯数。
Input

多组输入
输入的第一行包含两个整数n和a（1≤a≤n≤100）城市数量和马克居住的城市。
第二行包含n个整数t1,t2,…,tn（0≤ti≤1）。在第i个城市有ti个罪犯。

Output

输出马克抓捕的罪犯数量。

Sample Input

6 3
1 1 1 0 1 0
5 2
0 0 0 1 0
Sample Output

3
1
HINT

在第一个例子中，有6个城市，BCD会给马克的信息

距离马克0的城市有一个罪犯，马克会确定t3会有罪犯

距离马克1的城市有一个罪犯，马克不确定t2，t4哪座城市有罪犯

距离马克2的城市有两个罪犯，马克会确定t1和t5都会有罪犯

距离更大的城市为0个罪犯

所以第一个例子中输出结果为3，只有t1,t3,t5三个城市确定有罪犯

第二个例子中，BCD会给出只有距离为2的城市有一个罪犯，距离马克为2的城市也只有一个，所以第二个例子输出结果为1

---

代码块

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner cn = new Scanner(System.in);
        while (cn.hasNext()) {
            int n = cn.nextInt();
            int a = cn.nextInt();
            int[] t = new int[n];
            for (int i = 0; i < n; i++) {
                t[i] = cn.nextInt();
            }

            int count = 0;
            if (t[a - 1] > 0)
                count++;
            int i = a - 2, j = a;
            for (;; i--, j++) {
                if (i < 0 || j >= n)
                    break;
                if (t[i] + t[j] > 1) {
                    count += 2;
                }
            }
            while (i >= 0) {
                if (t[i] > 0)
                    count++;
                i--;
            }
            while (j < n) {
                if (t[j] > 0)
                    count++;
                j++;
            }
            System.out.println(count);
        }
    }
}