Go&Java算法之扑克牌中的顺子

434 阅读1分钟

这是我参与2022首次更文挑战的第1天,活动详情查看:2022首次更文挑战

扑克牌中的顺子

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

题解

算法一:排序+遍历(Java)

根据题意,此 5 张牌是顺子的 充分条件 如下:

除大小王外,所有牌无重复 ; 设此 5 张牌中最大的牌为 max ,最小的牌为 min (大小王除外),则需满足:max-min < 5

  • 先对数组执行排序
  • 判别重复: 排序数组中的相同元素位置相邻,因此可通过遍历数组,判断 nums[i] = nums[i + 1]是否成立来判重。
  • 获取最大 / 最小的牌: 排序后,数组末位元素 nums[4]为最大牌;元素 nums[joker]为最小牌,其中 joker 为大小王的数量。
class Solution {
    public boolean isStraight(int[] nums) {
        if (nums == null || nums.length == 0) {
            return false;
        }

        Arrays.sort(nums);

        int zeroNum = 0;
        for (int i = 0; i < nums.length - 1; i++) {
            if (nums[i] == 0) {
                zeroNum++;
            } else {
                if (nums[i] == nums[i + 1]) {
                    return false;
                }
            }
        }

        return nums[nums.length - 1] - nums[zeroNum] < 5;
    }
}

时间复杂度:O(NlogN)

空间复杂度:O(1)

算法一:排序+遍历(Go)

func isStraight(nums []int) bool {
	joker := 0
	sort.Ints(nums)
	for i:=0;i<4;i++ {
		if nums[i] == 0 {
			joker++
		} else if nums[i] == nums[i+1] {
			return false
		}
	}
	return nums[4] - nums[joker] < 5
}

时间复杂度:O(NlogN)

空间复杂度:O(1)

算法二:排序+Set(Java)

  • 遍历五张牌,遇到大小王(即 0 )直接跳过。
  • 判别重复: 利用 Set 实现遍历判重, Set 的查找方法的时间复杂度为 O(1);
  • 获取最大 / 最小的牌: 借助辅助变量 max 和 min ,遍历统计即可。
class Solution {
    public boolean isStraight(int[] nums) {
        Set<Integer> repeat = new HashSet<>();
        int max = 0, min = 14;
        for(int num : nums) {
            if(num == 0) continue; 
            max = Math.max(max, num); 
            min = Math.min(min, num); 
            if(repeat.contains(num)) return false; 
            repeat.add(num);
        }
        return max - min < 5;
    }
}

时间复杂度:O(N)

空间复杂度:O(N)