三数之和

前言

这是笔者在lintcode的刷题经历，本着共享的原则分享自己的解题方法。

三数之和系列

题目1:
给出一个有n个整数的数组S，在S中找到三个整数a, b, c，找到所有使得a + b + c = 0的三元组。

• 三元组(a, b, c)，要求a <= b <= c。
• 结果不能包含重复的三元组。

思路： 题目本身是很简单的，简单到我只有最朴素的想法。鉴于结果要求是排序的，因此思路具体如下：

• 将原数组排序
• 从N个数字中挑选3个，时间复杂度为N²级别

• 对所有结果进行筛选，然后从后往前删除重复的三元组（这里读者可以考虑下为什么不是从前往后删）

  代码如下：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    public List<List<Integer>> threeSum(int[] numbers) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (numbers.length <= 2) {
            return ret;
        }
        Arrays.sort(numbers);
        if (numbers[0] > 0 || numbers[numbers.length - 1] < 0) {
            return ret;
        }

        for (int i = 0; i < numbers.length - 2; i++) {
            location: for (int j = i + 1;;) {
                for (int k = numbers.length - 1;;) {
                    if (j == k) {
                        break location;
                    }
                    if (numbers[i] + numbers[j] + numbers[k] == 0) {
                        List<Integer> temp = new ArrayList<Integer>();
                        temp.add(numbers[i]);
                        temp.add(numbers[j]);
                        temp.add(numbers[k]);
                        ret.add(temp);
                        k--;
                    } else if (numbers[i] + numbers[j] + numbers[k] < 0) {
                        j++;
                    } else {
                        k--;
                    }

                }
            }
        }

        for (int i = ret.size() - 1; i > 0; i--) {
            for (int j = i - 1; j >= 0; j--) {
                List<Integer> l1 = ret.get(i);
                List<Integer> l2 = ret.get(j);
                if (l1.equals(l2)) {
                    ret.remove(i);
                    break;
                }
            }
        }

        return ret;
    }
}

题目2：
给一个包含 n 个整数的数组 S, 找到和与给定整数 target 最接近的三元组，返回这三个数的和。
思路其实同上，过于朴素，因此这里直接贴上代码：

public class Solution {
    public int threeSumClosest(int[] numbers, int target) {
        if (numbers.length < 3) {
            return -1;
        }
        int ret = numbers[0] + numbers[1] + numbers[2];
        int min = calculate(ret, target);

        for (int i = 0; i < numbers.length - 2; i++) {
            for (int j = i + 1; j < numbers.length - 1; j++) {
                for (int k = j + 1; j < numbers.length; k++) {
                    int temp = numbers[i] + numbers[j] + numbers[k];
                    if (min > calculate(temp, target)) {
                        ret = temp;
                        min = calculate(temp, target);
                    }
                }
            }
        }
        return ret;
    }

    private int calculate(int ret, int target) {
        if (ret > target) {
            return ret - target;
        }
        return target - ret;
    }
}

结语

谢谢您的阅读，希望对您有所帮助（笑哭，⁄(⁄ ⁄•⁄ω⁄•⁄ ⁄)⁄）