Leetcode.1033 移动石子直到连续

题目链接

Leetcode.1033 移动石子直到连续 Rating ： 1421

题目描述

三枚石子放置在数轴上，位置分别为 a，b，c。

每一回合，你可以从两端之一拿起一枚石子（位置最大或最小），并将其放入两端之间的任一空闲位置。形式上，假设这三枚石子当前分别位于位置 x, y, z且 $x<y<z$。那么就可以从位置 x或者是位置 z拿起一枚石子，并将该石子移动到某一整数位置 k处，其中 $x<k<z$ 且 $k\ne y$。

当你无法进行任何移动时，即，这些石子的位置连续时，游戏结束。

要使游戏结束，你可以执行的最小和最大移动次数分别是多少？ 以长度为 2 的数组形式返回答案：answer = [minimum_moves, maximum_moves]

示例 1：

输入：a = 1, b = 2, c = 5
输出：[1, 2]
解释：将石子从 5 移动到 4 再移动到 3，或者我们可以直接将石子移动到 3。

示例 2：

输入：a = 4, b = 3, c = 2
输出：[0, 0]
解释：我们无法进行任何移动。

提示：

• $1<=a<=100$
• $1<=b<=100$
• $1<=c<=100$
• $a!=b,b!=c,c!=a$

解法：分类讨论

先将 $a,b,c$按从小到大的顺序排序。

最小值：

• 当 $b-a=1$ and $c-a=1$时，不需要移动，故答案为 0
• 当 $b-a\le 2$ or $c-a\le 2$时，需要移动一次，故答案为 1
• 否则需要移动两次，故答案为 2

最大值：$c-a-2$

时间复杂度：$O(1)$

C++代码：

class Solution {
public:
    vector<int> numMovesStones(int a, int b, int c) {
        vector<int> arr{a,b,c};
        sort(arr.begin(),arr.end());    

        int ans;
        if(arr[1] - arr[0] == 1 && arr[2] - arr[1] == 1) ans = 0;
        else if(arr[1] - arr[0] <= 2 || arr[2] - arr[1] <= 2) ans = 1;
        else ans = 2;

        return {ans,arr[2] - arr[0] - 2};
    }
};