java中HashSet实现(转)

hashset底层的数据结构是hash表，hash表实现方式采用数组+链表，数组类型为HashNode，每个数组元素为链表的头指针，链表中存储经过hash函数散列后冲突的元素，数组的长度为26

hashset存储的元素类型为字符串，取每个字符串的首字符的ascall码作为hash函数的输入，数组的长度为10，散列函数h(x)=x%10。

HashNode代码如下：

1. public class HashNode {
2. private String msg;
3. private HashNode next;
4. public String getMsg() {
5. return msg;
6. }
7. public void setMsg(String msg) {
8. this.msg = msg;
9. }
10. public HashNode getNext() {
11. return next;
12. }
13. public void setNext(HashNode next) {
14. this.next = next;
15. }
16. public HashNode(String msg, HashNode next) {
17. this.msg = msg;
18. this.next = next;
19. }
20. public HashNode() {
21. }
22. }

hashset实现如下：

1. public class MyHashSet {
2. private HashNode[] nodes = new HashNode[10];
3. private int size = 0;
4. public MyHashSet() {
5. for (int i = 0; i < nodes.length; i++) {
6. nodes[i] = new HashNode();
7. }
8. }
9. public boolean add(String value) {
10. if (contains(value)) { // 如果有这个元素，就不插入
11. return false;
12. }
13. HashNode node = new HashNode(value, null);
14. int index = (int) value.charAt(0) % nodes.length; // 取第一个字符作为hash函数的输入
15. // 如果该链为空，直接插入，否则采用头插法
16. if (nodes[index].getNext() == null) {
17. nodes[index].setNext(node);
18. } else {
19. node.setNext(nodes[index].getNext());
20. nodes[index].setNext(node);
21. }
22. size++;
23. return true;
24. }
25. public boolean remove(String value) {
26. if (!contains(value)) {
27. return false;
28. }
29. int index = (int) value.charAt(0) % nodes.length;
30. HashNode node = nodes[index];
31. HashNode node2 = node.getNext();
32. while (node2 != null) {
33. if (node2.getMsg().equals(value)) {
34. node.setNext(node2.getNext());
35. size--;
36. break;
37. }
38. node = node.getNext();
39. node2 = node.getNext();
40. }
41. return true;
42. }
43. public void display() {
44. for (int i = 0; i < nodes.length; i++) {
45. HashNode node = nodes[i].getNext();
46. System.out.print(i + " :");
47. while (node != null) {
48. System.out.print(node.getMsg() + "  ");
49. node = node.getNext();
50. }
51. System.out.println();
52. }
53. }
54. public int size() {
55. return size;
56. }
57. public boolean contains(String value) {
58. int index = (int) value.charAt(0) % nodes.length;
59. HashNode node = nodes[index].getNext();
60. while (node != null) {
61. if (node.getMsg().equals(value)) {
62. return true;
63. }
64. node = node.getNext();
65. }
66. return false;
67. }
68. }

测试代码：

1. public class TestMyHashSet {
2. public static void main(String[] args) {
3. MyHashSet myHashSet = new MyHashSet();
4. myHashSet.add("hello");
5. myHashSet.add("hey");
6. myHashSet.add("apply");
7. myHashSet.add("你好");
8. myHashSet.add("你是谁");
9. myHashSet.add("cat");
10. myHashSet.add("dog");
11. myHashSet.add("cat");
12. myHashSet.add("你好");
13. System.out.println("包含'你好'? " + myHashSet.contains("你好"));
14. System.out.println("元素个数: " + myHashSet.size());
15. myHashSet.display();
16. myHashSet.remove("hello");
17. System.out
18. .println("*****************after remove 'hello'**********************");
19. myHashSet.display();
20. System.out.println("元素个数: " + myHashSet.size());
21. }
22. }

输出结果：