LeetCode 1955. 统计特殊子序列的数目

Michael阿明

发布于 2021-09-06 11:23:40

4190

发布于 2021-09-06 11:23:40

文章被收录于专栏：Michael阿明学习之路

文章目录

- 1. 题目
- 2. 解题

1. 题目

特殊序列是由正整数个 0 ，紧接着正整数个 1 ，最后正整数个 2 组成的序列。

比方说，[0,1,2] 和 [0,0,1,1,1,2] 是特殊序列。相反，[2,1,0] ，[1] 和 [0,1,2,0] 就不是特殊序列。

给你一个数组 nums （仅包含整数 0，1 和 2），请你返回 不同特殊子序列的数目 。由于答案可能很大，请你将它对 10^9 + 7 取余后返回。

一个数组的子序列是从原数组中删除零个或者若干个元素后，剩下元素不改变顺序得到的序列。如果两个子序列的下标集合不同，那么这两个子序列是不同的。

示例 1：
输入：nums = [0,1,2,2]
输出：3
解释：特殊子序列为 [0,1,2,2]，[0,1,2,2] 和 [0,1,2,2] 。

示例 2：
输入：nums = [2,2,0,0]
输出：0
解释：数组 [2,2,0,0] 中没有特殊子序列。

示例 3：
输入：nums = [0,1,2,0,1,2]
输出：7
解释：特殊子序列包括：
- [0,1,2,0,1,2]
- [0,1,2,0,1,2]
- [0,1,2,0,1,2]
- [0,1,2,0,1,2]
- [0,1,2,0,1,2]
- [0,1,2,0,1,2]
- [0,1,2,0,1,2]
 
提示：
1 <= nums.length <= 10^5
0 <= nums[i] <= 2

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/count-number-of-special-subsequences 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

class Solution {
public:
    int countSpecialSubsequences(vector<int>& nums) {
        long long p0 = 0, p1 = 0, p2 = 0, mod = 1e9+7;
        for(auto n : nums)
        {
            if(n==0)
            {
                p0 = (p0*2+1)%mod;//前面的0有多少种取法，2进制
            }
            else if(n==1)
            {
                p1 = (p0+p1*2)%mod;//以1结尾的种类
                //当前1是第一个，前面0结尾的有 p0种
                //当前1不取，前面以1结尾的有 p1种
                //当前1取，前面以1结尾的有 p1种
                // 三种情况合起来
            }
            else
            {
                p2 = (p1+p2*2)%mod;//同理
            }
        }
        return p2;
    }
};

156 ms 168.1 MB C++

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021/08/01 ，如有侵权请联系 cloudcommunity@tencent.com 删除

https