门| GATE CS Mock 2018年|套装2 |问题30

该篇文章为 GATE CS Mock 2018年的第30个问题。

这个问题要求我们编写Java代码，输出一个关于左导致右的路径方案。

代码实现

Java代码实现如下：

import java.util.Scanner;

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

    int[] result = new int[n];
    for (int i = n - 2; i >= 0; i--) {
      if (arr[i] == 1) {
        result[i] = i + 1;
      } else {
        int j = i + 1;
        while (arr[j] != 1 && j < n - 1) {
          j = result[j];
        }
        result[i] = j;
      }
    }

    for (int i = 0; i < n; i++) {
      System.out.print(result[i] + " ");
    }
  }
}

代码说明

主要思路是：

1. 读取用户输入的整数数组；
2. 初始化一个新的整数数组 result；
3. 反向遍历数组 arr，并按下述方式为 result 数组中的每个元素赋值：
   • 当前元素的值为1时，直接将 result 数组中对应元素设为 i+1;
   • 当前元素的值为 0 时，则在 result 数组中查找右侧相邻的第一个值为1的元素，如果找到，则将结果设置为该元素的下标；如果找不到，说明不存在从该下标可达到1的路径，将该元素的结果值设为 n（即不可达）；

最后输出result数组中的每个元素。

代码的时间复杂度为 O(n)，空间复杂度为 O(n)。

总结

此代码是一个基础的 Java 编程问题，但概念性较强。本题需要灵活使用数组处理思想来遍历输入的数组，并将结果保存在数组中。

对于初学者来说，需要加强对数组及其遍历方法的理解。