這是悅樂書的第369次更新,第397篇原創
java
今天介紹的是LeetCode算法題中Easy級別的第231題(順位題號是977)。給定一個整數數組A按有序遞增順序排序,返回每一個數字的平方,也按有序遞增順序返回。例如:算法
輸入:[ - 4,-1,0,3,10]
輸出:[0,1,9,16,100]數組
輸入:[ - 7,-3,2,3,11]
輸出:[4,9,9,49,121]數據結構
注意:翻譯
1 <= A.length <= 10000指針
-10000 <= A[i] <= 10000code
A按有序遞增順序排序。
排序
直接翻譯題目便可,藉助Arrays
的sort
方法來對結果數組排序。class
此解法的時間複雜度是O(N log(N))
,空間複雜度是O(N)
。數據結構與算法
public int[] sortedSquares(int[] A) { int n = A.length, i = 0; int[] result = new int[n]; for (int num : A) { result[i++] = num*num; } Arrays.sort(result); return result; }
不使用新數組來做爲結果數組返回,將A
中的負數變正數,再排序,再取平方。
此解法的時間複雜度是O(N log(N))
,空間複雜度是O(1)
。
public int[] sortedSquares2(int[] A) { int n = A.length; for (int i=0; i<n; i++) { A[i] = A[i]<0 ? -A[i] : A[i]; } Arrays.sort(A); for (int j=0; j<n; j++) { A[j] = A[j]*A[j]; } return A; }
使用計數排序,對A
中的數進行處理,而後計算平方。
此解法的時間複雜度是O(N)
,空間複雜度是O(N)
。
public int[] sortedSquares3(int[] A) { int n = A.length, index = 0; int[] sort = new int[10001]; int len = sort.length; for (int i=0; i<len; i++) { sort[i] = 1; } for (int num : A) { sort[num<0 ? -num : num]++; } int[] result = new int[n]; for (int j=0; j<sort.length; j++) { if (sort[j] > 1) { while (sort[j]-1 > 1) { result[index++] = j*j; sort[j]--; } result[index++] = j*j; } } return result; }
雙指針,先後比較兩個數,若是前面的數的絕對值大於後面的數,那麼結果數組中當前位元素爲前面的元素平方,反之就是後面的元素平方爲結果數組當前位元素。
此解法的時間複雜度是O(N)
,空間複雜度是O(N)
。
public int[] sortedSquares4(int[] A) { int n = A.length; int start = 0, end = n-1; int[] result = new int[n]; for (int i=n-1; i>=0; i--) { if (Math.abs(A[start]) > Math.abs(A[end])) { result[i] = A[start]*A[start]; start++; } else { result[i] = A[end]*A[end]; end--; } } return result; }
算法專題目前已連續日更超過七個月,算法題文章237
+篇,公衆號對話框回覆【數據結構與算法】、【算法】、【數據結構】中的任一關鍵詞,獲取系列文章合集。
以上就是所有內容,若是你們有什麼好的解法思路、建議或者其餘問題,能夠下方留言交流,點贊、留言、轉發就是對我最大的回報和支持!