数据结构:旋转数组

作者 : admin 本文共693个字,预计阅读时间需要2分钟 发布时间: 2024-06-10 共2人阅读

数据结构:旋转数组插图

方法1 (三次逆置法):

void reverse(int* nums, int start, int end) {
    while (start < end) {
        int temp = nums[start];
        nums[start] = nums[end];
        nums[end] = temp;
        start++;
        end--;
    }
}

void rotate(int* nums, int numsSize, int k) {
    k = k % numsSize;  // 防止 k 大于数组长度
    if (k == 0) return;  // 如果 k 为 0,则无需旋转

    reverse(nums, 0, numsSize - 1);      // 逆置整个数组
    reverse(nums, 0, k - 1);             // 逆置前 k 个元素
    reverse(nums, k, numsSize - 1);      // 逆置剩余的元素
}

时间复杂度:O(N),空间复杂度:O(1)

方法2(使用临时数组):

void rotate(int* nums, int numsSize, int k) {
    k = k % numsSize;  // 防止 k 大于数组长度
    if (k == 0) return;  // 如果 k 为 0,则无需旋转

    // 分配内存保存最后 k 个元素
    int* temp = (int*)malloc(sizeof(int) * k);
    if (temp == NULL) {
        printf("动态开辟失败
");
        return;
    }

    // 保存最后 k 个元素到临时数组中
    for (int i = 0; i = 0; i--) {
        nums[i + k] = nums[i];
    }

    // 将临时数组中的元素移到前面
    for (int i = 0; i < k; i++) {
        nums[i] = temp[i];
    }

    // 释放临时数组
    free(temp);
}

时间复杂度:O(N) ,空间复杂度:O(N)

本站无任何商业行为
个人在线分享 » 数据结构:旋转数组
E-->