题目描述
给你一个待查数组 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
的查询结果。
样例
输入: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] 。
输入:queries = [4,1,2,2], m = 4
输出:[3,1,2,0]
输入: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
算法分析
模拟
- 初始化
g[]
数组,元素分别是[1,2,3...m]
,枚举queries[]
数组中的每个元素,找到该元素在g[]
数组中的具体位置idx
,再将该位置idx
的元素插入到0
位置中(通过将前的元素进行置后)
时间复杂度 $O(n^2)$
Java 代码
class Solution {
public int[] processQueries(int[] queries, int m) {
int n = queries.length;
int[] ans = new int[n];
int[] g = new int[m];
int k = 0;
for(int i = 0;i < m;i ++) g[i] = i + 1;
for(int i = 0;i < n;i ++)
{
//找到当前值的位置
int idx = 0;
for(int j = 0;j < m;j ++)
if(queries[i] == g[j])
idx = j;
ans[k ++] = idx;
//交换位置
int t = g[idx];
for(int j = idx - 1;j >= 0;j --)
{
g[j + 1] = g[j];
}
g[0] = t;
}
return ans;
}
}