[Leetcode] 0717. 1 比特与 2 比特字符

717. 1 比特与 2 比特字符

点击上方，跳转至leetcode

题目描述

有两种特殊字符：

• 第一种字符可以用一比特 0 表示
• 第二种字符可以用两比特(10 或 11)表示

给你一个以 0 结尾的二进制数组 bits ，如果最后一个字符必须是一个一比特字符，则返回 true 。

示例 1:

输入: bits = [1, 0, 0]
输出: true
解释: 唯一的解码方式是将其解析为一个两比特字符和一个一比特字符。
所以最后一个字符是一比特字符。

示例 2:

输入：bits = [1,1,1,0]
输出：false
解释：唯一的解码方式是将其解析为两比特字符和两比特字符。
所以最后一个字符不是一比特字符。

提示:

• 1 <= bits.length <= 1000
• bits[i] 为 0 或 1

解法

根据题意，第一种字符一定以 0 开头，第二种字符一定以 1 开头。

我们可以对 bits 数组从左到右遍历。当遍历到 bits[i] 时，如果 bits[i]=0，说明遇到了第一种字符，将 i 的值增加 1；如果 bits[i]=1，说明遇到了第二种字符，可以跳过 bits[i+1](注意题目保证 bits 一定以 0 结尾，所以 bits[i] 一定不是末尾比特，因此 bits[i+1] 必然存在)，将 i 的值增加 2。

上述流程也说明 bits 的编码方式是唯一确定的，遇到1跳过下一个字符，遇到0跳过本字符。看看是否落到最后,
因此若遍历到 i=n−1，那么说明最后一个字符一定是第一种字符。

时间复杂度：\(O(n)\)，其中 \(n\) 是数组 \(bits\) 的长度。

空间复杂度：\(O(1)\)。

Python3

class Solution:
    def isOneBitCharacter(self, bits: List[int]) -> bool:
        i, n = 0, len(bits)
        while i < n - 1:
            i += bits[i] + 1
        return i == n - 1

# bits = [1, 0, 0]
bits = [1,1,1,0]
res = Solution().isOneBitCharacter(bits)
print(res)

C++

#include<iostream>
#include<vector>
using namespace std;


class Solution {
public:
    bool isOneBitCharacter(vector<int>& bits) {
        int i = 0, n = bits.size();
        while (i < n - 1) i += bits[i] + 1;
        return i == n - 1;
    }
};

int main(){

    // vector<int> bits = {1, 0, 0};
    vector<int> bits = {1,1,1,0};
    bool res = Solution().isOneBitCharacter(bits);
    cout << res << endl;
    return 0;
}

//g++ 711.cpp -std=c++11