PAT Basic Level 1008 数组元素循环右移问题 (思维)
程序员文章站
2022-03-22 13:36:58
...
题目链接:点击查看
题目描述:
一个数组A中存有N(N>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(M>=0)个位置,即将A中的数据由(A0 A1……AN-1)变换为(AN-M …… AN-1 A0 A1……AN-M-1)(最后M个数循环移至最前面的M个位置)。如果需要考虑程序移动数据的次数尽量少,要如何设计移动方法 ?
输入输出:
输入
6 2
1 2 3 4 5 6
输出
5 6 1 2 3 4
题目分析:
右移问题的算法为 ,对于大小为n 的数组 ,要想移动m位 ,现将 整个数组颠倒 ,再将前m位颠倒 ,最后将后n-m位颠倒。若m>n 则 对m取n的余数即可 ,因为移动倍数位位置不变。
代码:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int main()
{
int n,m;
cin>>n>>m;
vector<int>arr(n);
for(int i=0;i<n;i++)
{
cin>>arr[i];
}
m=m%n;//处理m比n大的情况
if(m!=0)
{
reverse(arr.begin(),arr.begin()+n);//现将整个数组颠倒
reverse(arr.begin(),arr.begin()+m);//将前m位颠倒
reverse(arr.begin()+m,arr.begin()+n);//将后n-m位 颠倒
}
cout<<arr[0];
for(int i=1;i<n;i++)
{
cout<<" "<<arr[i];
}
return 0;
}
上一篇: string和vector的基本接口函数