剑指offer—扑克牌中的顺子

剑指offer—扑克牌中的顺子

题目描述

从扑克牌中随机抽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]

思路

1：先把数组进行排序（小到大）
2：遍历数组
· 若 nums[i] == 0，zero++；
· 若后一个数 nums[i+1] == nums[i]，return false（有相同的牌肯定组不成顺子）；
· 若 nums[i+1] - nums[i] != 1，即需要判断 zero是否能够填补窟窿成顺子（填补不了返回 false）
3：最后返回 true

代码

    bool isStraight(vector<int>& nums) {
        int n = nums.size();
        sort(nums.begin(), nums.end());
        
        int zero = 0;
        for(int i = 0; i < (n-1); i++){
            if(nums[i] == 0) zero++;
            else if(nums[i+1] == nums[i]) return false;
            else if(nums[i+1] - nums[i] != 1){
                zero -= (nums[i+1] - nums[i] - 1);
            }
        }
        return zero>=0;
    }