只用旋转一次,总替换次数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次之后,整个数组就完成了移动