DEV Community

Sivakumar
Sivakumar

Posted on • Edited on

Data Structures & Algorithms using Rust: Squares of a Sorted Array

As part of Data Structures and Algorithms implementation using Rust, today we will explore Squares of a Sorted Array problem.

Leetcode problem link: https://leetcode.com/problems/squares-of-a-sorted-array/

Problem Statement
Given an array of integers A sorted in non-decreasing order, return an array of the squares of each number, also in sorted non-decreasing order. 
Enter fullscreen mode Exit fullscreen mode
Instructions
Example 1: Input: [-4,-1,0,3,10] Output: [0,1,9,16,100] Example 2: Input: [-7,-3,2,3,11] Output: [4,9,9,49,121] Note: 1 <= A.length <= 10000 -10000 <= A[i] <= 10000 A is sorted in non-decreasing order. 
Enter fullscreen mode Exit fullscreen mode
Solution

Here is my solution to above problem

impl Solution { pub fn sorted_squares(a: Vec<i32>) -> Vec<i32> { let mut ret_arr: Vec<i32> = a.iter().map(|x| x * x).collect(); ret_arr.sort_unstable(); ret_arr } } 
Enter fullscreen mode Exit fullscreen mode

Please feel free to share your comments.

Happy reading!!!

Top comments (0)