2023-03-02 02:15:45 +01:00
|
|
|
/* Problem source: https://leetcode.com/problems/squares-of-a-sorted-array/
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <iostream>
|
2023-03-02 06:23:06 +01:00
|
|
|
#include <memory>
|
2023-03-02 02:15:45 +01:00
|
|
|
#include <vector>
|
|
|
|
|
|
|
|
template <typename T> inline T absolute(T num) {
|
|
|
|
return num > -num ? num : -num;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename T> inline T square(T num) { return num * num; }
|
|
|
|
|
|
|
|
template <typename T>
|
2023-03-02 06:23:06 +01:00
|
|
|
std::unique_ptr<std::vector<T>> solution(const std::vector<T> &sorted_input) {
|
|
|
|
std::unique_ptr<std::vector<T>> sorted_output =
|
|
|
|
std::make_unique<std::vector<T>>(sorted_input.size());
|
2023-03-02 02:15:45 +01:00
|
|
|
unsigned long long p1 = 0, p2 = 1;
|
|
|
|
while (p2 < sorted_input.size()) {
|
|
|
|
if (absolute(sorted_input[p2]) <= absolute(sorted_input[p1]))
|
|
|
|
p1 = p2++;
|
|
|
|
else
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
// std::cout << sorted_input[p1] << ", " << sorted_input[p2] << "\n";
|
|
|
|
|
|
|
|
unsigned long long output_p = 0;
|
|
|
|
while (p1 < p2 && p2 < sorted_input.size()) {
|
|
|
|
if (absolute(sorted_input[p1]) < absolute(sorted_input[p2]))
|
2023-03-02 06:23:06 +01:00
|
|
|
(*sorted_output)[output_p++] = (square(sorted_input[p1--]));
|
2023-03-02 02:15:45 +01:00
|
|
|
else
|
2023-03-02 06:23:06 +01:00
|
|
|
(*sorted_output)[output_p++] = square(sorted_input[p2++]);
|
2023-03-02 02:15:45 +01:00
|
|
|
}
|
|
|
|
while (p1 < p2)
|
2023-03-02 06:23:06 +01:00
|
|
|
(*sorted_output)[output_p++] = square(sorted_input[p1--]);
|
2023-03-02 02:15:45 +01:00
|
|
|
while (p2 < sorted_input.size())
|
2023-03-02 06:23:06 +01:00
|
|
|
(*sorted_output)[output_p++] = square(sorted_input[p2++]);
|
2023-03-02 02:15:45 +01:00
|
|
|
return sorted_output;
|
|
|
|
}
|
|
|
|
|
|
|
|
int main() {
|
2023-03-02 06:23:06 +01:00
|
|
|
std::unique_ptr<std::vector<int>> result =
|
|
|
|
solution(std::vector{-4, -1, 0, 3, 10});
|
|
|
|
for (const int num : *result)
|
2023-03-02 02:15:45 +01:00
|
|
|
std::cout << num << ", ";
|
|
|
|
std::cout << std::endl;
|
|
|
|
}
|