featureLab/bitkylin-algorithm/section-1/leetcode1/RotateArray.java at master · bitkylin/featureLab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
/**
 * <p>给你一个数组，将数组中的元素向右轮转 <code>k</code><em>&nbsp;</em>个位置，其中&nbsp;<code>k</code><em>&nbsp;</em>是非负数。</p>
 *
 * <p>&nbsp;</p>
 *
 * <p><strong>示例 1:</strong></p>
 *
 * <pre>
 * <strong>输入:</strong> nums = [1,2,3,4,5,6,7], k = 3
 * <strong>输出:</strong> <code>[5,6,7,1,2,3,4]</code>
 * <strong>解释:</strong>
 * 向右轮转 1 步: <code>[7,1,2,3,4,5,6]</code>
 * 向右轮转 2 步: <code>[6,7,1,2,3,4,5]
 * </code>向右轮转 3 步: <code>[5,6,7,1,2,3,4]</code>
 * </pre>
 *
 * <p><strong>示例&nbsp;2:</strong></p>
 *
 * <pre>
 * <strong>输入：</strong>nums = [-1,-100,3,99], k = 2
 * <strong>输出：</strong>[3,99,-1,-100]
 * <strong>解释:</strong>
 * 向右轮转 1 步: [99,-1,-100,3]
 * 向右轮转 2 步: [3,99,-1,-100]</pre>
 *
 * <p>&nbsp;</p>
 *
 * <p><strong>提示：</strong></p>
 *
 * <ul>
 * <li><code>1 &lt;= nums.length &lt;= 10<sup>5</sup></code></li>
 * <li><code>-2<sup>31</sup> &lt;= nums[i] &lt;= 2<sup>31</sup> - 1</code></li>
 * <li><code>0 &lt;= k &lt;= 10<sup>5</sup></code></li>
 * </ul>
 *
 * <p>&nbsp;</p>
 *
 * <p><strong>进阶：</strong></p>
 *
 * <ul>
 * <li>尽可能想出更多的解决方案，至少有 <strong>三种</strong> 不同的方法可以解决这个问题。</li>
 * <li>你可以使用空间复杂度为&nbsp;<code>O(1)</code> 的&nbsp;<strong>原地&nbsp;</strong>算法解决这个问题吗？</li>
 * </ul>
 *
 * <ul>
 * </ul>
 *
 * <ul>
 * </ul>
 * <div><div>Related Topics</div><div><li>数组</li><li>数学</li><li>双指针</li></div></div><br><div><li>👍 1190</li><li>👎 0</li></div>
 */

package leetcode1;

public class RotateArray {

    public static void main(String[] args) {
        Solution solution = new RotateArray().new Solution();
    }

    /**
     * 数组多次翻转
     */
    class Solution {
        public void rotate(int[] nums, int k) {
            k = k % nums.length;
            solve(nums, 0, nums.length - 1);
            solve(nums, 0, k - 1);
            solve(nums, k, nums.length - 1);
        }

        private void solve(int[] nums, int i, int j) {
            while (i < j) {
                swap(nums, i++, j--);
            }
        }

        private void swap(int[] nums, int i, int j) {
            int k = nums[i];
            nums[i] = nums[j];
            nums[j] = k;
        }
    }

    /**
     * 使用额外的数组
     */
    class Solution2 {
        public void rotate(int[] nums, int k) {
            int[] res = new int[nums.length];
            for (int i = 0; i < nums.length; i++) {
                res[(i + k) % nums.length] = nums[i];
            }
            System.arraycopy(res, 0, nums, 0, nums.length);
        }
    }
}