蓝桥杯——基础练习——数组移动

问题描述

　　初始数组A[N]中为1,2,..,N，N个数字，现要进行M次操作，每次操作给定一个数字i，记其在数组中的位置为Bi，将A[1]..A[Bi]移到数组末尾。

输入格式

　　输入的第一行包含两个整数N,M。接下来M行，每行一个正整数，表示给定的数字i。

输出格式

　　一行，输出M次操作后的A数组。

一开始没读懂题，蓝桥杯题目给的解释太模棱两可了(@_@)，其实题目讲了那么长，输入M行，只用考虑最后一行就行了，搞不懂输入M行的意义在哪。。。

我这里用到的是队列来解决这个问题的，思路也很简单，但是时间复杂度、空间复杂度应该蛮高的

import java.util.LinkedList;
import java.util.Scanner;

public class Main {
    static LinkedList<Integer> list = new LinkedList<>();
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt(), M = sc.nextInt();
        int[] arr = new int[N];
        int[] m = new int[M];
        for (int i = 0; i < N; i++) {
            arr[i] = i + 1;
            list.add(i + 1);
        }

        for (int i = 0; i < M; i++) {
            m[i] = sc.nextInt();
        }

        arr = fun(m[M - 1]);

        for (int i : arr) {
            System.out.print(i + " ");
        }
    }

    public static int[] fun(int i) {
        for (int j = 0, k = 0; j < list.size(); j++) {
            if (list.get(k) <= i) {
                list.add(list.remove(k));
            } else {
                k++;
            }
        }

        return list.stream().mapToInt(Integer::valueOf).toArray();
    }

}