Solved 0973

Author: guangsheng.li01

src/a0973_k_closest_points_to_origin.rs

@@ -0,0 +1,38 @@
+/*
+ * [0973] k-closest-points-to-origin
+ */
+
+pub struct Solution {}
+
+// solution impl starts here
+
+impl Solution {
+    pub fn k_closest(points: Vec<Vec<i32>>, k: i32) -> Vec<Vec<i32>> {
+        let mut points = points;
+        points.sort_by_key(|p| p[0] * p[0] + p[1] * p[1]);
+        let mut closest_k = Vec::new();
+        for i in 0..k as usize {
+            closest_k.push(points[i].clone());
+        }
+        closest_k
+    }
+}
+
+// solution impl ends here
+
+// solution tests starts here
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_case0() {
+        assert_eq!(
+            Solution::k_closest(vec![vec![1, 3], vec![-2, 2]], 1),
+            vec![[-2, 2]]
+        );
+    }
+}
+
+// solution tests ends here

src/lib.rs

@@ -27,6 +27,7 @@ mod a0867_transpose_matrix;
 mod a0929_unique_email_addresses;
 mod a0931_minimum_falling_path_sum;
 mod a0937_reorder_data_in_log_files;
+mod a0973_k_closest_points_to_origin;
 mod a1021_remove_outermost_parentheses;
 mod a1047_remove_all_adjacent_duplicates_in_string;
 mod a1200_minimum_absolute_difference;