(C卷,200分)- 攀登者2（Java & JS & Python & C）

题目描述

攀登者喜欢寻找各种地图，并且尝试攀登到最高的山峰。

地图表示为一维数组，数组的索引代表水平位置，数组的元素代表相对海拔高度。其中数组元素0代表地面。

例如：[0,1,2,4,3,1,0,0,1,2,3,1,2,1,0]，代表如下图所示的地图，地图中有两个山脉位置分别为 1,2,3,4,5 和 8,9,10,11,12,13，最高峰高度分别为 4,3。最高峰位置分别为3,10。

一个山脉可能有多座山峰(高度大于相邻位置的高度，或在地图边界且高度大于相邻的高度)。

登山时会消耗登山者的体力(整数)，

• 上山时，消耗相邻高度差两倍的体力
• 下山时，消耗相邻高度差一倍的体力
• 平地不消耗体力

登山者体力消耗到零时会有生命危险。

例如，上图所示的山峰：

• 从索引0，走到索引1，高度差为1，需要消耗 2 * 1 = 2 的体力，
• 从索引2，走到索引3，高度差为2，需要消耗 2 * 2 = 4 的体力。
• 从索引3，走到索引4，高度差为1，需要消耗 1 * 1 = 1 的体力。

攀登者想要评估一张地图内有多少座山峰可以进行攀登，且可以安全返回到地面，且无生命危险。

例如上图中的数组，有3个不同的山峰，登上位置在3的山可以从位置0或者位置6开始，从位置0登到山顶需要消耗体力 1 * 2 + 1 * 2 + 2 * 2 = 8，从山顶返回到地面0需要消耗体力 2 * 1 + 1 * 1 + 1 * 1 = 4 的体力，按照登山路线 0 → 3 → 0 需要消耗体力12。攀登者至少需要12以上的体力（大于12）才能安全返回。

输入描述

第一行输入为地图一维数组

第二行输入为攀登者的体力

输出描述

确保可以安全返回地面，且无生命危险的情况下，地图中有多少山峰可以攀登。

用例

题目解析

本题考试时为核心代码模式，非ACM模式，即无需自己解析输入数据。

本题代码实现仍然以ACM模式处理，但是会将输入处理 与 算法逻辑 分开，大家只看算法逻辑即可。

 用例1图示：

用例说明中用的位置应该是从1开始计数的，换成数组索引的话，如下：

登山者只能登上位置9和11的山峰，6 → 9 → 6，13 → 11 → 13

登上索引9山峰的路线：6 → 9 → 6，消耗体力 = 9  < 13

登上索引11山峰的路线：13 → 11 → 13，消耗体力 = 6 < 13

另外还有一个索引2的山峰，有两条路线登山，分别为：

0 → 2 → 0，消耗体力 = 12 < 13

5 → 2 → 5，消耗体力 = 12 < 13

也能满足安全登山下山，但是题目用例说明没有给出，而题目输出是给的可攀登山峰数3，即索引位置2，9，11的三座山峰。

我的解题思路如下：

首先找到正向第一个地面（高度0）位置，然后从此位置开始攀登：

定义两个变量 upCost，downCost，分别代表上山体力消耗，下山体力消耗。

初始时都为0

假设索引 i 执行下一个位置，那么 height[i] – height[i-1] 就是高度差diff：

• 如果diff > 0，那么当前处于上坡路程，此时upCost += diff * 2，downCost += diff

上山时如果遇到上坡，那么相对应的，下山时，这段路程就变成了下坡

• 如果diff < 0，那么当前处于下坡路程，此时upCost -= diff，downCost -= diff * 2

上山时如果遇到下坡，

PS：为什么上山会遇到下坡路程？可以看下图，如果我想从索引6地面攀登到索引11山峰，那么中间必然要经过更高的索引9山峰，产生下坡路程

上山时如果遇到下坡，那么相对应的，下山时，这段路程就变成了上坡路。

注意：上面diff高度差 < 0，因此 upCost += -diff，就变为了 upCost -= diff。downCost同理。

• 如果diff == 0，那么当前不消耗体力

上面 diff = heights[i] – heights[i – 1]，如果 diff < 0 以及 diff == 0时，说明位置 i 的山不是山顶。

而 diff > 0 时，位置 i 是有可能为山顶的，我们只需要检查 heights[i] > heights[i+1]即可，比如

如果确定位置 i 是山顶，则攀登此山顶的体力消耗（上山+下山）= upCost + downCost

如果此时 upCost + downCost <= 自身体力（第二行输入），那么位置 i 山顶可攀登。

另外，我们在不断向后攀登的过程中，可能会重回地面（height[i] == 0），入下图绿色框部分

那么此时，对于后面的山峰来说，我们更优的攀登起始位置是绿色部分，而不是一开始的地面位置，因此，此时，我们应该将 upCost、downCost 重置为0，相当于从绿色部分重新攀登后面的山。

以上就是求解思路。但是存在漏洞，比如下图，我们攀登索引11的山峰，如果只正向考虑的话，则最优路径是 6 → 11 → 13

但是逆向考虑的话，会有更优解

即路径为 13 → 11 → 13

因此，我们应该正向、逆向都爬一次，这样才能保证所有山峰都能找到最优路径

2023.12.08

upCost 和 downCost 可以进行合并，具体请见代码注释。

攀登者体力必须要大于 上山、下山体力消耗之和，而不是大于等于。

JS算法源码

const rl = require("readline").createInterface({ input: process.stdin });
var iter = rl[Symbol.asyncIterator]();
const readline = async () => (await iter.next()).value;

// 输入处理
void (async function () {
  const heights = (await readline()).split(",").map(Number);
  const strength = parseInt(await readline());
  console.log(getResult(heights, strength));
})();

// 算法实现（本题实际考试为核心代码模式，因此考试时只需要写出此函数实现即可）
function getResult(heights, strength) {
  // 记录可攀登的山峰索引
  const idxs = new Set();

  // 正向攀登
  climb(heights, strength, idxs, true);
  // 逆序攀登
  climb(heights.reverse(), strength, idxs, false);

  return idxs.size;
}

function climb(heights, strength, idxs, direction) {
  // 找到第一个地面位置
  let j = 0;
  while (j < heights.length && heights[j] != 0) {
    j++;
  }

  let cost = 0; // 攀登体力总消耗（包括上山，下山）
  // let upCost = 0; // 上山体力消耗
  // let downCost = 0; // 下山体力消耗

  // 开始攀登
  for (let i = j + 1; i < heights.length; i++) {
    // 如果遇到了新的地面，则从新的地面位置重新计算攀登消耗的体力
    if (heights[i] == 0) {
      cost = 0;
      // upCost = 0;
      // downCost = 0;
      continue;
    }

    const diff = heights[i] - heights[i - 1]; // diff记录高度差

    if (diff > 0) {
      // 如果过程是上坡
      cost += diff * 3;
      // upCost += diff * 2; // 则上山时，体力消耗 = 高度差 * 2
      // downCost += diff; // 相反的下山时，体力消耗 = 高度差 * 1

      // 由于 height[i] > heights[i-1]，因此如果 height[i] > heights[i+1] 的话，位置 i 就是山顶
      if (i + 1 >= heights.length || heights[i] > heights[i + 1]) {
        // 计算攀登此山顶的上山下山消耗的体力和
        if (cost < strength) {
          // if (upCost + downCost < strength) {
          // 如果不超过自身体力，则可以攀登
          if (direction) {
            idxs.add(i);
          } else {
            idxs.add(heights.length - i - 1); // 需要注意，逆序heights数组后，我们对于的山峰位置需要反转
          }
        }
      }
    } else if (diff < 0) {
      cost -= diff * 3;
      // upCost -= diff; // 则上山时，体力消耗 = 高度差 * 1
      // downCost -= diff * 2; // 相反的下山时，体力消耗 = 高度差 * 2
      // heights[i] < heights[i-1]，因此位置i不可能是山顶
    }
  }
}

Java算法源码

import java.util.Arrays;
import java.util.HashSet;
import java.util.Scanner;

public class Main {
  // 输入处理
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);

    int[] heights = Arrays.stream(sc.nextLine().split(",")).mapToInt(Integer::parseInt).toArray();
    int strength = Integer.parseInt(sc.nextLine());

    System.out.println(getResult(heights, strength));
  }

  // 算法实现（本题实际考试为核心代码模式，因此考试时只需要写出此函数实现即可）
  public static int getResult(int[] heights, int strength) {
    // 记录可攀登的山峰索引
    HashSet<Integer> idxs = new HashSet<>();

    // 正向攀登
    climb(heights, strength, idxs, true);

    // 逆序攀登
    reverse(heights);
    climb(heights, strength, idxs, false);

    return idxs.size();
  }

  public static void climb(int[] heights, int strength, HashSet<Integer> idxs, boolean direction) {
    // 找到第一个地面位置
    int j = 0;
    while (j < heights.length && heights[j] != 0) {
      j++;
    }

    int cost = 0; // 攀登体力总消耗（包括上山，下山）
    //    int upCost = 0; // 上山体力消耗
    //    int downCost = 0; // 下山体力消耗

    // 开始攀登
    for (int i = j + 1; i < heights.length; i++) {
      // 如果遇到了新的地面，则从新的地面位置重新计算攀登消耗的体力
      if (heights[i] == 0) {
        cost = 0;
        //        upCost = 0;
        //        downCost = 0;
        continue;
      }

      int diff = heights[i] - heights[i - 1]; // diff记录高度差

      if (diff > 0) {
        // 如果过程是上坡
        cost += diff * 3;
        //        upCost += diff * 2; // 则上山时，体力消耗 = 高度差 * 2
        //        downCost += diff; // 相反的下山时，体力消耗 = 高度差 * 1

        // 由于 height[i] > heights[i-1]，因此如果 height[i] > heights[i+1] 的话，位置 i 就是山顶
        if (i + 1 >= heights.length || heights[i] > heights[i + 1]) {
          // 计算攀登此山顶的上山下山消耗的体力和
          if (cost < strength) {
            //          if (upCost + downCost <= strength) {
            // 如果小于自身体力，则可以攀登
            if (direction) {
              idxs.add(i);
            } else {
              idxs.add(heights.length - i - 1); // 需要注意，逆序heights数组后，我们对于的山峰位置需要反转
            }
          }
        }

      } else if (diff < 0) {
        // 如果过程是下坡
        cost -= diff * 3;
        //        upCost -= diff; // 则上山时，体力消耗 = 高度差 * 1
        //        downCost -= diff * 2; // 相反的下山时，体力消耗 = 高度差 * 2

        // heights[i] < heights[i-1]，因此位置i不可能是山顶
      }
    }
  }

  public static void reverse(int[] nums) {
    int i = 0;
    int j = nums.length - 1;

    while (i < j) {
      int tmp = nums[i];
      nums[i] = nums[j];
      nums[j] = tmp;
      i++;
      j--;
    }
  }
}

Python算法源码

# 输入获取
heights = list(map(int, input().split(",")))
strength = int(input())


def climb(idxs, direction):
    # 找到第一个地面位置
    j = 0
    while j < len(heights) and heights[j] != 0:
        j += 1

    # 上山体力消耗
    # upCost = 0
    # 下山体力消耗
    # downCost = 0

    # 攀登体力总消耗（包括上山，下山）
    cost = 0

    for i in range(j + 1, len(heights)):
        # 如果遇到了新的地面，则从新的地面位置重新计算攀登消耗的体力
        if heights[i] == 0:
            cost = 0
            # upCost = 0
            # downCost = 0
            continue

        # diff记录高度差
        diff = heights[i] - heights[i - 1]

        if diff > 0:
            # 如果过程是上坡
            cost += diff * 3
            # upCost += diff * 2  # 则上山时，体力消耗 = 高度差 * 2
            # downCost += diff  # 相反的下山时，体力消耗 = 高度差 * 1

            # 由于 height[i] > heights[i-1]，因此如果 height[i] > heights[i+1] 的话，位置 i 就是山顶
            if i + 1 >= len(heights) or heights[i] > heights[i + 1]:
                # 计算攀登此山顶的上山下山消耗的体力和
                if cost < strength:
                    # if upCost + downCost <= strength:
                    # 如果不超过自身体力，则可以攀登
                    if direction:
                        idxs.add(i)
                    else:
                        idxs.add(len(heights) - i - 1)  # 需要注意，逆序heights数组后，我们对于的山峰位置需要反转

        elif diff < 0:
            # 如果过程是下坡
            cost -= diff * 3
            # upCost -= diff  # 则上山时，体力消耗 = 高度差 * 1
            # downCost -= diff * 2  # 相反的下山时，体力消耗 = 高度差 * 2
            # heights[i] < heights[i-1]，因此位置i不可能是山顶


# 算法入口
def getResult():
    # 记录可攀登的山峰索引
    idxs = set()

    # 正向攀登
    climb(idxs, True)

    # 逆序攀登
    heights.reverse()
    climb(idxs, False)

    return len(idxs)


# 算法调用
print(getResult())

C算法源码

#include <stdio.h>

#define MAX_SIZE 100000

int canClimb[MAX_SIZE] = {0};
int canClimb_count = 0;

void climb(const int heights[], int heights_size, int strength, int direction) {
    // 找到第一个地面位置
    int j = 0;
    while (j < heights_size && heights[j] != 0) {
        j++;
    }

    int cost = 0; // 攀登体力总消耗（包括上山，下山）
//    int upCost = 0; // 上山体力消耗
//    int downCost = 0; // 下山体力消耗

    // 开始攀登
    for (int i = j + 1; i < heights_size; i++) {
        // 如果遇到了新的地面，则从新的地面位置重新计算攀登消耗的体力
        if (heights[i] == 0) {
            cost = 0;
//            upCost = 0;
//            downCost = 0;
            continue;
        }

        int diff = heights[i] - heights[i - 1]; // diff记录高度差

        if (diff > 0) {
            // 如果过程是上坡
            cost += diff * 3;
//            upCost += diff * 2; // 则上山时，体力消耗 = 高度差 * 2
//            downCost += diff; // 相反的下山时，体力消耗 = 高度差 * 1

            // 由于 height[i] > heights[i-1]，因此如果 height[i] > heights[i+1] 的话，位置 i 就是山顶
            if (i + 1 >= heights_size || heights[i] > heights[i + 1]) {
                // 计算攀登此山顶的上山下山消耗的体力和
                if (cost < strength) {
//                if (upCost + downCost < strength) {
                    // 需要注意，逆序heights数组后，我们对于的山峰位置需要反转
                    int idx = direction ? i : heights_size - i - 1;

                    if(!canClimb[idx]) {
                        // 如果不超过自身体力，则可以攀登
                        canClimb[i] = 1;
                        canClimb_count++;
                    }
                }
            }

        } else if (diff < 0) {
            // 如果过程是下坡
            cost -= diff * 3;
//            upCost -= diff; // 则上山时，体力消耗 = 高度差 * 1
//            downCost -= diff * 2; // 相反的下山时，体力消耗 = 高度差 * 2
            // heights[i] < heights[i-1]，因此位置i不可能是山顶
        }
    }
}

void reverse(int nums[], int nums_size) {
    int i = 0;
    int j = nums_size - 1;

    while (i < j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;

        i++;
        j--;
    }
}

// 算法实现（本题实际考试为核心代码模式，因此考试时只需要写出此函数实现即可）
int getResult(int heights[], int heights_size, int strength) {
    climb(heights, heights_size, strength, 1);

    reverse(heights, heights_size);
    climb(heights, heights_size, strength, 0);

    return canClimb_count;
}

// 输入处理
int main() {
    int heights[MAX_SIZE];
    int heights_size = 0;

    while (scanf("%d", &heights[heights_size++])) {
        if (getchar() != ',') break;
    }

    int strength;
    scanf("%d", &strength);

    printf("%dn", getResult(heights, heights_size, strength));

    return 0;
}