菜雞每日一題系列打卡80天
每天一道算法題目
小夥伴們一起留言打卡
堅持就是勝利,我們一起努力!
題目描述(引自LeetCode)
給定一個排序數組,你需要在原地刪除重複出現的元素,使得每個元素最多出現兩次,返回移除後數組的新長度。
不要使用額外的數組空間,你必須在原地修改輸入數組並在使用 O(1) 額外空間的條件下完成。
示例 1:
給定 nums = [1,1,1,2,2,3],
函數應返回新長度 length = 5, 並且原數組的前五個元素被修改爲 1, 1, 2, 2, 3 。
你不需要考慮數組中超出新長度後面的元素。
示例 2:
給定 nums = [0,0,1,1,1,1,2,3,3],
函數應返回新長度 length = 7, 並且原數組的前五個元素被修改爲 0, 0, 1, 1, 2, 3, 3 。
你不需要考慮數組中超出新長度後面的元素。
題目分析
這是每日一題——刪除排序數組中的重複項的變形題目,所不同的是,之前的題目要求每個元素只出現一次,而本題要求每個元素出現不超過兩次。
因此,這兩道題處理的方式其實並無本質區別,只是本題的處理要稍許複雜一些,需要引入額外的計數變量,記錄當前元素的出現次數。
話不多說,上代碼!
代碼實現
class Solution {
public int removeDuplicates(int[] nums) {
// 特殊情況處理
if (nums == null || nums.length == 0) return 0;
// 計數變量,用於記錄當前元素出現次數
int count = 1;
// 結果變量
int flag = 1;
for (int i = 1; i < nums.length; i++) {
if (nums[i] == nums[i - 1]) {
// 截至目前,當前元素出現不超過2次
if (++count <= 2) nums[flag++] = nums[i];
} else {
// 重置計數變量
count = 1;
nums[flag++] = nums[i];
}
}
// 返回結果
return flag;
}
}
代碼分析
對代碼進行分析,程序對數組進行了一次遍歷,因此,時間複雜度爲O(n);而就空間而言,僅僅使用了常數級別的額外空間,因此,空間複雜度爲O(1)。
執行結果
相關鏈接
學習 | 工作 | 分享
????長按關注“有理想的菜雞”
只有你想不到,沒有你學不到