剑指offer - 45 -- 扑克牌顺子 - C++

题目描述

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。

示例1

输入

[0,3,2,6,4]

返回值

true

题解

方法一：set+遍历

我们分两种情况考虑，
一. 如果vector中不包含0的情况：
那么如何判断呢？因为需要是顺子，所以首先不能有重复值， 如果没有重复值，那么形如[1 2 3 4 5]
[5 6 7 8 9]， 会发现最大值与最小值的差值应该小于5.

二. 如果vector中包含0：
发现除去0后的值，判断方法和1中是一样的。

所以根据如上两个条件，算法过程如下：

1. 初始化一个set，最大值max_ = 0, 最小值min_ = 14
2. 遍历数组， 对于大于0的整数，没有在set中出现，则加入到set中，同时更新max_, min_
3. 如果出现在了set中，直接返回false
4. 数组遍历完，最后再判断一下最大值与最小值的差值是否小于5

class Solution {
public:
    bool IsContinuous( vector<int> numbers ) {
        if (numbers.size() <= 0)
            return false;
        set<int> st;
        int max_ = 0, min_ = 14;
        for(int i=0; i<numbers.size(); i++){
            if (numbers[i] > 0){
                if(st.count(numbers[i]) > 0)
                    return false;
                st.insert(numbers[i]);
                max_ = max(max_, numbers[i]);
                min_ = min(min_, numbers[i]);
            }
        }
        return max_ - min_ < 5;
    }
};

或者可以直接遍历，记录下每个数字出现的个数即可，非0数字出现次数大于1时直接返回false

class Solution {
public:
    bool IsContinuous( vector<int> numbers ) {
        if(numbers.size() != 5)
            return false;
        vector<int> res(14, 0);
        int min = INT_MAX;
        int max = INT_MIN;
        for(int i=0; i<numbers.size(); i++){
            res[numbers[i]] += 1;
            if(numbers[i] == 0)
                continue;
            if(res[numbers[i]] > 1)
                return false;
            if(numbers[i] > max)
                max = numbers[i];
            if(numbers[i] < min)
                min = numbers[i];
        }
        return max - min < 5;
    }
};

时间复杂度：O(N)
空间复杂度：O(N)

方法二：排序+遍历

根据方法一的分析，实现上如果不用set判断是否有重复值的话，还可以先排序，然后如果有重复值，那么肯定相邻。

class Solution {
public:
    bool IsContinuous( vector<int> numbers ) {
        if(numbers.size() <= 0)
            return false;
        sort(numbers.begin(), numbers.end());  //先排序
        int min_idx = 0;
        for(int i=0; i<numbers.size(); i++){
            if(numbers[i] == 0){   //排序过的数组，0在最前面，找到第一个不为0的数，即为最小数字的下标
                min_idx += 1;
                continue;
            }
            if((i+1 < numbers.size()) && (numbers[i] == numbers[i+1]))   //如果有重复值，那么肯定是相邻的，则不可能成为顺子
                return false;
        }
        return numbers.back() - numbers[min_idx] < 5;
    }
};

时间复杂度：O(NlogN)
空间复杂度：O(1)