算法面试题——TLV解码（python）

题目描述

TLV编码是按[Tag Length Value]格式进行编码的，一段码流中的信元用Tag标识，Tag在码流中唯一不重复，Length表示信元Value的长度，Value表示信元的值。
码流以某信元的Tag开头，Tag固定占一个字节，Length固定占两个字节，字节序为小端序。
现给定TLV格式编码的码流，以及需要解码的信元Tag，请输出该信元的Value。
输入码流的16进制字符中，不包括小写字母，且要求输出的16进制字符串中也不要包含小写字母；码流字符串的最大长度不超过50000个字节。

输入描述

输入的第一行为一个字符串，表示待解码信元的Tag；
输入的第二行为一个字符串，表示待解码的16进制码流，字节之间用空格分隔。

输出描述

输出一个字符串，表示待解码信元以16进制表示的Value。

用例

输入

31
32 01 00 AE 90 02 00 01 02 30 03 00 AB 32 31 31 02 00 32 33 33 01 00 CC

输出

32 33

解析

本题考查的是对字符串、数组和集合操作的理解，首先输入一个16进制的信元Tag，然后再输入一串字符串找到第一行输入信元Tag的Value值。
需要解析的信元的Tag是31，
从码流的起始处开始匹配，
第一个信元的Tag是32，信元长度为1（0100，小端序表示为1）；
第二个信元的Tag是90，其长度为2（0200），返回01 02；
第三个信元的Tag是30，其长度为3（0300），返回AB 32 31；
第四个信元的Tag是31，其长度为2（0200），所以返回长度后面的两个字节即可，即3233。

代码

def print_hex(arr, pos, length):
    res = ' '.join([format(i, '02X') for i in arr[pos:pos + length]])
    print(res)


if __name__ == '__main__':
    target = int(input(), 16)
    arr = [int(i, 16) for i in input().split()]
    pos = 0
    while pos < len(arr):
        t = arr[pos]
        length = (arr[pos + 2] << 8) | arr[pos + 1]
        if target == t:
            print_hex(arr, pos + 3, length)

        pos += length + 3