剑指offer 45. 扑克牌顺子

原题

LL今天心情特别好,因为他去买了一副扑克牌,发现里面居然有2个大王,2个小王(一副牌原本是54张^_)…他随机从中抽出了5张牌,想测测自己的手气,看看能不能抽到顺子,如果抽到的话,他决定去买体育彩票,嘿嘿！！“红心A,黑桃3,小王,大王,方片5”,“Oh My God!”不是顺子…LL不高兴了,他想了想,决定大\小 王可以看成任何数字,并且A看作1,J为11,Q为12,K为13。上面的5张牌就可以变成“1,2,3,4,5”(大小王分别看作2和4),“So Lucky!”。LL决定去买体育彩票啦。 现在,要求你使用这幅牌模拟上面的过程,然后告诉我们LL的运气如何， 如果牌能组成顺子就输出true，否则就输出false。为了方便起见,你可以认为大小王是0。

Reference Answer

思路分析

听起来很花哨，其实就是含规则的连续数判定。

• 先统计王的数量，再把牌排序，如果后面一个数比前面一个数大于1以上，那么中间的差值就必须用王来补了。
• 看王的数量够不够，如果够就返回true，否则返回false。

# -*- coding:utf-8 -*-
class Solution:
    def IsContinuous(self, numbers):
        # write code here
        if len(numbers) < 5:
            return False
        zero = 0
        base = 0
        sum_gap = temp_gap = 0

        sort_numbers = sorted(numbers)
        for x in sort_numbers:
            if x == 0:
                zero += 1
            elif x != 0 and base == 0:
                base = x
            else:
                temp_gap = x - base - 1
                sum_gap += temp_gap
                if temp_gap == -1:
                    return False
                base = x
        if sum_gap <= zero:
            return True
        else:
            return False
                

C++ version:

class Solution {
public:
    bool IsContinuous( vector<int> numbers ) {
        int length = numbers.size();
        if (length < 5){
            return false;
        }
        int base = 0, gap = 0, zeros = 0;
        sort(numbers.begin(), numbers.end());
        for(int i = 0; i < length; i++){
            if(numbers[i] == 0){
                zeros ++;
            }
            else{
                if(base == 0){
                    base = numbers[i];
                }
                else{
                    gap += (numbers[i] - base - 1);
                    base = numbers[i];
                    if(gap == -1){
                        return false;
                    }
                }
            }
        }
        return (gap <= zeros)? true:false;
    }
};