Leetcode刷题之轮转数组
- 一、题目描述
- 二、题目解析
- 方法一:三段逆置
- 题目代码
- 方法二:利用memcpy进行拷贝
- 运行结果
一、题目描述
轮转数组
给定一个整数数组 nums,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例 1:
输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右轮转 1 步: [7,1,2,3,4,5,6]
向右轮转 2 步: [6,7,1,2,3,4,5]
向右轮转 3 步: [5,6,7,1,2,3,4]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
二、题目解析
本题目,我给大家介绍两种解决方法,解决方法如下:
方法一:三段逆置
①、先将前n-k个数据进行逆置,这里需要注意的是最后一个前n-k个元素的下标,应该是n-k-1.
②、第二步,将后k个数据再进行逆置操作。
第三步:整体对两部分进行逆置操作。
题目代码
//三段逆置
void resver(int *a,int left,int right)
{while(left<right){int tmp=a[left];a[left]=a[right];a[right]=tmp;left++;right--;}
}void rotate(int* nums, int numsSize, int k)
{if(k>numsSize){k%=numsSize;}resver(nums,0,numsSize-k-1);//第一步,前n-k个元素逆置resver(nums,numsSize-k,numsSize-1);//第二步,后k个元素逆置resver(nums,0,numsSize-1);//整体逆置
}
需要注意的是,当k大于numsizes时,我们需要对k进行取余操作。例如当k比numsizes大于一时,相当于只进行一次轮转。
方法二:利用memcpy进行拷贝
①、利用malloc函数创建一个临时的数组,存放数据。
②、用memcpy函数将前numsSize-k个数据元素拷贝到tmp的第k个元素位置。
③、将nums数组中剩下的k个元素,拷贝到tmp的起始位置开始。
④、将tmp数组中的所有元素重新拷贝回nums数组中去,并且free掉临时数组tmp.
void rotate(int* nums, int numsSize, int k){if(k>numsSize){k%=numsSize;}int *tmp=(int*)malloc(sizeof(int)*numsSize);memcpy(tmp+k,nums,sizeof(int)*(numsSize-k));memcpy(tmp,nums+(numsSize-k),sizeof(int)*k);memcpy(nums,tmp,sizeof(int)*numsSize);free(tmp);
}