【id:20】【1分】D. DS顺序表之循环移位

题目描述

顺序表的移位是循环移位，例如顺序表：1，2，3，4，5，6。如果左移1位，即原来的头元素移动到末尾，其它元素向左移1位，变成2，3，4，5，6，1。同理，如果右移1位，即原来的尾元素移动到头，其它元素向右移1位，变成6，1，2，3，4，5。以下是移位的多个例子：

原数据：1，2，3，4，5，6

左移3位：4，5，6，1，2，3，与原数据对比

右移4位：3，4，5，6，1，2，与原数据对比

请编写程序实现顺序表的循环移位操作

输入

第1行输入n表示顺序表包含的·n个数据

第2行输入n个数据，数据是小于100的正整数

第3行输入移动方向和移动的位数，左移方向为0，右移方向为1

第4行输入移动方向和移动的位数，左移方向为0，右移方向为1

注意：移动操作是针对上一次移动后的结果进行的

输出

第一行输出创建后，顺序表内的所有数据，数据之间用空格隔开

第二行输出第一次移位操作后，顺序表内的所有数据，数据之间用空格隔开

第三行输出第二次移位操作后，顺序表内的所有数据，数据之间用空格隔开

样例查看模式 

正常显示查看格式

输入样例1 <-复制

5
11 22 33 44 55
0 2
1 4

输出样例1

11 22 33 44 55 
33 44 55 11 22 
44 55 11 22 33 

#include<iostream>
using namespace std;
int* change(int choice, int* a, int n, int num)
{
    int* b = new int[n];
    if (choice == 0)
    {
        for (int i = 0; i < n; i++)
        {
            int index = i;
            if (index - num < 0)
            {
                index = index - num + n;
            }
            else
            {
                index -= num;
            }
            b[index] = a[i];
        }
        return b;
   }
    else
    {
        for (int i = 0; i < n; i++)
        {
            int index = i;
            if (index +num >= n)
            {
                index += num;
                index -= n;
            }
            else
            {
                index += num;
            }
            b[index] = a[i];
        }
        return b;
    }
}
int main()
{
    int n;
    cin >> n;
    int* arr = new int[n];
    for (int i = 0; i < n; i++)
    {
        cin >> arr[i];
    }
    for (int i = 0; i <n; i++)
    {
        cout << arr[i] << " ";
    }
    cout << endl;
    int choice;
    cin >> choice;
    int num;
    cin >> num;
    int* b = change(choice, arr, n, num);
    for (int i = 0; i < n; i++)
    {
        cout << b[i] << " ";
    }
    cout << endl;
    cin >> choice;
    cin >> num;
    int* c = change(choice, b, n, num);
    for (int i = 0; i < n; i++)
    {
        cout <<c[i] << " ";
    }
    cout << endl;

    return 0;
}