PAT(Basic Level) 1008 数组元素循环右移问题 (20)

让我们定义 dn 为：dn = pn+1 - pn，其中 pi 是第i个素数。显然有 d1=1 且对于n>1有 dn 是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。

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

输入格式

每个输入包含一个测试用例，第1行输入N ( 1<=N<=100)、M（M>=0）；第2行输入N个整数，之间用空格分隔。

输出格式

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

输入样例

6 2
1 2 3 4 5 6

输出样例

5 6 1 2 3 4

代码实现

C语言

#include<stdio.h>
#include<malloc.h>

int main() {
    int n;
    int move_bits;

    scanf("%d %d", &n, &move_bits);
    int* A = (int*)malloc(n * sizeof(int));

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

    if ( move_bits > n ) {
        move_bits -= n;
    }

    int flag = n - move_bits - 1;
    for ( flag; flag >= 0; flag-- ) {
        int tmp_flag = flag;
        for ( tmp_flag; tmp_flag - flag < move_bits; tmp_flag++ ) {
            int t;
            t = A[tmp_flag];
            A[tmp_flag] = A[tmp_flag + 1];
            A[tmp_flag + 1] = t;
        }
    }

    for ( int i = 0; i < n; i++) {
        if ( i == n - 1 ) {
            printf("%d", A[i]);
        } else {
            printf("%d ", A[i]);
        }
    }
    return 0;
}

Python

move_str = input().split()
num_str = input().split()

move_num = int(move_str[1])
num_str_len = len(num_str)

if move_num > num_str_len:
    move_num -= num_str_len

if move_num > 0:
    output_str = ' '.join(num_str[0 - move_num:] + num_str[:num_str_len - move_num])
else:
    output_str = ' '.join(num_str)

print(output_str)