[PTA]自测-3 数组元素循环右移问题

Posted 2021-06-26 Spring-_-Bear

一个数组A中存有N（>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥0）个位置，即将A中的数据由（A₀A₁⋯A_N-1）变换为（A_N-M⋯A_N-1A₀ A₁⋯A_N-M-1）（最后M个数循环移至最前面的M个位置）。如果需要考虑程序移动数据的次数尽量少，要如何设计移动的方法？

输入格式:

每个输入包含一个测试用例，第1行输入N（1≤N≤100）和M（≥0）；第2行输入N个整数，之间用空格分隔。

输出格式:

在一行中输出循环右移M位以后的整数序列，之间用空格分隔，序列结尾不能有多余空格。

输入样例:

6 2
1 2 3 4 5 6

输出样例:

5 6 1 2 3 4

• 提交结果：

• 源码：

#include <stdio.h>
#define MAXN 100

// 将数组a中的元素循环右移m个单位
int ArrayShift(int a[], int n, int m);

int main()
{
    int a[MAXN], n, m;

    scanf("%d %d", &n, &m);

    for (int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
    }

    ArrayShift(a, n, m);

    // 输出循环右移后的数组
    for (int i = 0; i < n; i++)
    {
        if (i != 0)
        {
            printf(" ");
        }

        printf("%d", a[i]);
    }
    printf("\\n");

    return 0;
}

// 将数组a中的元素循环右移m个单位
int ArrayShift(int a[], int n, int m)
{
    // 动态内存分配,创建一个新的数组临时存储循环右移后的数组
    int* p = (int*)malloc(sizeof(int) * n);

    m %= n;   // 避免m过大时重复移动，求模后只移动一次

    for (int i = 0; i < n; i++)
    {
        p[i] = a[(n - m + i) % n];   // a[i]右移m下后其下标变为(n-m+i)%n，需推导求出
    }

    for (int i = 0; i < n; i++)   // 对a[i]数组重新赋值
    {
        a[i] = p[i];
    }

    // 释放内存
    free(p);

    return 0;
}