蓝桥杯赛前冲刺补习第三课——《典型问题的递归框架》

第一题

长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左，有的朝右。
每只蚂蚁都只能沿着杆子向前爬，速度是1厘米/秒。
当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。
这些蚂蚁中，有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。
请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

【数据格式】
第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。
接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。

要求输出1个整数，表示最后感冒蚂蚁的数目。
例如，输入：
3
5 -2 8
程序应输出：1
再例如，输入：
5
-10 8 -20 12 25
程序应输出：3

---

解法：
  这里笔者想引入另外一个题目，就是普通的蚂蚁爬杆：

  有一根27厘米的细木杆，在第3厘米、7厘米、11厘米、17厘米、23厘米这五个位置上各有一只蚂蚁。木杆很细，不能同时通过两只蚂蚁。开始时，蚂蚁头朝向任意（左或右），它们只会朝前或掉头，不会后退。当任意两只蚂蚁碰头是，两只蚂蚁会同时调头朝反向走。假设蚂蚁每秒走1厘米，编写程序，求所有蚂蚁都离开木杆的最短时间和最长时间。
  这道题相对蚂蚁感冒来说更容易一些，这道题的解法有一个核心要点，就是当两只蚂蚁碰面的时候应该如何操作？
  比如可以当数组中这两个变量的值的绝对值相等时，将这两个值的正负号翻一下——放到现实意义中就是两个蚂蚁替换了一下序号继续向着原来的方向走了。一旦这样想问题之后，我们立马就可以知道其实这道题只需要计算离终点最远的那只蚂蚁走过的时间就好。

  但是蚂蚁感冒和单纯的爬杆又不同，我们得到的最终结果是有几只蚂蚁会感冒，杆子的长度并没有做限制，这样我们就不能类比爬杆的递归出口条件。但是却可以引用其中蚂蚁碰面的思想。

上面的引用叙述只是一个引子，现在我们开始正式解决蚂蚁感冒的问题：
  首先，第一只蚂蚁感冒了；与其面对面而行的蚂蚁也一定会在将来被传染。在传染的同时，我们不用贴合现实地将蚂蚁反转，而是可以让蚂蚁以原本的方向接着前行，这就是我引用爬杆问题的目的——蚂蚁转不转向其实没有关系。
  到这里，相信大家已经看懂了，这道题的递归过程是多么的简单，这里我的第一次思考写得比较复杂，先给出附带详细注释的代码：

/**
 * 蚂蚁感冒第一次思考
 * 
 * @param antsStatus 蚂蚁当前的感染状态
 * @param distances 蚂蚁当前当前位置
 * @param k 当前状态下，将来能感染最多蚂蚁的蚂蚁位置
 * @return
 */
public static int coughingAntSpreadingFirstThink(boolean[] antsStatus, int[] distances, int k) {
    /*main方法
        Scanner scanner = new Scanner(new BufferedInputStream(System.in));

        int   n         = scanner.nextInt();
        int[] distances = new int[n];

        for (int i = 0; i < n; i++) {
            distances[i] = scanner.nextInt();
        }

        boolean[] antsStatus = new boolean[n];
        Arrays.fill(antsStatus, false);
        antsStatus[0] = true;

        System.out.println(coughingAntSpreadingFirstThink(antsStatus, distances, 0));
     */
    /*
        递归出口，说明上次寻找未找到继续向后传染的对象
     */
    if (k == -1) {
        int count = 0;
        for (boolean bool : antsStatus) {
            if (bool) {
                count++;
            }
        }
        return count;
    }

    int drection   = distances[k] > 0 ? 1 : -1;//当前要考虑的蚂蚁的行走方向
    int finalPoint = -1;//用来保存最终感染到的蚂蚁位置
    /*
        由于每次感染只考虑和当前考虑的蚂蚁方向相反的蚂蚁
        所以每次过程感染的蚂蚁都是同向的
        而下次过程中，这些同向的蚂蚁所要感染的蚂蚁是一个大集合中包含多个真子集的形式
        换言之，我们只需要找到感染集合最大，也就是距离最远的那只蚂蚁
        就可以知道下次过程中那些蚂蚁将被感染
     */
    for (int i = 0; i < distances.length; i++) {
  //遍历距离数组
        if (drection > 0){
        //当当前考虑点的方向向右，其感染范围是所有距离左端距离比该点大的点
        //且已经被感染的点不做考虑
            if (Math.abs(distances[i]) > Math.abs(distances[k]) && !antsStatus[i]) {
                //如果距离绝对值满足要求，则判断这只蚂蚁是否与考虑的蚂蚁相向而行
                //只有面对面走来的蚂蚁才会被感染
                int d = distances[i] > 0 ? 1 : -1;
                if (d != drection) {
                    finalPoint = i;//感染完成，将下标保存
                    antsStatus[i] = true;
                }
            }
        } else {
        //当当前考虑点的方向向右，其感染范围是所有距离左端距离比该点小的点
        //且已经被感染的点不做考虑
            if (Math.abs(distances[i]) < Math.abs(distances[k]) && !antsStatus[i]) {
                //如果距离绝对值满足要求，则判断这只蚂蚁是否与考虑的蚂蚁相向而行
                //只有面对面走来的蚂蚁才会被感染
                int d = distances[i] > 0 ? 1 : -1;
                if (d != drection) {
                    finalPoint = i;//感染完成，将下标保存
                    antsStatus[i] = true;
                }
            }
        }
    }

    //如果此次过程成功感染，则要开始每只蚂蚁的移动
    if (finalPoint != -