只用旋转一次,总替换次数n次,比经典的3次旋转更优。
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
* 旋转数组
* @param n int整型 数组长度
* @param m int整型 右移距离
* @param a int整型vector 给定数组
* @return int整型vector
*/
// 找到最小公倍数
int findMin(int a, int b){
int n = 0;
int max = a > b ? a : b;
int min = a > b ? b : a;
int cnt = max / min;
while (1) {
n = cnt * min;
if(n % max == 0) break;
++cnt;
}
return n;
}
vector<int> solve(int n, int m, vector<int>& a) {
// write code here
int tmp = 0;
int pre = a[0];
int idx = 0;
m = m % a.size();
if(m == 0) return a;
int max = findMin(m, a.size());
for(int i = 0; i < a.size(); i++){
idx += m;
tmp = a[idx % a.size()];
a[idx % a.size()] = pre;
pre = tmp;
if(idx >= max){
idx = idx % a.size() + 1;
pre = a[idx];
}
}
return a;
}
};
首先需要知道如下事实:
- 最多需要交换n次
- 不能重复移动同一位置的数字
- 因为idx 可能大于size,所以使用idx % size 作为实际的索引
- (idx + (n * k)) % n = idx,如果idx每次都加m,则最终一定会形成循环,并且 idx 最早开始重复的时候一定是 idx 加上 m 和 n 的最小公倍数的时候
因此,每当idx重复的时候,将idx+1,这样就能保证每次交换的位置都是不重复的,并且上一轮的交换已经形成环,并且环中的每个位置都已经完成交换。这样n次之后,整个数组就完成了移动

京公网安备 11010502036488号