15. 3Sum

15. 3Sum
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
    	List<List<Integer>> result=new ArrayList<>();
    	Arrays.sort(nums);
   	 	for(int i=0;i<nums.length;i++){
      		if(i>0&&nums[i]==nums[i-1]){
        		continue;
      		}
      	int sum=nums[i],start=i+1,end=nums.length-1;
      	while(start<end){
        	if(nums[start]+nums[end]==-sum){
          		List<Integer> temp=new ArrayList<>();
          		temp.add(sum);
          		temp.add(nums[start]);
          		temp.add(nums[end]);
          		result.add(temp);
          		while (start<end&&nums[start+1]==nums[start]){
            		start++;
          		}
          		start++;
          		while (start<end&&nums[end]==nums[end-1]){
            		end--;
          		}
          		end--;
        	}else if(nums[start]+nums[end]>-sum){
          		end--;
        	}else {
          		start++;
        	}
      	}
    }
    return result;
    }
}
在 NumPy 中,`sum` 函数用于计算数组元素的总和。该函数可以作用于多维数组,并且可以通过指定 `axis` 参数来控制求和的方向。基本语法如下: ```python numpy.sum(a, axis=None, dtype=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>) ``` - `a`:输入的数组。 - `axis`:指定沿着哪个轴进行求和。如果为 `None`,则对数组所有元素求和;如果为整数,则沿着该轴求和。 - `dtype`:指定输出结果的数据类型。 - `out`:可选参数,用于指定输出数组。 - `keepdims`:若为 `True`,则在输出中保留被求和的轴,长度为 1。 - `initial`:指定求和的初始值。 - `where`:指定一个布尔数组,仅对 `where` 为 `True` 的位置求和。 ### 示例 #### 1. 一维数组求和 ```python import numpy as np arr = np.array([1, 2, 3, 4, 5]) total = np.sum(arr) print(total) # 输出 15 ``` #### 2. 二维数组按轴求和 ```python arr = np.array([[1, 2], [3, 4]]) row_sum = np.sum(arr, axis=1) # 按行求和 col_sum = np.sum(arr, axis=0) # 按列求和 print(row_sum) # 输出 [3 7] print(col_sum) # 输出 [4 6] ``` #### 3. 指定初始值 ```python arr = np.array([1, 2, 3]) total = np.sum(arr, initial=10) print(total) # 输出 16 ``` #### 4. 使用 `keepdims` 保留维度 ```python arr = np.array([[1, 2], [3, 4]]) total = np.sum(arr, axis=1, keepdims=True) print(total) # 输出 [[3], [7]] ``` ### 注意事项 - `np.sum` 返回的是一个新的数组或标量,原始数组不会被修改。 - 若数组为空,`np.sum` 返回 `0.0`[^1]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值