1 package Test.CollectionIterator;
2 import java.util.Iterator;
3 public class MyHashSet2 implements Iterable{
4 //1.数组+链表 一个add 方法
5 private Node[] arr;
6 private int size;//返回list中的元素个数
7 public int size() {
8 return size;
9 }
10 public boolean add(E e) {
11 boolean flag = false;
12 //1.判断arr数组是否存在 不存在就创建 存在就添加
13 if (arr == null) {
14 arr = new Node[16];//1.1不存在 创建
15 }
16 //1.2 添加
17 //2.根据分组的下标判断 数组下标处 有没有node元素
18 int index = e.hashCode() % 16;
19 if (arr[index] == null) {
20 arr[index] = new Node(e, null);//2.1没有直接创建
21 size++;
22 flag = true;
23 } else {//2.2有的话比较
24 //3.如果hash值和equals值相等 就不添加 否则添加
25 Node node = arr[index];
26 E e1 = (E)node.getE();
27 Node next;
28 do {//3.1循环判读是否相等相等就不添加
29 if (e1.hashCode() == e.hashCode() && e1.equals(e)) {
30 return flag;//相等不添加
31 }//
32 next = node.getNext();
33 if (next != null) {
34 e1 = (E)next.getE();
35 }
36 } while (next != null);//这步是比较完数组某个链表上的全部元素
37 //3.2不相等 开始添加
38 arr[index] = new Node(e, node);//2.1没有直接创建
39 size++;
40 flag = true;
41 }
42 return flag;
43 }
44 @Override
45 public Iterator iterator() {
46 return new Itr();
47 }
48 private class Itr implements Iterator{
49 private int curser;//记录 节点数组 的下标
50 private int count;
51 private Node currentNode;
52
53 @Override
54 public boolean hasNext() {
55 return count!=size;
56 }
57
58 @Override
59 public E next() {
60 if (count>=size){
61 throw new RuntimeException("没有找到下一个");
62 }
63 currentNode=arr[curser];
64 if(currentNode==null){
65 while (true){
66 curser++;
67 if(curser>16)throw new RuntimeException("没有找到下一个");
68 currentNode=arr[curser];
69 if(currentNode==null){
70 curser++;
71 }else {
72 count++;
73 return (E) currentNode.getE();
74 }
75 }
76 }else{
77 count++;
78 Node now =currentNode;
79 if (currentNode.getNext()==null){
80 curser++;
81 }else {
82 currentNode=currentNode.getNext();
83 }
84 return (E)now.getE();
85 /* count++;
86 if (currentNode.getNext()==null){
87 curser++;
88 }else {
89 currentNode=currentNode.getNext();
90 }//这刚好链表上只有一个元素 没有修改currentNode 自己写bug
91 return currentNode.getE();*/
92 }
93 }
94 }
95 // 该内部类 实现 链表结构
96 private class Node {
97 private E e;
98 private Node next;
99 public Node() {
100 }
101 public Node(E e, Node next) {
102 this.e = e;
103 this.next = next;
104 }
105 public E getE() {
106 return e;
107 }
108 public void setE(E e) {
109 this.e = e;
110 }
111 public Node getNext() {
112 return next;
113 }
114 public void setNext(Node next) {
115 this.next = next;
116 }
117 }
118 }
Original: https://www.cnblogs.com/acman-mr-lee/p/16413241.html
Author: ACMAN-Mr.Lee
Title: 手写 HashSet的底层 和 迭代器
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/644376/
转载文章受原作者版权保护。转载请注明原作者出处！