1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#![allow(dead_code)]
pub fn max_profit(prices: Vec<i32>) -> i32 {
if prices.len() < 2 {
0
} else {
let mut profit = 0;
for i in 0..prices.len() - 1 {
let diff_price = prices[i + 1] - prices[i];
if diff_price > 0 {
profit += diff_price;
}
}
profit
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test1() {
assert_eq!(max_profit(vec![7, 1, 5, 3, 6, 4]), 7);
assert_eq!(max_profit(vec![7, 6, 4, 3, 1]), 0);
assert_eq!(max_profit(vec![6, 1, 3, 2, 4, 7]), 7);
}
}