1409_查询带键的排列

难度：中等

题目

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

解题

容易想到的方法就是对于每一个查询项，先在排列 P 中找到查询项的位置，将其加入答案，然后将其删除，再添加到头部。

具体的，可以用 List 实现。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public int[] processQueries(int[] queries, int m){
  List<Integer> p = new ArrayList<>();
  for (int i = 1; i <= m; i++) {
    p.add(i);
  }
  int[] ans = new int[queries.length];
  for (int i = 0; i < queries.length; i++) {
    int query = queries[i];
    int pos = -1;
    for (int j = 0; j < m; j++) {
      if (p.get(j)==query){
        pos = j;
        break;
      }
    }
    ans[i] = pos;
    p.remove(pos);
    p.add(0,query);
  }
  return ans;
}