Re: [閒聊] 每日leetcode

作者: yam276 ('_')   2025-04-27 18:41:32
3392. Count Subarrays of Length Three With a Condition
https://leetcode.com/problems/count-subarrays-of-length-three-with-a-condition/
簡單來說 任意三個數的子陣列切片
中間的數 要是兩邊相加除二
思考:
Sliding Windows
而且 Rust 居然有內建 .widnows() 函數
沒內建的語言用
for i in 0..nums.len()-2 {
let a = nums[i];
let b = nums[i+1];
let c = nums[i+2];
// ...
}
Code:
impl Solution {
pub fn count_subarrays(nums: Vec<i32>) -> i32 {
nums.windows(3)
.filter(|w| w[1] % 2 == 0 && w[1] / 2 == w[0] + w[2])
.count() as i32
}
}

Links booklink

Contact Us: admin [ a t ] ucptt.com