1409.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/4/25 20:49
# @Author  : Fhh
# @File    : 1409.py
# Good good study,day day up!
"""给你一个待查数组 queries ，数组中的元素为 1 到 m 之间的正整数。
请你根据以下规则处理所有待查项 queries[i]（从 i=0 到 i=queries.length-1）：
一开始，排列 P=[1,2,3,...,m]。
对于当前的 i ，请你找出待查项 queries[i] 在排列 P 中的位置（下标从 0 开始），
然后将其从原位置移动到排列 P 的起始位置（即下标为 0 处）。注意， queries[i] 在 P 中的位置就是 queries[i] 的查询结果。
请你以数组形式返回待查数组  queries 的查询结果。
示例 1：
输入：queries = [3,1,2,1], m = 5
输出：[2,1,2,1]
解释：待查数组 queries 处理如下：
对于 i=0: queries[i]=3, P=[1,2,3,4,5], 3 在 P 中的位置是 2，接着我们把 3 移动到 P 的起始位置，得到 P=[3,1,2,4,5] 。
对于 i=1: queries[i]=1, P=[3,1,2,4,5], 1 在 P 中的位置是 1，接着我们把 1 移动到 P 的起始位置，得到 P=[1,3,2,4,5] 。
对于 i=2: queries[i]=2, P=[1,3,2,4,5], 2 在 P 中的位置是 2，接着我们把 2 移动到 P 的起始位置，得到 P=[2,1,3,4,5] 。
对于 i=3: queries[i]=1, P=[2,1,3,4,5], 1 在 P 中的位置是 1，接着我们把 1 移动到 P 的起始位置，得到 P=[1,2,3,4,5] 。
因此，返回的结果数组为 [2,1,2,1] 。
示例 2：
输入：queries = [4,1,2,2], m = 4
输出：[3,1,2,0]
示例 3：
输入：queries = [7,5,5,8,3], m = 8
输出：[6,5,0,7,5]
提示：
1 <= m <= 10^3
1 <= queries.length <= m
1 <= queries[i] <= m
"""
from typing import List


class Solution:
    def processQueries(self, queries: List[int], m: int) -> List[int]:
        if m <= 0 or not queries:
            return []
        p = dict()
        for i in range(1, m + 1):
            p[i] = i - 1
        res = []
        for i in queries:
            res.append(p[i])
            for j in p:
                if p[j] < p[i]:
                    p[j] += 1
            p[i] = 0
        return res


s = Solution()
print(s.processQueries(queries = [4,1,2,2], m = 4))