PAT乙级-1008（数组元素循环右移问题）

 一、解题思路：

解题思路是使用rotate 函数来实现数组的循环右移。下面是具体的步骤：

首先，从标准输入中读取数组的长度 N 和右移位数 M。
然后，创建一个空的 vector 容器 v 来存储输入的数组元素。
通过一个 for 循环，读取输入的 N 个整数，并将它们依次添加到 vector 容器 v 中。
接下来，对右移位数 M 进行处理，通过取模运算符 %，确保 M 的值在数组长度 N 内，避免不必要的循环移动。
使用 rotate 函数，将数组向右移动 M 个位置。rotate 函数接受三个参数：旋转的起始位置、旋转的中间位置、旋转的结束位置。在这里，我们将起始位置设置为数组的末尾，中间位置为末尾减去 M（即右移的位数），结束位置为数组的末尾。
最后，通过迭代器遍历输出右移后的数组元素，注意在输出时处理好元素之间的空格，确保输出格式正确。

二、我的代码：

#include<iostream>

#include<vector>

#include<algorithm>

using namespace std;

int main(){

    int N,M;

    vector<int> v;

    cin>>N>>M;

    for(int i=0;i<N;i++){

        int k;

        cin>>k;

        v.push_back(k);

    }

    rotate(v.begin(),v.begin()+(v.size()-M%v.size()),v.end());

    vector<int>::iterator it = v.begin();

    for(;it!=v.end();it++){

        cout<<*it<<" ";

    }

    return 0;

}

 

三、ai改进代码：

#include<iostream>

#include<vector>

#include<algorithm>

using namespace std;

 

int main(){

    int N, M;

    cin >> N >> M;

 

    if (N <= 0 || M < 0) {

        cout << "Invalid input!" << endl;

        return 1;

    }

 

    vector<int> v(N);

    for(int i = 0; i < N; i++){

        cin >> v[i];

    }

 

    M %= N;

    rotate(v.rbegin(), v.rbegin() + M, v.rend());

 

    for(auto it = v.begin(); it != v.end(); ++it){

        if(it != v.begin()) cout << " ";

        cout << *it;

    }

 

    return 0;

}

四、改进理由

输入校验：可以在读取 N 和 M 后，检查它们的范围是否合法。

对数组长度取余：在计算右移位数时，使用了取余操作符 %，这是一个好的做法，可以确保 M 的值小于数组长度 N。

输出格式：在输出数组元素时，可以使用更简洁的方法，不必在最后一个元素后面输出空格。

使用 auto 关键字：可以使用 auto 关键字来简化代码，特别是在迭代器的声明和循环中。

 

五、总结

利用c++的stl库函数简化算法

今日学习与分享rotate（）函数