Python基础练习 1008 数组元素循环右移问题 (20 分)
本文已参与「新人创作礼」活动,一起开启掘金创作之路。
问题描述
一个数组A中存有N(>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(≥0)个位置,即将A中的数据由(A0 A1 ⋯AN−1 )变换为(AN−M ⋯AN−1 A0 A1 ⋯AN−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
Python版本 代码如下
在python版本中,我使用的循环右移的方法是利用列表思维,插入数据以及删除数据完成的。
n, m = map(int, input().split())
s = input().split()
for i in range(m):
s.insert(0, s[-1]) ##insert()指定位置插入元素
s.pop() ##pop()指定位置弹出元素,默认弹出最后一个元素(index=-1)
print(" ".join(s))
C++版本 代码如下
在C++版本中,我使用的思路是先将数组前面元素逆置,然后将元素后剩下的元素逆置,最后将整个数组逆置即可。
#include <iostream>
using namespace std;
int num[101];
int main(){
int N, M;
cin >> N >> M;
for (int i = 0; i < N; i++)
cin >> num[i];
M = M % N;
for (int i = N - M, j = N - 1; i < j; i++, j--)
swap(num[i],num[j]);
for (int i = 0, j = N - M - 1; i < j; i++, j--)
swap(num[i],num[j]);
for (int i = 0, j = N - 1; i < j; i++, j--)
swap(num[i],num[j]);
for (int i = 0; i < N - 1; i++)
cout << num[i] << " ";
cout << num[N - 1] << endl;
}