Update 0188.买卖股票的最佳时机IV.md about rust

This commit is contained in:
fwqaaq
2023-06-30 11:33:23 +08:00
committed by GitHub
parent 4aab5bf815
commit 2444c084b1

View File

@ -439,6 +439,28 @@ function maxProfit(k: number, prices: number[]): number {
}; };
``` ```
Rust
```rust
impl Solution {
pub fn max_profit(k: i32, prices: Vec<i32>) -> i32 {
let mut dp = vec![vec![0; 2 * k as usize + 1]; prices.len()];
for v in dp[0].iter_mut().skip(1).step_by(2) {
*v = -prices[0];
}
for (i, &p) in prices.iter().enumerate().skip(1) {
for j in (0..2 * k as usize - 1).step_by(2) {
dp[i][j + 1] = dp[i - 1][j + 1].max(dp[i - 1][j] - p);
dp[i][j + 2] = dp[i - 1][j + 2].max(dp[i - 1][j + 1] + p);
}
}
dp[prices.len() - 1][2 * k as usize]
}
}
```
<p align="center"> <p align="center">