需求如题,代码如下。来自LeetCode 1851. 包含每个查询的最小区间
if __name__ == '__main__':
queries = [5,3,4,2]
qindex = list(range(len(queries))) # 生成列表[0,1,2,...,n-1],其中 n = len(queries)
qindex.sort(key=lambda i: queries[i]) # 按照queries[i]的对应值对qindex排序,其实就是保持queries数组不变,对它的下标进行排序
print(qindex)
# 输出结果是[3, 1, 2, 0],自己可以改变queries 的值尝试一下
本文已整合至LeetCode刷题小技巧(Python版)