向左旋转 K 个位置的 C++ 列表
假设我们有一个数字列表。我们必须定义一种可以向左旋转该列表 k 个元素的方法。
所以,如果输入类似 [5,4,7,8,5,6,8,7,9,2], k = 2,则输出将为 [8,5,6,8,7,9,2,5,4,7]
要解决这个问题,我们将按照以下步骤进行 −
定义数组 ret
n := 数字的大小
k := k mod n
对于 i := k,当 i < n,更新 (i 增加 1) 执行 −
将 nums[i] 插入 ret 的末尾
对于 i := 0,当 i < k,更新 (i 增加 1) 执行 −
将 nums[i] 插入 ret 的末尾
返回 ret
让我们参考以下实现,以便更好地理解 −
示例
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto> v) { cout << "["; for (int i = 0; i < v.size(); i++) { cout << v[i] << ", "; } cout << "]" << endl; } class Solution { public: vector<int> solve(vector<int>& nums, int k) { vector <int> ret; int n = nums.size(); k %= n; for(int i = k; i < n; i++){ ret.push_back(nums[i]); } for(int i = 0; i < k; i++){ ret.push_back(nums[i]); } return ret; } }; main(){ Solution ob; vector<int> v = {5,4,7,8,5,6,8,7,9,2}; print_vector(ob.solve(v, 3)); }
输入
{5,4,7,8,5,6,8,7,9,2}, 2
输出
[8, 5, 6, 8, 7, 9, 2, 5, 4, 7, ]
广告