Leetcode--Rust--簡單4

121. 買賣股票的最佳時機

/** 
給定一個數組，它的第 i 個元素是一支給定股票第 i 天的價格。
如果你最多隻允許完成一筆交易（即買入和賣出一支股票一次），設計一個算法來計算你所能獲取的最大利潤。
注意：你不能在買入股票前賣出股票。

示例 1:
輸入: [7,1,5,3,6,4]
輸出: 5
解釋: 在第 2 天（股票價格 = 1）的時候買入，在第 5 天（股票價格 = 6）的時候賣出，最大利潤 = 6-1 = 5 。
     注意利潤不能是 7-1 = 6, 因爲賣出價格需要大於買入價格。
示例 2:
輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這種情況下, 沒有交易完成, 所以最大利潤爲 0。
 */
struct Solution {}
impl Solution {
    pub fn max_profit(prices: Vec<i32>) -> i32 {
        if prices.len() <= 1 {
            return 0;
        }

        let mut buy_price: i32 = prices[0];
        let mut profit = 0;
        for i in 1..prices.len() {
            if prices[i] < buy_price {
                buy_price = prices[i];
            } else {
                profit = i32::max(profit, prices[i] - buy_price);
            }
        }

        profit
    }
}

// submission codes end

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_121() {
        assert_eq!(Solution::max_profit(vec![7, 1, 5, 3, 6, 4]), 5);
        assert_eq!(Solution::max_profit(vec![7, 6, 4, 3, 1]), 0);
    }
}

217. 存在重複元素

/** 
 * 給定一個整數數組，判斷是否存在重複元素。

如果任意一值在數組中出現至少兩次，函數返回 true 。如果數組中每個元素都不相同，則返回 false 。
示例 1:
輸入: [1,2,3,1]
輸出: true

示例 2:
輸入: [1,2,3,4]
輸出: false

示例 3:
輸入: [1,1,1,3,3,4,3,2,4,2]
輸出: true
 */
struct Solution {}
impl Solution {
    pub fn contains_duplicate(nums: Vec<i32>) -> bool {
        if nums.is_empty() {
            return false;
        }

        let mut nums = nums;
        nums.sort_unstable();
        let mut prev = nums[0];
        for i in 1..nums.len() {
            if nums[i] == prev {
                return true;
            }

            prev = nums[i];
        }

        false
    }
}

// submission codes end

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_217() {
        assert_eq!(Solution::contains_duplicate(vec![1]), false);
        assert_eq!(Solution::contains_duplicate(vec![]), false);
        assert_eq!(Solution::contains_duplicate(vec![1, 2, 3, 4]), false);
        assert_eq!(Solution::contains_duplicate(vec![1, 2, 3, 1]), true);
    }
}

219. 存在重複元素 II

/** 
給定一個整數數組和一個整數 k，判斷數組中是否存在兩個不同的索引 i 和 j，使得 nums [i] = nums [j]，並且 i 和 j 的差的 絕對值 至多爲 k。
示例 1:
輸入: nums = [1,2,3,1], k = 3
輸出: true

示例 2:
輸入: nums = [1,0,1,1], k = 1
輸出: true

示例 3:
輸入: nums = [1,2,3,1,2,3], k = 2
輸出: false
 */
use std::collections::HashMap;
struct Solution {}
impl Solution {
    pub fn contains_nearby_duplicate(nums: Vec<i32>, k: i32) -> bool {
        if nums.is_empty() {
            return false;
        }

        let mut res = HashMap::with_capacity(nums.len());

        for (i, &num) in nums.iter().enumerate() {
            match res.get(&num) {
                Some(v) => {
                    if i - v <= k as usize {
                        return true;
                    }
                    res.insert(num, i);
                }
                None => {
                    res.insert(num, i);
                }
            }
        }
        false
    }
}

// submission codes end

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_219() {
        assert_eq!(
            Solution::contains_nearby_duplicate(vec![1, 2, 3, 1, 2, 3], 2),
            false
        );
        assert_eq!(
            Solution::contains_nearby_duplicate(vec![1, 2, 3, 1, 2, 3], 3),
            true
        );
    }
}