本篇博客提供两种解题思路,一是直接不改变数组的内容,直接输出指定格式;二是通过特定算法,不使用额外数组的情况下以最快的方式将数组元素循环右移。
题目内容
一个数组A中存有N(>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(≥0)个位置,如果需要考虑程序移动数据的次数尽量少,要如何设计移动的方法?
输入格式
每个输入包含一个测试用例,第1行输入N(1≤N≤100)和M(≥0);第2行输入N个整数,之间用空格分隔。
输出格式
在一行中输出循环右移M位以后的整数序列,之间用空格分隔,序列结尾不能有多余空格。
输入样例
6 2
1 2 3 4 5 6
输出样例
5 6 1 2 3 4
解法一
指定格式直接进行输出
思路:
假设循环2位,则先输出数组后端的两个元素,再输出数组元素前端的剩余元素即可。注意通过取余操作简化循环多位的情况。
#include <iostream>
#include <vector>
using namespace std;
//注意m为0时的临界情况
int main()
{
int n, m, x;
cin>>n>>m;
vector<int> a(n);
for(int i = 0; i < n; i++)
{
cin>>x;
a[i] = x;
}
m = m%n;
int flag = 1;
for(int i = n-m; i<n; i++)
{
if(flag == 1)
flag = 0;
else
cout<<" ";
cout<<a[i];
}
for(int i = 0; i < n-m; i++)
{
if(flag == 1)
flag = 0;
else
cout<<" ";
cout<<a[i];
}
return 0;
}
解法二
不适用额外数组快速实现数组元素循环右移。
解题速录如下所示,以输入样例为例
初始数组:1 2 3 4 5 6
将后两位反转:1 2 3 4 6 5
将前四位反转:4 3 2 1 6 5
将数组所有元素反转:5 6 1 2 3 4
代码实现:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
int n, m, x;
cin>>n>>m;
vector<int> a(n);
for(int i = 0; i < n; i++)
{
cin>>x;
a[i] = x;
}
m = m%n;
vector<int>::iterator iter = a.end()-1;
iter = iter-m;
reverse(a.begin(), ++iter); //反转指定范围的元素
reverse(iter, a.end());
reverse(a.begin(), a.end());
int flag = 1;
for(int i = 0; i<n; i++)
{
if(flag == 1)
flag = 0;
else
cout<<" ";
cout<<a[i];
}
return 0;
}
在实现时遇到了两个问题:
(1)注意a.end()并不是指向最后一个元素,而是指向最后一个元素后面一个的元素,类似于字符串数组中最后的’\0’
(2)注意reverse(a.begin(),a.end())是反转a.begin()到a.end()减一之间的元素。