PAT-BASIC1008——数组元素循环右移问题

我的PAT-BASIC代码仓：https://github.com/617076674/PAT-BASIC

原题链接：https://pintia.cn/problem-sets/994805260223102976/problems/994805316250615808

题目描述：

知识点：数组

思路：每个位置只移动一次，用一个数组flag来标记该位置是否已经被移动过

如果每次只向右移动一位，那我们对于每个位置就需要移动M次，太浪费时间了。事实上，对于一个位置i，我们是知道其下一个位置将会是(i + M) % N。

我们为什么要每次只移动1位呢？但是如果一次性移动M位又有个问题，位置i到达位置(i + M) % N，那么原来(i + M) % N位置的值也将移动到下一个位置，什么时候其下一个位置刚好是位置i，以此形成一个循环呢？

所以，在移动的时候，我们需要一个标记数组flag来记录该位置是否已经被移动。在遍历原数组移动的时候，如果该位置已经被移动过了，显然我们不需要再次移动该位置。否则，就移动该位置，并且带动一系列值一起移动。

时间复杂度是O(N)，空间复杂度是O(1)。

C++代码：

#include<iostream>
#include<vector>

using namespace std;

int main() {
	int n;
	int m;
	cin >> n >> m;

	if (m >= n) {
		m %= n;
	}

	vector<int> array;
	for (int i = 0; i < n; i++) {
		int temp;
		cin >> temp;
		array.push_back(temp);
	}

	vector<bool> flag;
	for (int i = 0; i < n; i++) {
		flag.push_back(false);
	}

	for (int i = 0; i < array.size(); i++) {
		if (flag[i]) {
			continue;
		}
		flag[i] = true;
		int temp = array[i];
		int index = i;
		while (true) {
			int next = index - m;
			if (next < 0) {
				next += n;
			}
			if (next == i) {
				array[index] = temp;
				break;
			}
			array[index] = array[next];
			flag[next] = true;
			index = next;
		}
	}
	
	for (int i = 0; i < array.size(); i++) {
		cout << array[i];
		if (i != array.size() - 1) {
			cout << " ";
		}
	}
}

C++解题报告：

JAVA代码：

import java.util.ArrayList;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        int[] nums = new int[n];
        for (int i = 0; i < n; i++) {
            nums[i] = scanner.nextInt();
        }
        if(m >= n){
            m = m % n;
        }
        boolean[] flag = new boolean[n];
        for (int i = 0; i < n; i++) {
            if(flag[i]){
                continue;
            }
            ArrayList<Integer> arrayList = new ArrayList<>();
            arrayList.add(i);
            for (int j = 1;; j++) {
                if((i + j * m) % n == i){
                    break;
                }
                arrayList.add((i + j * m) % n);
            }
            remove(nums, arrayList);
            for (int j = 0; j < arrayList.size(); j++) {
                flag[arrayList.get(j)] = true;
            }
        }

        for (int i = 0; i < n; i++) {
            System.out.print(nums[i]);
            if(i != n - 1){
                System.out.print(" ");
            }
        }
    }

    private static void remove(int[] nums, ArrayList<Integer> arrayList){
        int temp = nums[arrayList.get(arrayList.size() - 1)];
        for (int i = arrayList.size() - 1; i >= 1; i--) {
            nums[arrayList.get(i)] = nums[arrayList.get(i - 1)];
        }
        nums[arrayList.get(0)] = temp;
    }
}

JAVA解题报告：