java 自习室 day15

leetcode 16题 最接近的三数之和

题目

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数，使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

例如，给定数组 nums = [-1，2，1，-4], 和 target = 1.

与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
思路与三数之和类似

package com.leetcode.medium;

import java.util.Arrays;

public class ThreeSumClosest {
        public int threeSumClosest(int[] nums, int target) {
            Arrays.sort(nums);
            int sum=0;
            int closeNum = nums[0] + nums[1] + nums[2];
            for(int i=0;i<nums.length-2;i++){
                int left=i+1;
                int right = nums.length - 1;
                while(left<right){
                    sum = nums[i]+nums[left]+nums[right];
                    if(Math.abs(sum-target)<Math.abs(closeNum-target)){
                        closeNum = sum;
                    }
                    if(sum>target){while(right>1 && nums[right--]==nums[right]){}}
                    else if(sum<target){while(left<nums.length-1 && nums[left++]==nums[left]){}}
                    else{
                        return target;
                    }
                }
            }
            return closeNum;
        }

        public static void main(String[] args) {
            int[] num = {0,1,1,1};
            int target = 100;
            ThreeSumClosest tnc = new ThreeSumClosest();
            int close = tnc.threeSumClosest(num,target);
        }
    }

leetcode 923题 三数之和的多种可能

题目

给定一个整数数组 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

首先计算nums数组中数字的频度，然后遍历0到target这一区间的数，找到i,j,k，用排列组合的方法算出result。

package com.leetcode.medium;

public class ThreeSumMulti {
    public int threeSumMulti(int[] nums,int target){
        long result=0;
        int length = nums.length;
        int mod = (int)Math.pow(10,9)+7;
        long[] fr = new long[101];
        for (int i = 0; i < length; i++) {
            fr[nums[i]]++;
        }
        for (int i = 0; i <= target; i++) {
            if(i>100 || fr[i]==0){continue;}
            for (int j = i; j <= target; j++) {
                if(j>100 || fr[j]==0){continue;}
                int k = target - i - j;
                if(k>100 | k<0 | k<j || fr[k] ==0 ){
                    continue;
                }
                if(i==j & j==k){
                    result += fr[i]*(fr[i]-1)*(fr[i]-2)/6;
                }
                else if (i==j & j!=k){
                    result += fr[i]*(fr[i]-1)/2 * fr[k];
                }
                else if (i!=j & j==k){
                    result += fr[i] * fr[j] *(fr[j]-1)/2;
                }
                else{
                    result += fr[i]*fr[j]*fr[k];
                }
            }

        }

        return  (int)(result % mod);
    }

    public static void main(String[] args) {
        int[] nums = {0,0,0};

        int target = 0;
        ThreeSumMulti tsm = new ThreeSumMulti();
        int result = tsm.threeSumMulti(nums,target);
    }
}