Cuadrado de matriz ordenada
1. Tema
Enlace al título del enlace.
Le proporciona una matriz de números enteros ordenados en orden no decreciente y devuelve una nueva matriz compuesta por el cuadrado de cada número , que también debe ordenarse en orden no decreciente .
Entrada: nums = [-4,-1,0,3,10]
Salida: [0,1,9,16,100]
Explicación: Después de elevar al cuadrado, la matriz se convierte en [16,1,0,9,100]
Después de ordenar, la matriz se convierte en [0,1,9,16,100]
Entrada: números = [-7,-3,2,3,11]
Salida: [4,9,9,49,121]
- 1 <= nums.longitud <= 104
- -104 <= números[i] <= 104
- nums se ordena en orden no decreciente
2. Idea
2.1 Solución violenta
Cuadrar primero, luego ordenar
Función sort() incorporada de C++, sort(first, last, comp), sort range is [first, last)
comp default value is (operator<), puede personalizar
el método de clasificación predeterminado de la función bool como orden ascendente
La complejidad del tiempo es: O ( n + n ∗ log ( n ) ) = O ( n ∗ log ( n ) ) O(n+n*log(n)) = O(n*log(n))O ( n+norte∗registro ( n ) ) _ _=O ( n∗registro ( n ) ) _ _
2.2 Método de doble puntero
yoEl puntero i apunta acomenzar comenzarb e g i n ,jjEl puntero j apunta alfinal finalfin kk _
_k apunta al finalde la nueva matrize n dDebido a que
la matriz esuna matriz en orden no decreciente, el elemento con el valor máximo después del cuadrado aparece en ambos extremos de la matriz
如果A [ i ] ∗ A [ i ] < A [ j ] ∗ A [ j ] A[i] * A[i] < A[j] * A[j]un [ yo ]∗un [ yo ]<un [ j ]∗UN [ j ]那么resultado [ k − − ] = UN [ j ] ∗ UN [ j ] ; resultado[k--] = A[j] * A[j];resultado [ k _ _ _ _ _−− ]=un [ j ]∗A [ j ] ;
如果A [ i ] ∗ A [ i ] > = A [ j ] ∗ A [ j ] A[i] * A[i] >= A[j] * A[j]un [ yo ]∗un [ yo ]>=un [ j ]∗UN [ j ]那么resultado [ k − − ] = UN [ yo ] ∗ UN [ yo ] ; resultado[k--] = A[i] * A[i];resultado [ k _ _ _ _ _−− ]=un [ yo ]∗un [ yo ] ;
La complejidad del tiempo es: O ( n ) O(n)O ( n )
3. Implementación del código
3.1 Solución violenta
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 Método de doble puntero
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;
}
};