20.4.12 周赛 查询带键的排列 中等

image.png
时间复杂度O(m*n),空间复杂度不会分析。。。

题目

给你一个待查数组 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

解题思路

  1. 找待查项在P中的索引,然后比这个索引小的索引全都后移一位,最后该索引改回0。

代码思路

  1. map P的key存正整数1~m,value存正整数的索引,初始化时按升序排列;
  2. 循环遍历待查数组queries,找到当前元素在P的索引,放进结果数组result;
  3. 放进后更新P中的元素位置,因为至多要移动m次,所以循环条件设置为小于等于m,但是只有遍历的索引小于当前元素在P的索引,才会移动,全部移动完毕后,再将当前元素移到P的初始位置。

代码

class Solution {
public:
    vector<int> processQueries(vector<int>& queries, int m) {
        vector<int> result;
        map<int, int> P;
        for(int i = 1; i <= m; i++) P[i] = i-1;
        for(int i = 0; i < queries.size(); i++){
            result.push_back(P[queries[i]]);
            for(int j = 1; j <= m; j++){
                if(P[j] < P[queries[i]]) P[j]++;
            }
            P[queries[i]] = 0;
        }
        return result;
    }
};
posted @ 2020-04-12 13:10  肥斯大只仔  阅读(95)  评论(0编辑  收藏  举报