X

LEETCODE 1409. Queries on a Permutation With Key 解题思路分析

题目大意:

查询带键的排列

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

如果想查看本题目是哪家公司的面试题,请参考以下免费链接: https://leetcode.jp/problemdetail.php?id=1409

解题思路分析:

本题我们可以维护一个数组pos[],记录当前所有数字的位置,初始时数字的位置对应其本身的数值减一(数字是从1开始,而位置下标从0开始)。即:

pos[1]=0; // 数字1在第0位
pos[2]=1; // 数字2在第1位
pos[3]=2; // 数字3在第2位
...
pos[n]=n-1; // 数字n在第n-1位

接下来我们循环queries数组中的每一个数字,利用pos数组查看当前数字所在的位置,并将该位置计入返回结果。然后,由于当前数字会从当前位置移动到数组的首位,所以,当前数字的位置变为0,其他小于当前数字的位置都要加一,位置大于当前数字的位置保持不变。重复此步骤,直到循环完所有 queries 中的数字。

实现代码:

public int[] processQueries(int[] queries, int m) {
    int[] pos = new int[m+1];
    for(int i=1;i<=m;i++){
        pos[i]=i-1;
    }
    int[] res = new int[queries.length];
    for(int i=0;i<queries.length;i++){
        int q = queries[i];
        int p = pos[q];
        res[i]=p;
        for(int j=1;j<=m;j++){
            if(pos[j]<p) pos[j]++;
        }
        pos[q]=0;
    }
    return res;
}

本题解法执行时间为7ms。

Runtime: 7 ms, faster than 86.34% of Java online submissions for Queries on a Permutation With Key.

Memory Usage: 39.7 MB, less than 100.00% of Java online submissions for Queries on a Permutation With Key.

本网站文章均为原创内容,并可随意转载,但请标明本文链接
如有任何疑问可在文章底部留言。为了防止恶意评论,本博客现已开启留言审核功能。但是博主会在后台第一时间看到您的留言,并会在第一时间对您的留言进行回复!欢迎交流!
本文链接: http://leetcode.jp/leetcode-1409-queries-on-a-permutation-with-key-解题思路分析/
Categories: leetcode
kwantong: