华为OD机试 Python 【报文重排序】

题目

你手里有一堆乱七八糟的消息片段，每个片段后面都跟着一个数字，那个数字就像是每个片段的编号。你需要按照这些数字，将消息片段整合成一个完整的消息，并把那些数字扔掉。

输入：

第一行告诉你有几个消息片段，记作N (0 < N ≤ 1000)。

第二行就是这N个片段，每个片段后都跟着它的编号。
输出：

把这些片段按编号的顺序组合起来

示例：
输入：
4
rolling3 stone4 like1 a2

输出：
like a rolling stone

解释："like"的编号是1，所以它是第一个。

代码

import re

def assemble_message(num_fragments, fragments