剑指 Offer 61. 扑克牌中的顺子

从扑克牌中随机抽5张牌,判断是不是一个顺子,即这5张牌是不是连续的。2~10为数字本身,A为1,J为11,Q为12,K为13,而大、小王为 0 ,可以看成任意数字。A 不能视为 14。

示例 1:

输入: [1,2,3,4,5]
输出: True

示例 2:

输入: [0,0,1,2,5]
输出: True

限制:

数组长度为 5 

数组的数取值为 [0, 13] .

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/bu-ke-pai-zhong-de-shun-zi-lcof 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

Solution 1

因为输入数组长度一定为5,所以不必太在意效率,封装命名加强可读性
如果要提高效率,可以通过快排后用哈希表进行统计

class Solution {
    public boolean isStraight(int[] nums) {
        if (nums.length != 5)
            return false;
        // for (int num : nums)
        // if (num < 0 || num > 13)
        // throw new RuntimeException();
        int largest = getLargest(nums);
        for (int next = largest - 1; next > largest - 5; next--) {
            if (next == peekLargest(nums))
                getLargest(nums);
            else if (getZero(nums))
                ;
            else
                return false;
        }
        return true;
    }

    private int peekLargest(int[] nums) {
        int largestIndex = 0;
        for (int i = 1; i < nums.length; i++) {
            if (nums[i] > nums[largestIndex])
                largestIndex = i;
        }
        int largestNum = nums[largestIndex];
        return largestNum;
    }

    private int getLargest(int[] nums) {
        int largestIndex = 0;
        for (int i = 1; i < nums.length; i++) {
            if (nums[i] > nums[largestIndex])
                largestIndex = i;
        }
        int largestNum = nums[largestIndex];
        nums[largestIndex] = -1;
        return largestNum;
    }

    private boolean peekZero(int[] nums) {
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == 0) {
                return true;
            }
        }
        return false;
    }

    private boolean getZero(int[] nums) {
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == 0) {
                nums[i] = -1;
                return true;
            }
        }
        return false;
    }
}