LeetCode-705. 设计哈希集合

不使用任何内建的哈希表库设计一个哈希集合（HashSet）。

实现 MyHashSet 类：

void add(key) 向哈希集合中插入值 key 。
bool contains(key) 返回哈希集合中是否存在这个值 key 。
void remove(key) 将给定值 key 从哈希集合中删除。如果哈希集合中没有这个值，什么也不做。
 
示例：

输入：
["MyHashSet", "add", "add", "contains", "contains", "add", "contains", "remove", "contains"]
[[], [1], [2], [1], [3], [2], [2], [2], [2]]
输出：
[null, null, null, true, false, null, true, null, false]

解释：
MyHashSet myHashSet = new MyHashSet();
myHashSet.add(1);      // set = [1]
myHashSet.add(2);      // set = [1, 2]
myHashSet.contains(1); // 返回 True
myHashSet.contains(3); // 返回 False ，（未找到）
myHashSet.add(2);      // set = [1, 2]
myHashSet.contains(2); // 返回 True
myHashSet.remove(2);   // set = [1]
myHashSet.contains(2); // 返回 False ，（已移除）
 

提示：

0 <= key <= 106
最多调用 104 次 add、remove 和 contains 。

 

 

哈希函数：开平方+1

冲突解决：开链法

#include <iostream>
#include <vector>
#include <stdio.h>
#include <math.h>

using namespace std;

/* 哈希方法：开平方+1，开链法 */
typedef struct Node {
	int key;
	struct Node* pnext;
	Node(int v) :key(v), pnext(NULL) {};
}node;

class MyHashSet {
public:
	/** Initialize your data structure here. 初始值用:-1 */
	MyHashSet() {
		arr.clear();
		for (int i = 0; i <= 1001; i++) {
			Node* temp = new Node(-1);
			arr.push_back(temp);
		}
		TrueKey = 0;
	}

	void add(int key) {
		TrueKey = sqrt(key) + 1;
		Node* p = arr[TrueKey];

		while (p->pnext!= NULL) {
			if (p->key == key) {
				return;
			}
			p = p->pnext;
		}
		if (p->key == key) {
			return;
		}
		else {
			Node* insert = new Node(key);
			p->pnext = insert;
		}

	}

	void remove(int key) {
		TrueKey = sqrt(key) + 1;
		Node* p = arr[TrueKey];

		while (p!= NULL) {
			if (p->key == key) {
				p->key = -1;
				break;
			}
			p = p->pnext;
		}
	}

	/** Returns true if this set contains the specified element */
	bool contains(int key) {
		TrueKey = sqrt(key) + 1;
		Node* p = arr[TrueKey];

		while (p != NULL) {
			if (p->key == key) {
				return true;
			}
			p = p->pnext;
		}

		return false;
	}

	/* 打印函数 */
	void print(int key) {
		TrueKey = sqrt(key) + 1;
		Node* p = arr[TrueKey];
		while (p != NULL) {
			cout <<"p:"<< p->key << endl;
			p = p->pnext;
		}
	}

private:
	int TrueKey;
	vector<Node*> arr;
};


int main() {
	MyHashSet myHashSet;
	myHashSet.add(1);     
	myHashSet.add(2);     
	cout << myHashSet.contains(1) << endl; 
	cout << myHashSet.contains(3) << endl;
	myHashSet.add(2);      
	cout << myHashSet.contains(2) << endl;
	myHashSet.remove(2);
	cout << myHashSet.contains(2) << endl;
	myHashSet.add(1000000);
	cout << myHashSet.contains(1000000) << endl;


	system("pause");
	return 0;
}