给你一个待查数组 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]
思路
无需创建移动P数组,用一个数组记录1-m每个数字i在P数组中的位置pos[i-1],之后的变化只需改变位置即可
Java实现
class Solution {
public int[] processQueries(int[] queries, int m) {
// 不需要创建和移动P数组,创建如下数组pos记录1-m在P数组中的位置,P[i] 代表数字i在P数组中位置,
// 初始情况下,数组i在P数组中的下标是i-1,即pos[i] = i;
int[] pos = new int[m];
for(int i = 0 ; i< m ; i++){
pos[i] = i;
}
int[] res = new int[queries.length];
for(int i = 0; i<queries.length; i++){
// res[i] 表示queries[i]在数组P中的位置即pos[queries[i]-1]
res[i] = pos[queries[i]-1];
for(int j = 0; j<m; j++ ){
// 将在 queries[i] 前面的数字往后移
if(pos[j]<pos[queries[i]-1]){
pos[j]+=1;
}
}
// 将queries[i] 对应的数组,位置放到最前面,即位置为0
pos[queries[i]-1] = 0;
}
return res;
}
}