PAT乙级 1008.数组元素循环右移问题

205

取余运算循环输出

题目


原题链接

一个数组A中存有N(N>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(M>=0)个位置,即将A中的数据由(A~0~A~1~……A~N-1~)变换为(A~N-M~ …… A~N-1~ A~0~A~1~……A~N-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



解析


取余运算即可,需要额外注意的是,m可能会大于n

代码


C++解法


#include <iostream>
using namespace std;

const int maxn = 105;
int a[maxn];

int main() {
    int n, m;
    cin >> n >> m;
    m %= n;
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
    }
    for (int i = 0; i < n; ++i) {
        if (i)
            cout << " ";
        cout << a[(i - m + n) % n];
    }
    cout << endl;
    return 0;
}


Python解法


(n, m) = input().split()
a = input().split(' ')
n = int(n)
m = int(m) % n
print(" ".join([a[i - m] for i in range(n)]))


Java解法


import java.util.*;

class Main {
    public static final int maxn = 105;
    public static int[] a = new int[maxn];

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int m = in.nextInt() % n;
        for (int i = 0; i < n; ++i) {
            a[i] = in.nextInt();
        }
        for (int i = 0; i < n; ++i) {
            if (i > 0) {
                System.out.print(" ");
            }
            System.out.print(a[(i + n - m) % n]);
        }
        System.out.print("\n");

        in.close();
    }
}
发布评论
  • 点击查看/关闭被识别为广告的评论