Leetcode1389,数组在指定位置插入元素,C代码实现(考虑原数组元素连续和不连续两种情况)

Leetcode1389

我们学习的链表插入元素,是O(1)的操作
数组插入元素,如果原数组中的元素是连续存放的,如[0, 1 ,2 ,3 ,4 ,5 ,NULL, NULL, NULL],这样的情景比较常见,链表也是这种情形。
但是,也有原数组中的元素没有连续存放的情况,如[0, 1, NULL, 2, 3, NULL, 4, 5, NULL],这样的情景就要留心了。

下面是综合考虑以上两种情况的代码实现,同样是Leetcode1389的AC代码

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */

int isClear(int* returnAns, int index){
    int ret = 1;
    if(returnAns[index] != -1)ret = 0;
    return ret;
}

/*输入范围
  1 <= nums.length, index.length <= 100
  nums.length == index.length
  0 <= nums[i] <= 100
  0 <= index[i] <= i

  来源:力扣(LeetCode)
  链接:https://leetcode-cn.com/problems/create-target-array-in-the-given-order
  著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
 */
int* createTargetArray(int* nums, int numsSize, int* index, int indexSize, int* returnSize){
    int* returnAns = (int *)malloc(numsSize*sizeof(int));
    //returnAns数组初始化为-1
    for(int i=0;i<numsSize;i++){
        returnAns[i] = -1;
    }
    *returnSize = numsSize;
    int j;
    for(int i=0;i<indexSize;i++){
        if(isClear(returnAns, index[i])==1){//当前index[i]位置处干净
            returnAns[index[i]] = nums[i];
        }else{
            //把index[i]~末尾处的值向右平移一个单位长度
            //计算连续长度
            for(j=index[i];(isClear(returnAns, j)!=1);j++);//输入范围保证了j不会越界
            for(int k=j;k>index[i];k--){//此处注意判定条件是k>index[i],而非k>=index[i]
                returnAns[k] = returnAns[k-1];
            }
            //把nums[i]填到index[i]位置处
            returnAns[index[i]] = nums[i];
        }
    }
    return returnAns;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章