LeetCode(1):两数之和

写在前面:基本全部参考大神“Grandyang”的博客,附上网址:http://www.cnblogs.com/grandyang/p/4130379.html

写在这里,是为了做笔记,同时加深理解,希望有识之士一起加油。

Easy!

题目描述

给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

这道题给了我们一个数组,还有一个目标数target,让我们找到两个数字,使其和为target,乍一看就感觉可以用暴力搜索,但是猜到OJ肯定不会允许用暴力搜索这么简单的方法,于是去试了一下,果然是Time Limit Exceeded,这个算法的时间复杂度是O(n^2)。那么只能想个O(n)的算法来实现,由于暴力搜索的方法是遍历所有的两个数字的组合,然后算其和,这样虽然节省了空间,但是时间复杂度高。一般来说,我们为了提高时间的复杂度,需要用空间来换,这算是一个trade off吧,我们只想用线性的时间复杂度来解决问题,那么就是说只能遍历一个数字,那么另一个数字呢,我们可以事先将其存储起来,使用一个HashMap,来建立数字和其坐标位置之间的映射,我们都知道HashMap是常数级的查找效率,这样,我们在遍历数组的时候,用target减去遍历到的数字,就是另一个需要的数字了,直接在HashMap中查找其是否存在即可,注意要判断查找到的数字不是第一个数字,比如target是4,遍历到了一个2,那么另外一个2不能是之前那个2,整个实现步骤为:先遍历一遍数组,建立HashMap映射,然后再遍历一遍,开始查找,找到则记录index。

C++参考答案一:

 1 class Solution {
 2 public:
 3     vector<int> twoSum(vector<int>& nums, int target) {
 4         unordered_map<int, int> m;
 5         vector<int> res;
 6         for (int i = 0; i < nums.size(); ++i) {
 7             m[nums[i]] = i; //先遍历一遍数组,建立HashMap映射
 8         }
//然后再遍历一遍,开始查找,找到则记录index
9 for (int i = 0; i < nums.size(); ++i) { 10 int t = target - nums[i]; 11 if (m.count(t) && m[t] != i) {//if里面的条件用于判断查找到的数字不是第一个数字 12 res.push_back(i); 13 res.push_back(m[t]); 14 break; 15 } 16 } 17 return res; 18 } 19 };

或者可以写得更简洁一些,把两个for循环合并成一个(注意到答案一中两个for循环的循环条件是一模一样的):

C++参考答案二:

 1 class Solution {
 2 public:
 3     vector<int> twoSum(vector<int>& nums, int target) {
 4         unordered_map<int, int> m;
 5         for (int i = 0; i < nums.size(); ++i) {
 6             if (m.count(target - nums[i])) {
 7                 return {i, m[target - nums[i]]};
 8             }
 9             m[nums[i]] = i;
10         }
11         return {};
12     }
13 };

解决思路:

第一个for循环把数组遍历一遍,建立map数据,第二个for循环进行查找,找到符合条件的则记录index。

知识点回顾:(注:下面知识点的页码标注,均依据《C++ primer plus》第六版)

容器(P695)

容器概念:容器是存储在其他对象的对象。被存储的对象必须是同一种类型的。不能将任何类型的对象存储在容器中,具体的说,类型必须是可赋值构造的和可赋值的。

vector(P195)

vector是C++众多容器类型中的一个,是一个十分有用的容器。是数组的一种类表示,它能够像容器一样存放各种类型的对象,简单地说,vector是一个能够存放多种类型的动态数组,能够增加和压缩数据。vector在C++标准模板库中的部分内容,它是一个多功能的,能够操作多种数据结构和算法的模板类和函数库。

使用vector需要注意以下几点:

1、如果你要表示的向量长度较长(需要为向量内部保存很多数),容易导致内存泄漏,而且效率会很低;

2、Vector作为函数的参数或者返回值时,需要注意它的写法:

 double Distance(vector<int>&a, vector<int>&b) 其中的“&”绝对不能少!!!

实例:vector<int>test;

//建立一个vector,int为数组元素的数据类型,test为动态数组名

简单的使用方法如下:

vector<int>test;//建立一个vector

test.push_back(1);

test.push_back(2);//把1和2压入vector,这样test[0]就是1,test[1]就是2。

1 、基本操作

(1)头文件#include<vector>.

(2)创建vector对象,vector<int> vec;

(3)尾部插入数字:vec.push_back(a);

(4)使用下标访问元素,cout<<vec[0]<<endl;记住下标是从0开始的。

(5)使用迭代器访问元素.

vector<int>::iterator it;

for(it=vec.begin();it!=vec.end();it++)

    cout<<*it<<endl;

(6)插入元素:    vec.insert(vec.begin()+i,a);在第i+1个元素前面插入a;

(7)删除元素:    vec.erase(vec.begin()+2);删除第3个元素

vec.erase(vec.begin()+i,vec.end()+j);删除区间[i,j-1];区间从0开始

(8)向量大小:vec.size();

(9)清空:vec.clear();

特别提示:这里有begin()与end()函数、front()与back()的差别

2、重要说明

vector的元素不仅仅可以是int,double,string,还可以是结构体,但是要注意:结构体要定义为全局的,否则会出错。

 1 #include<stdio.h>  
 2 #include<algorithm>  
 3 #include<vector>  
 4 #include<iostream>  
 5 using namespace std;
 6 
 7 typedef struct rect
 8 {
 9     int id;
10     int length;
11     int width;
12 
13     //对于向量元素是结构体的,可在结构体内部定义比较函数,下面按照id,length,width升序排序。  
14     bool operator< (const rect &a)  const
15     {
16         if (id != a.id)
17             return id<a.id;
18         else
19         {
20             if (length != a.length)
21                 return length<a.length;
22             else
23                 return width<a.width;
24         }
25     }
26 }Rect;
27 
28 int main()
29 {
30     vector<Rect> vec;
31     Rect rect;
32     rect.id = 1;
33     rect.length = 2;
34     rect.width = 3;
35     vec.push_back(rect);
36     vector<Rect>::iterator it = vec.begin();
37     cout << (*it).id << ' ' << (*it).length << ' ' << (*it).width << endl;
38     
39     return 0;
40     cin.get();
41     
42 }

3、算法

(1) 使用reverse将元素翻转:需要头文件#include<algorithm>

reverse(vec.begin(),vec.end());将元素翻转,即逆序排列!

(在vector中,如果一个函数中需要两个迭代器,一般后一个都不包含)

(2)使用sort排序:需要头文件#include<algorithm>,

sort(vec.begin(),vec.end());(默认是按升序排列,即从小到大).

可以通过重写排序比较函数按照降序比较,如下:

定义排序比较函数:

bool Comp(const int &a,const int &b)
{
    return a>b;
}
调用时:sort(vec.begin(),vec.end(),Comp),这样就降序排序。 

输出Vector的中的元素   

vector<float> vecClass; 

int nSize = vecClass.size();   

 //打印vecClass,方法一:  

1 for(int i=0;i<nSize;i++)    
2 {    
3    cout<<vecClass[i]<<"     ";    
4 }    
5    cout<<endl;   

需要注意的是:以方法一进行输出时,数组的下标必须保证是整数。

 //打印vecClass,方法二:

1 for(int i=0;i<nSize;i++)    
2 {    
3    cout<<vecClass.at(i)<<"     ";    
4 }    
5    cout<<endl; 

//打印vecClass,方法三:输出某一指定的数值时不方便

1 for(vector<float>::iterator it = vecClass.begin();it!=vecClass.end();it++)    
2 {    
3     cout<<*it<<"   ";    
4 }    
5     cout<<endl; 

二维数组的使用:

 1 #include "stdafx.h"  
 2 #include <cv.h>  
 3 #include <vector>   
 4 #include <iostream>   
 5 using namespace std;
 6 int main()
 7 {
 8     using namespace std;
 9     int out[3][2] = { 1, 2,
10         3, 4,
11         5, 6 };
12     vector <int*> v1;
13 
14     v1.push_back(out[0]);
15     v1.push_back(out[1]);
16     v1.push_back(out[2]);
17 
18     cout << v1[0][0] << endl;//1  
19     cout << v1[0][1] << endl;//2  
20     cout << v1[1][0] << endl;//3  
21     cout << v1[1][1] << endl;//4  
22     cout << v1[2][0] << endl;//5  
23     cout << v1[2][1] << endl;//6  
24 
25     return 0;
26 }

vector中insert()的用法详解

insert() 函数有以下三种用法: 
1、在指定位置loc前插入值为val的元素,返回指向这个元素的迭代器
2、在指定位置loc前插入num个值为val的元素 

3、在指定位置loc前插入区间[start, end)的所有元素 

 1 //创建一个vector,置入字母表的前十个字符   
 2 vector <char> Avector;   
 3 for( int i=0; i < 10; i++ )   
 4   Avector.push_back( i + 65 );   
 5   
 6   
 7 //插入四个C到vector中   
 8 vector <char>::iterator theIterator = Avector.begin();   
 9 Avector.insert( theIterator, 4, 'C' );   
10   
11   
12 //显示vector的内容   
13 for( theIterator = Avector.begin(); theIterator != Avector.end(); theIterator++ )   
14   cout < < *theIterator;   

这段代码需要再完善方可运行,运行结果将显示:CCCCABCDEFGHIJ

unordered_map的定义

详见原网页:http://www.cplusplus.com/reference/unordered_map/unordered_map/

1 template < class Key,
2 class T,
3 class Hash = hash<Key>,
4 class Pred = equal_to<Key>,
5 class Alloc = allocator< pair<const Key,T> >> class unordered_map;

模版参数说明:

Key 
主键的类型。 
在类模板内部,使用其别名为 key_type 的成员类型。 

被映射的值的类型。 
在类模板内部,使用其别名为 mapped_type 的成员类型。 
Hash 
一元谓词,以一个 Key 类型的对象为参数,返回一个基于该对象的 size_t 类型的唯一值。可以是函数指针(Function pointer)类型或函数对象(Function object)类型。在类模板内部,使用其别名为 hasher 的成员类型。 
Pred 
二元谓词,以两个 Key 类型的对象为参数,返回一个 bool 值,如果第一个参数等价于第二个参数,该 bool 值为 true,否则为 false。默认为 std::equal_to.可以是函数指针类型(Function pointer)类型或函数对象(Function object)类型.在类模板内部,使用其别名为 key_equal 的成员类型。 
Alloc 
容器内部用来管理内存分配及释放的内存分配器的类型。这个参数是可选的,它的默认值是 std::allocator,这个是一个最简单的非值依赖的(Value-independent)内存分配器。在类模板内部,使用其别名为 allocator_type 的成员类型。  

unordered_map是一个关联容器,存储key,value.其中元素并没有特别的次序关系 。
特点: 
1. 关联容器中的元素是通过主键(Key)而不是它们在容器中的绝对位置来引用的。 
2. 无序(Unordered)无序容器通过 hash 表来组织它们的元素,允许通过主键快速地访问元素。 
3. 映射(Map)每个元素为一个值(Mapped value)绑定一个键(Key):以主键来标志主要内容等于被映射值的元素。 
4. 键唯一(Unique keys)容器中不存在两个元素有相同的主键。 
5. 能够感知内存分配器的(Allocator-aware)容器使用一个内存分配器对象来动态地处理它的存储需求。

在 unordered_map 内部,元素不会按任何顺序排序,而是通过主键的 hash 值将元素分组放置到 
各个槽(Bucket,也可译成“桶”)中,这样就能通过主键快速地访问各个对应的元素 
(平均耗时为一个常量,即时间复杂度为 O(1))。

成员函数

=================迭代器========================= 
begin 返回指向容器起始位置的迭代器(iterator) 
end 返回指向容器末尾位置的迭代器 
cbegin 返回指向容器起始位置的常迭代器(const_iterator) 
cend 返回指向容器末尾位置的常迭代器 
=================Capacity================ 
size 返回有效元素个数 
max_size 返回 unordered_map 支持的最大元素个数 
empty 判断是否为空 
=================元素访问================= 
operator[] 访问元素 
at 访问元素 
=================元素修改================= 
insert 插入元素 
erase 删除元素 
swap 交换内容 
clear 清空内容 
emplace 构造及插入一个元素 
emplace_hint 按提示构造及插入一个元素 
================操作========================= 
find 通过给定主键查找元素 
count 返回匹配给定主键的元素的个数 
equal_range 返回值匹配给定搜索值的元素组成的范围 
================Buckets====================== 
bucket_count 返回槽(Bucket)数 
max_bucket_count 返回最大槽数 
bucket_size 返回槽大小 
bucket 返回元素所在槽的序号 
load_factor 返回载入因子,即一个元素槽(Bucket)的最大元素数 
max_load_factor 返回或设置最大载入因子 
rehash 设置槽数 
reserve 请求改变容器容量

测试代码:

 1 #include <unordered_map>
 2 #include <iostream>
 3 #include <string>
 4 
 5 using namespace std;
 6 
 7 void PrintIntDoubleUnOrderedMap(unordered_map<int, double>& m, char* pre)
 8 {
 9     unordered_map<int, double>::iterator iter;
10     cout << pre;
11     for (iter = m.begin(); iter != m.end(); ++iter)
12         cout << "(" << iter->first << ", " << iter->second << ")" << " ";
13     cout << endl;
14 }
15 
16 void UnOrderedMapExp1()
17 {
18     unordered_map<int, double> m;
19     //没有key,就自动创建
20     m[0] = 1.11;
21     //普通插入,使用类型转换
22     m.insert(unordered_map<int, double>::value_type(1, 2.22));
23     //带暗示的插入,pair<int, double>就相当于上面的unordered_map<int ,double>
24     m.insert(m.end(), pair<int, double>(2, 3.33));
25     PrintIntDoubleUnOrderedMap(m, "插入元素之前的m:");
26 
27     //插入一个范围
28     unordered_map<int, double> m2;
29     m2.insert(unordered_map<int, double>::value_type(3, 4.44));
30     m2.insert(unordered_map<int, double>::value_type(4, 5.44));
31     m2.insert(unordered_map<int, double>::value_type(5, 6.44));
32     m.insert(m2.begin(), m2.end());
33 
34     m.emplace(6, 5.55);
35     m.emplace_hint(m.end(), 7, 3.09);
36     m.at(5) = 3.333333;
37 
38     PrintIntDoubleUnOrderedMap(m, "插入元素之后m:");
39 
40     unordered_map<int, double>::iterator iter;
41     iter = m.find(4);
42     if (iter != m.end())
43     {
44         cout << "m.find(4): ";
45         cout << "(" << iter->first << ", " << iter->second << ")" << endl;
46     }
47 
48     if (iter != m.end())
49     {
50         m.erase(iter);
51     }
52     PrintIntDoubleUnOrderedMap(m, "删除主键为4的元素之后m:");
53 
54     //遍历删除
55     for (iter = m.begin(); iter != m.end(); ++iter)
56     {
57         if (iter->first == 2)
58         {
59             m.erase(iter);
60             break;
61         }
62     }
63 
64     //内部数据
65     cout << "bucket_count:" << m.bucket_count() << endl;
66     cout << "max_bucket_count:" << m.max_bucket_count() << endl;
67     cout << "bucket_size:" << m.bucket_size(0) << endl;
68     std::cout << "load_factor:" << m.load_factor() << std::endl;
69     std::cout << "max_load_factor:" << m.max_load_factor() << std::endl;
70     PrintIntDoubleUnOrderedMap(m, "删除主键为2的元素后的foo1:");
71     m.clear();
72     PrintIntDoubleUnOrderedMap(m, "清空后的foo1:");
73 }
74 
75 int main()
76 {
77     UnOrderedMapExp1();
78 
79     return 0;
80 }

以类作为key,value

只是用STL提供的基本类型int, char, long等和stringz作为key,value,STL提供了哈希函数和比较函数。但是用自己定义的类时,需要自己定义哈希函数和比较函数

 1 //hash函数
 2 template <typename T>  
 3 class hash  
 4 {  
 5     public:  
 6         size_t operator()(const T& o) const { return 0; }  
 7 };  
 8 
 9 //compare函数
10 template <typename T>  
11 class equal_to  
12 {  
13   public:  
14     bool operator()(const T& a, const T& b) const { return a == b; }  
15 };  

下面以一个学号姓名为例子来实现

 1 #include <unordered_map>
 2 #include <iostream>
 3 #include <string>
 4 
 5 using namespace std;
 6 
 7 //自己设计类,作为key和value
 8 
 9 
10 //学号
11 class Number
12 {
13     string str;
14 public:
15     Number() { }
16     Number(string s) { str = s; }
17 
18     const string& get() const
19     {
20         return str;
21     }
22 };
23 
24 //姓名
25 class Name 
26 {
27     string str;
28 public:
29     Name() {}
30     Name(string s) { str = s; }
31 
32     const string& get() const
33     {
34         return str;
35     }
36 };
37 
38 
39 //哈希函数对象实现
40 // 必须为const
41 class MyHash 
42 {
43 public:
44 
45     size_t operator()(const Number& num) const
46     {
47         hash<string> sh;        //使用STL中hash<string>
48         return sh(num.get());
49     }
50 };
51 
52 //实现equal_to对象
53 //必须为const
54 class MyEqualTo {
55 public:
56 
57     bool operator()(const Number& n1, const Number& n2) const
58     {
59         return n1.get() == n2.get();
60     }
61 };
62 
63 int main()
64 {
65     unordered_map<Number, Name, MyHash, MyEqualTo> map;
66     map.emplace(Number("1000"), Name("A"));
67     map.emplace(Number("1001"), Name("G"));
68     map.emplace(Number("1002"), Name("E"));
69     map.emplace(Number("1003"), Name("D"));
70 
71 
72     unordered_map<Number, Name, MyHash, MyEqualTo>::iterator iter;
73     Number num("1001");
74     iter = map.find(num);
75 
76     if (iter != map.end())
77         cout << "Number: " << iter->first.get() << "," << "Name: " << iter->second.get() << endl;
78 
79     else
80         cout << "Not found!" << endl;
81 
82     return 0;
83 }

官方解答:

方法一:暴力法

暴力法很简单。遍历每个元素x,并查找是否存在一个值与target-x相等的目标元素。Java代码:

 1 public int[] twoSum(int[] nums, int target) {
 2     for (int i = 0; i < nums.length; i++) {
 3         for (int j = i + 1; j < nums.length; j++) {
 4             if (nums[j] == target - nums[i]) {
 5                 return new int[] { i, j };
 6             }
 7         }
 8     }
 9     throw new IllegalArgumentException("No two sum solution");
10 }

复杂度分析:

  • 时间复杂度:O(n2), 对于每个元素,我们试图通过遍历数组的其余部分来寻找它所对应的目标元素,这将耗费O(n) 的时间。因此时间复杂度为 O(n2​​)。

  • 空间复杂度:O(1)

方法二:两遍哈希表

为了对运行时间复杂度进行优化,我们需要一种更有效的方法来检查数组中是否存在目标元素。如果存在,我们需要找出它的索引。保持数组中的每个元素与其索引相互对应的最好方法是什么?哈希表。

通过以空间换取速度的方式,我们可以将查找时间从 O(n)O(n) 降低到 O(1)O(1)。哈希表正是为此目的而构建的,它支持以 近似 恒定的时间进行快速查找。我用“近似”来描述,是因为一旦出现冲突,查找用时可能会退化到 O(n)O(n)。但只要你仔细地挑选哈希函数,在哈希表中进行查找的用时应当被摊销为 O(1)O(1)。

一个简单的实现使用了两次迭代。在第一次迭代中,我们将每个元素的值和它的索引添加到表中。然后,在第二次迭代中,我们将检查每个元素所对应的目标元素(target - nums[i]targetnums[i])是否存在于表中。注意,该目标元素不能是 nums[i]nums[i] 本身!

 1 public int[] twoSum(int[] nums, int target) {
 2     Map<Integer, Integer> map = new HashMap<>();
 3     for (int i = 0; i < nums.length; i++) {
 4         map.put(nums[i], i);
 5     }
 6     for (int i = 0; i < nums.length; i++) {
 7         int complement = target - nums[i];
 8         if (map.containsKey(complement) && map.get(complement) != i) {
 9             return new int[] { i, map.get(complement) };
10         }
11     }
12     throw new IllegalArgumentException("No two sum solution");
13 }

复杂度分析:

  • 时间复杂度:O(n), 我们把包含有 n 个元素的列表遍历两次。由于哈希表将查找时间缩短到 O(1) ,所以时间复杂度为 O(n)。

  • 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表中存储了 n 个元素。

方法三:一遍哈希表

事实证明,我们可以一次完成。在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。

 1 public int[] twoSum(int[] nums, int target) {
 2     Map<Integer, Integer> map = new HashMap<>();
 3     for (int i = 0; i < nums.length; i++) {
 4         int complement = target - nums[i];
 5         if (map.containsKey(complement)) {
 6             return new int[] { map.get(complement), i };
 7         }
 8         map.put(nums[i], i);
 9     }
10     throw new IllegalArgumentException("No two sum solution");
11 }

复杂度分析:

  • 时间复杂度:O(n), 我们只遍历了包含有 n个元素的列表一次。在表中进行的每次查找只花费 O(1)的时间。

  • 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储 n 个元素。





原文地址:https://www.cnblogs.com/ariel-dreamland/p/8662906.html