剑指 Offer -- 扑克牌顺子(四十五)

扑克牌顺子(四十五)

题目描述
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。

题目重新描述:
上面扯了那么一堆…就是说大小王认为是 0, 可以替换其他任何牌, 问一副牌是不是顺子?

代码(牛客上已 AC)

使用 count 统计 numbers 中 0 的个数, 并让 idx 指向第一个非 0 的元素; 如果 idx > 5, 即 0 的元素多于 4 个, 返回 false.

之后对于剩下的非 0 元素, 计算 diff = numbers[i + 1] - numbers[i], 如果 diff == 0, 说明出现了对子, 直接返回 false; 然后观察用 0 可以填补多少个空缺, 如果 diff == 1, 说明两个数字是连续的, count -= (diff - 1) 不会影响 count 的值.

最后只要保证 0 的个数 count >= 0, 就可以返回 true 了.

class Solution {
public:
    bool IsContinuous(vector<int> numbers) {
        if (numbers.size() != 5) return false;
        std::sort(numbers.begin(), numbers.end());
        int count = 0;
        int idx = 0;
        while (numbers[idx] == 0) { count ++; idx ++; }
        if (idx > 5) return false;
        for (int i = idx; i + 1 < numbers.size(); ++i) {
            int diff = numbers[i + 1] - numbers[i];
            if (diff == 0) return false;
            count -= (diff - 1);
        }
        if (count < 0) return false;
        return true;
    }
};