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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章