[力扣笔记]189.轮转数组

解题思路

利用了平摊分析的思想，另外开辟一个同样大小的数组来存储每个数的“势能”（需要向右移动的次数）
向右移动一次，势能降低；向左移动一次，势能增加
每次循环遍历数组的每个元素，查看元素的势能，将其与目标位置的元素对换并改变相应势能

在循环结束条件的设置上，一开始我是用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;
        }
    }
}
}