LeetCode 刷题1---两数之和

Posted 2021-02-02 最美的时光-

/**
 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
 你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
 示例:
 给定 nums = [2, 7, 11, 15], target = 9
 因为 nums[0] + nums[1] = 2 + 7 = 9
 所以返回 [0, 1]
 */

public class demo1 {
    //普通解法
    public static int[] twoSum(int[] nums,int target){
        HashMap<Integer,Integer> m = new HashMap<Integer, Integer>();
        int res[] = new int[2];
        for(int i=0;i<nums.length;++i){
            if(m.containsKey(target-nums[i])){
                res[0]=i;
                res[1]=m.get(target-nums[i]);
                break;
            }
            m.put(nums[i],i);
        }
        return res;
    }
    //二分法解决方案

/**
 * 创建一个辅助类数组，对辅助类进行排序，使用两个指针，开始时分别指向数组的两端，
 * 看这两个下标对应的值是否等于目标值，如果等于就从辅助类中找出记录的下标，构造好返回结果，返回。
 * 如果大于就让右边的下标向左移，进入下一次匹配，如果小于就让左边的下标向右移动，进入下一次匹配，
 * 直到所有的数据都处理完
 */

    /**
     * 辅助类
     */
    private static class Node implements Comparable<Node> {
        int val; // 值
        int idx; // 值对应的数组下标

        public Node() {
        }

        public Node(int val, int idx) {
            this.val = val;
            this.idx = idx;
        }

        // 比较方法
        public int compareTo(Node o) {
            if (o == null) {
                return -1;
            }
            return this.val - o.val;
        }
    }
    public static int[] twoSum2(int[] nums, int target) {
        // 用于保存返回结果
        int[] result = {0, 0};

        // 创建辅助数组
        Node[] tmp = new Node[nums.length];
        for (int i = 0; i < nums.length; i++) {
            tmp[i] = new Node(nums[i], i);
        }

        // 对辅助数组进行排序
        Arrays.sort(tmp);

        // 记录辅助数组中左边一个值的下标
        int lo = 0;
        // 记录辅助数组中右边一个值的下标
        int hi = nums.length - 1;

        // 从两边向中间靠陇进行求解
        while (lo < hi) {
            // 如果找到结果就设置返回结果，并且退出循环
            if (tmp[lo].val + tmp[hi].val == target) {

                if (tmp[lo].idx > tmp[hi].idx) {
                    result[0] = tmp[hi].idx + 1;
                    result[1] = tmp[lo].idx + 1;
                } else {
                    result[0] = tmp[lo].idx + 1;
                    result[1] = tmp[hi].idx + 1;
                }
                break;
            }
            // 如果大于目标值，右边的下标向左移动
            else if (tmp[lo].val + tmp[hi].val > target) {
                hi--;
            }
            // 如果小于目标值，左边的下标向右移动
            else {
                lo++;
            }
        }
        return result;
    }
    public static void main(String args[]) {
        long startTime = System.currentTimeMillis();
        int arr[]={5,6,7,8,9,10,1,12,3,4,2,3,4,5,7,11};
        int target=20;
        int res[]=twoSum(arr,target);
        for(int i=0;i<res.length;++i){
            System.out.println(res[i]);
        }
        long endTime = System.currentTimeMillis();
        System.out.println("程序运行时间：" + (endTime - startTime) + "ms");
    }
}