其他分享
首页 > 其他分享> > 贪心-2350. 不可能得到的最短骰子序列

贪心-2350. 不可能得到的最短骰子序列

作者:互联网

问题描述

给你一个长度为 n 的整数数组 rolls 和一个整数 k 。你扔一个 k 面的骰子 n 次,骰子的每个面分别是 1 到 k ,其中第 i 次扔得到的数字是 rolls[i] 。

请你返回 无法 从 rolls 中得到的 最短 骰子子序列的长度。

扔一个 k 面的骰子 len 次得到的是一个长度为 len 的 骰子子序列 。

注意 ,子序列只需要保持在原数组中的顺序,不需要连续。

示例 1:

输入:rolls = [4,2,1,2,3,3,2,4,1], k = 4
输出:3
解释:所有长度为 1 的骰子子序列 [1] ,[2] ,[3] ,[4] 都可以从原数组中得到。
所有长度为 2 的骰子子序列 [1, 1] ,[1, 2] ,... ,[4, 4] 都可以从原数组中得到。
子序列 [1, 4, 2] 无法从原数组中得到,所以我们返回 3 。
还有别的子序列也无法从原数组中得到。
示例 2:

输入:rolls = [1,1,2,2], k = 2
输出:2
解释:所有长度为 1 的子序列 [1] ,[2] 都可以从原数组中得到。
子序列 [2, 1] 无法从原数组中得到,所以我们返回 2 。
还有别的子序列也无法从原数组中得到,但 [2, 1] 是最短的子序列。
示例 3:

输入:rolls = [1,1,3,2,2,2,3,3], k = 4
输出:1
解释:子序列 [4] 无法从原数组中得到,所以我们返回 1 。
还有别的子序列也无法从原数组中得到,但 [4] 是最短的子序列。

提示:
n == rolls.length
1 <= n <= 105
1 <= rolls[i] <= k <= 105

问题求解

每次都去寻找所有元素,只有集齐所有元素才能完成拼接到更长的序列。

class Solution:
    def shortestSequence(self, rolls: List[int], k: int) -> int:
        res = 1
        record = set()
        for roll in rolls:
            if roll not in record:
                record.add(roll)
                if len(record) == k:
                    res += 1
                    record = set()
        return res

标签:骰子,得到,数组,从原,序列,2350,rolls,贪心
来源: https://www.cnblogs.com/hyserendipity/p/16536150.html