1
0
Fork 0
mirror of https://gitlab.com/mfocko/LeetCode.git synced 2024-11-14 01:49:41 +01:00
LeetCode/rs/intersection-of-two-arrays.rs

18 lines
484 B
Rust
Raw Normal View History

use std::collections::HashSet;
impl Solution {
pub fn intersection(mut nums1: Vec<i32>, mut nums2: Vec<i32>) -> Vec<i32> {
// if nums1 is bigger, swap them
if nums1.len() > nums2.len() {
return Self::intersection(nums2, nums1);
}
let nums: HashSet<i32> = nums1.into_iter().collect();
nums2.sort_unstable();
nums.into_iter()
.filter(|x| nums2.binary_search(x).is_ok())
.collect()
}
}