update 15.3sum about rust

This commit is contained in:
fw_qaq
2022-10-22 16:42:48 +08:00
committed by GitHub
parent e48ed8a586
commit fddea62f38

View File

@ -655,7 +655,7 @@ impl Solution {
```Rust
// 双指针法
use std::collections::HashSet;
use std::cmp::Ordering;
impl Solution {
pub fn three_sum(nums: Vec<i32>) -> Vec<Vec<i32>> {
let mut result: Vec<Vec<i32>> = Vec::new();
@ -667,21 +667,20 @@ impl Solution {
if i > 0 && nums[i] == nums[i - 1] { continue; }
let (mut left, mut right) = (i + 1, len - 1);
while left < right {
if nums[i] + nums[left] + nums[right] > 0 {
right -= 1;
// 去重
while left < right && nums[right] == nums[right + 1] { right -= 1; }
} else if nums[i] + nums[left] + nums[right] < 0 {
left += 1;
// 去重
while left < right && nums[left] == nums[left - 1] { left += 1; }
} else {
match (nums[i] + nums[left] + nums[right]).cmp(&0){
Ordering::Equal =>{
result.push(vec![nums[i], nums[left], nums[right]]);
// 去重
right -= 1;
left +=1;
while left < right && nums[right] == nums[right + 1] { right -= 1; }
while left < right && nums[left] == nums[left - 1] { left += 1; }
right -=1;
while left < right && nums[left] == nums[left - 1]{
left += 1;
}
while left < right && nums[right] == nums[right+1]{
right -= 1;
}
}
Ordering::Greater => right -= 1,
Ordering::Less => left += 1,
}
}
}