LeetCode 剑指 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] .

题目链接

思路

1.先把非零（即不是大小王的）元素取出然后排序，对零（大小王）计数；
2.循环判断非零元素是否有相同元素，有相同的不会构成顺子；
3.依次循环后面的元素减去前面的元素再-1，最后对差值求和如果求和的数据小于等于所有零的个数则满足要求，相当于把大小王变成任意数插进去。
有坑的地方，注释处，零（大小王）居然可以大于2，一副牌中不是最多一对大小王吗？

题解

class Solution {
public:
    bool isStraight(vector<int>& nums) {
        int cnt=0;
        vector<int> res;
        for(int i=0;i<nums.size();i++){
            if(nums[i]==0) cnt++;
            else res.push_back(nums[i]);
        }
        //if(cnt>2) return false;
        sort(res.begin(),res.end());
        int sum=0;
        for(int i=1;i<res.size();i++){
            if(res[i]==res[i-1]) return false;
            else sum+=res[i]-res[i-1]-1;
        }
        return sum<=cnt;
    }
};