旋转数组

题目链接

Solution

模拟。
新建一个vector,先加入后面n-m个元素,然后加入前面m个元素即可。
代码实现较为简单,需要注意下m如果大于n的情况,需要让m模n。

Code

class Solution {
public:
    vector<int> solve(int n, int m, vector<int>& a) {
        vector<int> ans;
        m %= n;
        for (int i = n - m; i < n; ++i) ans.push_back(a[i]);
        for (int i = 0; i < n - m; ++i) ans.push_back(a[i]);
        return ans;
    }
};