TH40 牛牛摇骰子

描述

一维数轴，初始位置在原点，每次可以选择向左或者向右移动0或3或7或11个单位

N次询问，每次给出一个坐标arr[i]，求从0点走到arr[i]需要的最少次数

示例1

输入：

[1,4,14]

返回值：

[3,2,2]

说明：

从0到1最少需要3次(0->7->4->1)（走法不唯一，比如0->11->4->1也只需要3次）

从0到4最少需要2次(0->7->4)

从0到14最少需要2次(0->7->14)

示例2

输入：

[6,25]

返回值：

[2,3]

说明：

从0到6最少需要2次(0->3->6)

从0到25最少需要3次(0->7->18->25)

备注：

对于20%的数据1≤N≤5,1≤arr[i]≤1e5

对于50%的数据1≤N≤1e5,1≤arr[i]≤5e5

对于100%的数据1≤N≤1e5,1≤arr[i]≤1e9

import java.util.*;

public class Solution {
    public int[] getMinimumSteps(int[] arr) {
        List<Integer> result = new ArrayList<>();
        int maxCoordinate = 0;
        for (int a : arr) {
            maxCoordinate = Math.max(maxCoordinate, a);
        }
        
        // 初始化dp数组，dp[i]表示到达坐标i所需的最小步数
        int[] dp = new int[maxCoordinate + 1];
        Arrays.fill(dp, Integer.MAX_VALUE);
        dp[0] = 0;

        // 动态规划求解
        for (int i = 0; i <= maxCoordinate; i++) {
            if (dp[i] != Integer.MAX_VALUE) {
                // 向左移动3个单位
                if (i - 3 >= 0) {
                    dp[i - 3] = Math.min(dp[i - 3], dp[i] + 1);
                }
                // 向右移动7个单位
                if (i + 7 <= maxCoordinate) {
                    dp[i + 7] = Math.min(dp[i + 7], dp[i] + 1);
                }
                // 向左移动11个单位
                if (i - 11 >= 0) {
                    dp[i - 11] = Math.min(dp[i - 11], dp[i] + 1);
                }
                // 向右移动0个单位（原地不动）
                dp[i] = Math.min(dp[i], dp[i] + 1);
            }
        }

        // 将结果存入ArrayList并转换为int数组返回
        for (int coordinate : arr) {
            result.add(dp[coordinate]);
        }
        
        return result.stream().mapToInt(Integer::intValue).toArray();
    }
}

这段代码首先通过遍历输入数组 arr 确定最大坐标值，然后初始化一个动态规划数组 dp，其中 dp[i] 表示达到坐标 i 所需的最小步数。接着利用动态规划的方法更新数组，根据题目给出的移动规则（向左或向右移动0、3、7、11个单位）来计算从原点到每个位置的最小步数。最后将结果存入 ArrayList 并转换为 int 数组返回。