解题思路
利用了平摊分析的思想,另外开辟一个同样大小的数组来存储每个数的“势能”(需要向右移动的次数)
向右移动一次,势能降低;向左移动一次,势能增加
每次循环遍历数组的每个元素,查看元素的势能,将其与目标位置的元素对换并改变相应势能
在循环结束条件的设置上,一开始我是用flag = flag + a[i](i=0,1,2,…),结果到最后一个测试用例时提示溢出了,我改为当a[i]=0时flag=flag+1就通过了。我也不清除具体原因是什么,希望有大佬帮忙能解答
代码
#include <stdio.h>
#include <string.h>
int main(){
int* nums;
int numSize;
numSize = 7;
nums = (int*)malloc(sizeof(int)*numSize);
nums[0] = 1;
nums[1] = 2;
nums[2] = 3;
nums[3] = 4;
nums[4] = 5;
nums[5] = 6;
nums[6] = 7;
int i;
int k=3;
rotate(nums,numSize,k);
for(i=0;i<numSize;i++)
{
printf("%d\n",nums[i]);
}
}
void rotate(int* nums, int numsSize, int k){
int a[numsSize];
int i;
for(i=0;i<numsSize;i++)
{
a[i] = k;
}
int flag;
flag = 1;
while(flag!=0)
{
flag = 0;
for(i=0;i<numsSize;i++)
{
int j;
j=a[i];
j = j % numsSize;
if(i+j<numsSize&&a[i]!=0)
{
if(a[i+j]!=0)
{
int temp;
temp = nums[i+j];
nums[i+j] = nums[i];
nums[i] = temp;
a[i] = a[i+j]+j;
if(a[i+j]>=numsSize)
{
a[i] = a[i]%numsSize;
}
a[i+j] = 0;
}
}
else if(i+j>=numsSize && a[i]!=0 )
{
j = (i+j)-numsSize;
if(a[j]!=0)
{
int temp;
temp = nums[j];
nums[j] = nums[i];
nums[i] = temp;
a[i] = a[j]-(i-j);
a[j] = 0;
}
}
}
for(i=0;i<numsSize;i++)
{
if(a[i]!=0)
{
flag = flag+1;
}
}
}
}