LeetCode 三数之和的多种可能,c++

最新推荐文章于 2023-10-17 15:22:14 发布

Nicolas Lee

最新推荐文章于 2023-10-17 15:22:14 发布

阅读量384

点赞数 1

CC 4.0 BY-SA版权

文章标签： LeetCOde 排列组合

本文链接：https://blog.csdn.net/qq_42835910/article/details/84593806

本文深入探讨了三数之和算法，旨在寻找数组中满足特定条件的三个数，使其和等于目标值。通过详细分析示例，展示了如何计算满足条件的组合数量，并提供了高效的算法实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

三数之和的多种可能
给定一个整数数组 A，以及一个整数 target 作为目标值，返回满足 i < j < k 且 A[i] + A[j] + A[k] == target 的元组 i, j, k 的数量。

由于结果会非常大，请返回结果除以 10^9 + 7 的余数。

示例 1：

输入：A = [1,1,2,2,3,3,4,4,5,5], target = 8
输出：20
解释：
按值枚举（A[i]，A[j]，A[k]）：
(1, 2, 5) 出现 8 次；
(1, 3, 4) 出现 8 次；
(2, 2, 4) 出现 2 次；
(2, 3, 3) 出现 2 次。
示例 2：

输入：A = [1,1,2,2,2,2], target = 5
输出：12
解释：
A[i] = 1，A[j] = A[k] = 2 出现 12 次：
我们从 [1,1] 中选择一个 1，有 2 种情况，
从 [2,2,2,2] 中选出两个 2，有 6 种情况。

提示：

3 <= A.length <= 3000
0 <= A[i] <= 100
0 <= target <= 300

class Solution {
public:
    int threeSumMulti(vector<int>& A, int target) {
        int len=A.size();
	long long count=0;
	long long a[101]={0};
    for(int i=0;i<len;i++){
    	a[A[i]]++; 
    }
	int i,j, k;
        cout<<a[0]<<endl;
	long long mod=1E9 + 7;
    for(i=0;i<=100;i++){
    	for(j=i;j<=100;j++){
    		if(i+j>target){
    			break;
    		}
    		for(k=j;k<=100;k++){
    			if(i+j+k==target){ 
    				if(i==j && j==k){
    					count+=(a[i]*(a[i]-1)*(a[i]-2))/6;
                        cout<<count<<endl;
    				}else if( j==k){
    					count+=a[i]*a[j]*(a[j]-1)/2;
    				} else if(i==j ){
    					count+=a[k]*a[i]*(a[i]-1)/2;
    				}else if(i==k){
    					count+=a[j]*a[i]*(a[i]-1)/2;
    				}else {
						count+=a[i]*a[j]*a[k];				
					}
					count%=mod;
    			}else if( i+j+k>target){
    				break;
    			}
    		}    		
    	}
    }
    return count;
    }
};