ソートされた配列の平方
1. トピック
リンクのタイトルへのリンク。非降順でソートされた整数配列 nums を
指定し、各数値の 2 乗で構成される新しい配列を返します。これも非降順でソートする必要があります。
入力: nums = [-4,-1,0,3,10]
出力: [0,1,9,16,100]
説明: 2 乗後、配列は [16,1,0,9,100] になります。
ソート後、配列はは [0,1,9,16,100] になります
入力: nums = [-7,-3,2,3,11]
出力: [4,9,9,49,121]
- 1 <= nums.length <= 104
- -104 <= nums[i] <= 104
- nums は降順ではない順にソートされます
2. アイデア
2.1 暴力的な解決策
まず四角形にしてから並べ替えます
C++ 組み込みの sort() 関数、sort(first、last、comp)、ソート範囲は [first、last) comp のデフォルト値は (operator<)、 bool 関数のデフォルトのソート方法を昇順として
カスタマイズできます。
時間計算量は次のとおりです。O ( n + n ∗ log ( n ) ) = O ( n ∗ log ( n ) ) O(n+n*log(n)) = O(n*log(n))O ( n+n∗l o g ( n ) )=O ( n∗l o g ( n ) )
2.2 ダブルポインタ方式
iiiポインタは開始点を指します。始まり、jj _ _ _ _jポインタはend endを指しますエンドック_ _
_k は新しい配列の終端配列は非減少順の配列なので、正方形以降の最大値を持つ要素が配列の両端に現れます。
結果A [ i ] ∗ A [ i ] < A [ j ] ∗ A [ j ] A[i] * A[i] < A[j] * A[j]あ[ i ]∗あ[ i ]<A [ j ]∗A [ j ]那么結果 [ k − − ] = A [ j ] ∗ A [ j ] ; 結果[k--] = A[j] * A[j];結果[ k _ _ _ _ _−− ]=A [ j ]∗A [ j ] ;
結果A [ i ] ∗ A [ i ] > = A [ j ] ∗ A [ j ] A[i] * A[i] >= A[j] * A[j]あ[ i ]∗あ[ i ]>=A [ j ]∗A [ j ]那么結果 [ k − − ] = A [ i ] ∗ A [ i ] ; 結果[k--] = A[i] * A[i];結果[ k _ _ _ _ _−− ]=あ[ i ]∗A [ i ] ;
時間計算量は次のとおりです: O ( n ) O(n)O ( n )
3. コードの実装
3.1 暴力的な解決策
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
for(int i = 0; i < nums.size(); i++){
nums[i] *= nums[i];
}
sort(nums.begin(), nums.end());
return nums;
}
};
3.2 ダブルポインタ方式
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
// 创建保存返回值的新数组
vector<int> result(nums.size(), 0);
// 数组最后一个位置的index
int k = nums.size() - 1;
// 利用双指针对数组进行遍历,初始化双指针分别指向数组两端
for (int i = 0, j = nums.size() - 1; i <= j; ){
// 注意这里for循环的写法
if(nums[i] * nums[i] < nums[j] * nums[j]){
result[k--] = nums[j] * nums[j];
j--;
}
else{
result[k--] = nums[i] * nums[i];
i++;
}
}
return result;
}
};