rustgym 0.2.0

struct Solution;

impl Solution {
    fn delete_and_earn(nums: Vec<i32>) -> i32 {
        let n = 10001;
        let mut sum: Vec<i32> = vec![0; n];
        let mut dp: Vec<i32> = vec![0; n];
        for x in nums {
            sum[x as usize] += x;
        }
        dp[1] = sum[1];
        for i in 2..n {
            dp[i] = i32::max(sum[i] + dp[i - 2], dp[i - 1]);
        }
        dp[n - 1]
    }
}

#[test]
fn test() {
    let nums = vec![3, 4, 2];
    let res = 6;
    assert_eq!(Solution::delete_and_earn(nums), res);
    let nums = vec![2, 2, 3, 3, 3, 4];
    let res = 9;
    assert_eq!(Solution::delete_and_earn(nums), res);
}