描述
试设计一个算法,将数组a中的元素a[0]至a[n-1]循环右移k位,并要求只用一个元素大小的附加存储,元素移动或交换次数为O(n)。
输入
先输入一个大于1且小于100的正整数n,再输入n个整数存到数组a中,最后输入一个小于n正整数k,
输出
循环移动k位后输出。
输入样例
5
2 6 15 39 5
3
2 6 15 39 5
3
输出样例
15
39
5
2
39
5
2
6
算法:
1、将整个数组置换(a[i]<=>a[n-1-i])
2、将前k个数置换
3、将后n-k个数置换
要点:
对于a[i],第一次交换后为a[n-1-i],成为后n-k个数,第二次交换后为a[k+i],向右移动(k+i)-i = k位;
对于a[n-1-i],第一次交换后为a[i],成为前k个数,第二次交换后为a[k-1-i],向右移动(n-1-(n-1-i))+(k-1-i+1)=k位;
#include <iostream>
#include <cstdio>
using namespace std;
//交换
void Swap(int *a, int *b)
{
int temp = *a;
*a = *b;
*b = temp;
}
void Reverse(int *a, int *b)
{
//地址比较
while(a < b)
Swap(a++, b--);
}
void Shift(int a[], int k, int n)
{
Reverse(a, a+n-1); //全部交换
Reverse(a, a+k-1); //前k个交换
Reverse(a+k, a+n-1); //剩下部分交换
}
int main()
{
int a[100], n, k;
scanf("%d", &n);
for(int i = 0; i < n; i++)
scanf("%d", &a[i]);
scanf("%d", &k);
Shift(a, k, n);
for(int i = 0; i < n; i++)
printf("%d%c", a[i], i != n-1 ? ' ':'\n');
return 0;
}