LeetCode-in-Rust

287. Find the Duplicate Number

Medium

Given an array of integers nums containing n + 1 integers where each integer is in the range [1, n] inclusive.

There is only one repeated number in nums, return this repeated number.

You must solve the problem without modifying the array nums and uses only constant extra space.

Example 1:

Input: nums = [1,3,4,2,2]

Output: 2

Example 2:

Input: nums = [3,1,3,4,2]

Output: 3

Constraints:

Follow up:

Solution

impl Solution {
    pub fn find_duplicate(nums: Vec<i32>) -> i32 {
        let mut arr = vec![0; nums.len() + 1];
        for &num in &nums {
            arr[num as usize] += 1;
            if arr[num as usize] == 2 {
                return num;
            }
        }
        0
    }
}