1
Fork 0

Solve set-mismatch.

This commit is contained in:
Bauke 2022-04-12 15:26:20 +02:00
parent e5fb926c91
commit 0a065d259b
Signed by: Bauke
GPG Key ID: C1C0F29952BCF558
3 changed files with 44 additions and 0 deletions

View File

@ -14,6 +14,7 @@ pub mod palindrome_number;
pub mod plus_one;
pub mod reverse_integer;
pub mod roman_to_integer;
pub mod set_mismatch;
pub mod two_sum;
pub mod valid_anagram;
pub mod valid_palindrome;

View File

@ -0,0 +1,31 @@
pub fn find_error_nums(numbers: Vec<i32>) -> Vec<i32> {
// Uncomment FromIterator in Leetcode since they're using Rust Edition 2018.
// Edition 2021 has this trait in its prelude.
// use std::iter::FromIterator;
use std::collections::HashSet;
let range = 1..=numbers.len() as i32;
let mut numbers_set = HashSet::<_>::new();
let mut duplicate = 0;
for number in numbers {
if numbers_set.insert(number) {
continue;
}
duplicate = number;
}
let mut missing = 0;
for number in range {
if !numbers_set.contains(&number) {
missing = number;
break;
}
}
vec![duplicate, missing]
}

12
tests/set_mismatch.rs Normal file
View File

@ -0,0 +1,12 @@
use leetcode::set_mismatch::find_error_nums;
use test_case::test_case;
#[test_case(&[1, 2, 2, 4], &[2, 3]; "example 1")]
#[test_case(&[1, 1], &[1, 2]; "example 2")]
#[test_case(&[2, 2], &[2, 1]; "edge 1")]
#[test_case(&[3, 2, 3, 4, 6, 5], &[3, 1]; "edge 2")]
#[test_case(&[3, 2, 2], &[2, 1]; "edge 3")]
fn test_set_mismatch(input: &[i32], expected: &[i32]) {
assert_eq!(find_error_nums(input.to_vec()), expected);
}