算法前端算法程序员

跟我一起刷leetCode算法题2之Two Sum

2017-07-29  本文已影响178人  打铁大师

1.Two Sum

这是leetCode第1题,难度Easy

题目

Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.

大致意思是:
给你一个整型数组,要求你返回两个数的索引,它们的和是一个特定的目标。你可以假设每个输入都存在一种结果(返回其中一种结果就行),你不可以使用同一个元素两次。

Example:

 Given nums = [2, 7, 11, 15], target = 9,

  Because nums[0] + nums[1] = 2 + 7 = 9,
  return [0, 1].

下面是我的代码:

  /**
   * @param {number[]} nums
   * @param {number} target
   * @return {number[]}
 */
var twoSum = function(nums, target) {
    var l = nums.length;
    for(var i=0;i<l;i++){
        for(var j=i+1;j<l;j++){
            if(nums[i]+nums[j]===target){
                 result = [i,j];
                 return result;
            }
        }
    }
};

简单测试一下:

 twoSum([3,2,4,5],7) // result: [0,2]
 twoSum([3,2,4,5,8,11,28,16,13,8,7,21],44) //result:[6,7]

OK测试通过了,我很兴奋,这题目真简单。

可是,我的代码执行效率很低。因为我的做法很暴力,循环遍历每个数字,然后寻找是否有一个数字和当前位置的数字的和,等于target。

这种做法的时间复杂度为:O(n^2);

高手们的做法:

其实我们可以一次性完成,不需要在循环中嵌套循环。

高手们的做法:迭代数组并将元素作为对象的属性,索引作为该属性的值。每次都检查一下,对象中是否存在target与当前元素的差值。如果存在,我们已经找到一个解决方案并立即返回。

var twoSum = function(nums, target) {
    var map = {};
    for(var i = 0; i < nums.length; i++) {
        var check = nums[i];
        var search = target - check;
        if(map[search] === undefined) {
            map[check] = i;
        } else {
            return [map[search], i];
        }
    }
};

下面,测试一下两种算法花费的时间大小:

按照下面的代码,创建一个模块Array.js,导出一个长度为10000000的整型数组:

node  product.js

product.js文件的代码如下

var fs = require("fs");
var nums =[];
var boundary = 10000000;
for(var i=0;i<boundary;i++){
    nums.push(Math.floor(Math.random()*boundary));
}
//转成字符串
var str = "var array=["+nums+"];module.exports=array;";
//写入Array.js文件,生成了js代码
fs.writeFile('Array.js', str,  function(err) {
   if (err) {
       return console.error(err);
   }
   console.log("数据写入成功!");
});

使用下面的代码测试算法:

var nums=require('./Array.js');
var start = new Date().getTime();
console.log("result : "+twoSum(nums,877656));
var end = new Date().getTime();
console.log("time: "+(end-start));

第一种算法的结果:

result : 4,5008617
time: 124

第二种算法的结果:

result : 977,24257
time: 22

因为算法不同,可能找到结果不同。
但是,第二种算法的时间,明显更快。

再看下另一个测试结果

 var nums=require('./Array.js');
var start = new Date().getTime();
 console.log("result : "+twoSum(nums,45678));
var end = new Date().getTime();
console.log("time: "+(end-start));

第一种算法的结果:

result : 154,5721777
time: 3893

第二种算法的结果:

result : 42904,74847
time: 72

结果还是第二种算法更快。

上一篇下一篇

猜你喜欢

热点阅读